Skip to content

Commit 8c7a2b5

Browse files
committed
add unit test for visualization properties
1 parent 001197e commit 8c7a2b5

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tests/unit/metadata/test_metadata.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -877,6 +877,19 @@ def test_copy(self, metadata_instance):
877877
assert isinstance(copied_metadata, Metadata)
878878
assert metadata_instance.to_dict() == copied_metadata.to_dict()
879879

880+
@patch('graphviz.Digraph')
881+
def test_visualize_properties(self, mock_digraph, metadata_instance):
882+
"""Test the ``visualize`` method properties"""
883+
# Run
884+
metadata_instance.visualize()
885+
886+
# Assert
887+
mock_digraph.assert_called_with(
888+
'Metadata',
889+
format=None,
890+
node_attr={'shape': 'Mrecord', 'fillcolor': '#B7E9FF', 'style': 'filled'},
891+
)
892+
880893
@patch('sdv.metadata.multi_table.visualize_graph')
881894
def test_visualize_with_sequence_key_and_index(self, visualize_graph_mock):
882895
"""Test the ``visualize`` method with sequence key and index"""

0 commit comments

Comments
 (0)