From fb67bc651bdc87c4517a02dc9df75cad92f4ae8b Mon Sep 17 00:00:00 2001 From: R-Palazzo Date: Mon, 13 Nov 2023 15:21:38 -0600 Subject: [PATCH] add unit test --- .../multi_table/test_base_multi_table_report.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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 index a5f7c4d4..cbabefa6 100644 --- 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')