update test

sdv-dev · Nov 16, 2023 · 52aa57c · 52aa57c
1 parent 1551d07
commit 52aa57c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/integration/reports/multi_table/test_diagnostic_report.py b/tests/integration/reports/multi_table/test_diagnostic_report.py
@@ -172,13 +172,12 @@ def test_get_details_with_table_name(self):
 
         pd.testing.assert_frame_equal(details, expected_dataframe)
 
+    @pytest.mark.filterwarnings('error::UserWarning')
     def test_metadata_without_relationship(self):
-        """Test that no warning is raised when the metadata does not contain relationships."""
         # Setup
         real_data, synthetic_data, metadata = load_demo(modality='multi_table')
         del metadata['relationships']
         report = DiagnosticReport()
 
         # Run and Assert
-        with pytest.warns(None):
-            report.generate(real_data, synthetic_data, metadata)
+        report.generate(real_data, synthetic_data, metadata)