testing metadata is returned

sdv-dev · Sep 12, 2024 · 987e583 · 987e583
1 parent cef7ec4
commit 987e583
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit/metadata/test_metadata.py b/tests/unit/metadata/test_metadata.py
@@ -554,7 +554,7 @@ def test_detect_from_dataframes(self, mock_metadata):
         data = {'guests': guests_table, 'hotels': hotels_table}
 
         # Run
-        Metadata.detect_from_dataframes(data)
+        metadata = Metadata.detect_from_dataframes(data)
 
         # Assert
         mock_metadata.return_value.detect_table_from_dataframe.assert_any_call(
@@ -564,6 +564,7 @@ def test_detect_from_dataframes(self, mock_metadata):
             'hotels', hotels_table
         )
         mock_metadata.return_value._detect_relationships.assert_called_once_with(data)
+        assert metadata == mock_metadata.return_value
 
     def test_detect_from_dataframes_bad_input(self):
         """Test that an error is raised if the dictionary contains something other than DataFrames.