Change metadata visualization background color (#2762)

Author: sarahmish

sdv/metadata/visualization.py

@@ -6,6 +6,7 @@
 import graphviz
 
 DEFAULT_SDTYPES = ['id', 'numerical', 'categorical', 'datetime', 'boolean']
+BACKGROUND_COLOR = '#B7E9FF'
 
 
 def create_columns_node(columns):
@@ -95,7 +96,7 @@ def visualize_graph(nodes, edges, filepath=None):
     digraph = graphviz.Digraph(
         'Metadata',
         format=graphviz_extension,
-        node_attr={'shape': 'Mrecord', 'fillcolor': 'lightgoldenrod1', 'style': 'filled'},
+        node_attr={'shape': 'Mrecord', 'fillcolor': BACKGROUND_COLOR, 'style': 'filled'},
     )
 
     for name, label in nodes.items():

tests/unit/metadata/test_metadata.py

@@ -877,6 +877,19 @@ def test_copy(self, metadata_instance):
         assert isinstance(copied_metadata, Metadata)
         assert metadata_instance.to_dict() == copied_metadata.to_dict()
 
+    @patch('graphviz.Digraph')
+    def test_visualize_properties(self, mock_digraph, metadata_instance):
+        """Test the ``visualize`` method properties"""
+        # Run
+        metadata_instance.visualize()
+
+        # Assert
+        mock_digraph.assert_called_with(
+            'Metadata',
+            format=None,
+            node_attr={'shape': 'Mrecord', 'fillcolor': '#B7E9FF', 'style': 'filled'},
+        )
+
     @patch('sdv.metadata.multi_table.visualize_graph')
     def test_visualize_with_sequence_key_and_index(self, visualize_graph_mock):
         """Test the ``visualize`` method with sequence key and index"""