add unit test

sdv-dev · Nov 13, 2023 · fb67bc6 · fb67bc6
1 parent b4b9644
commit fb67bc6
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/unit/reports/multi_table/test_base_multi_table_report.py b/tests/unit/reports/multi_table/test_base_multi_table_report.py
@@ -304,3 +304,18 @@ def test_get_visualization_without_table_name(self):
 
         with pytest.raises(ValueError, match=expected_error_message):
             report.get_visualization('Property_1')
+
+    def test_get_visualization_for_structure_property(self):
+        """Test the ``get_visualization`` method for the structure property."""
+        # Setup
+        report = BaseMultiTableReport()
+        report._properties = {
+            'Data Structure': Mock()
+        }
+        report._properties['Data Structure'].get_visualization = Mock()
+
+        # Run
+        report.get_visualization('Data Structure', 'Table_1')
+
+        # Assert
+        report._properties['Data Structure'].get_visualization.assert_called_once_with('Table_1')