assert table_names

sdv-dev · Nov 21, 2023 · 38a79ca · 38a79ca
1 parent f150da4
commit 38a79ca
Showing 1 changed file with 1 addition 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
@@ -184,6 +184,7 @@ def test_generate(self, mock_generate):
         report.generate(real_data, synthetic_data, metadata)
 
         # Assert
+        assert report.table_names == ['Table_1', 'Table_2']
         mock_generate.assert_called_once_with(real_data, synthetic_data, metadata, True)
 
     def test__check_table_names(self):