assert_called_once

sdv-dev · Nov 2, 2023 · 7522d31 · 7522d31
1 parent d0dc937
commit 7522d31
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sdmetrics/reports/multi_table/_properties/__init__.py b/sdmetrics/reports/multi_table/_properties/__init__.py
@@ -22,6 +22,6 @@
     'InterTableTrends',
     'Synthesis',
     'Structure',
-    'DataValidity'
+    'DataValidity',
     'RelationshipValidity',
 ]
diff --git a/tests/unit/reports/multi_table/_properties/test_relationship_validity.py b/tests/unit/reports/multi_table/_properties/test_relationship_validity.py
@@ -173,7 +173,7 @@ def test_get_score(
 
         assert score == 0.5
         progress_bar.update.assert_called()
-        assert progress_bar.update.call_count == 1
+        progress_bar.update.assert_called_once()
         mock_compute_average.assert_called_once()
         pd.testing.assert_frame_equal(relationship_validity.details, expected_details_property)
 
@@ -218,7 +218,7 @@ def test_get_score_raises_errors(
         assert pd.isna(score)
         pd.testing.assert_frame_equal(relationship_validity.details, expected_details_property)
         progress_bar.update.assert_called()
-        assert progress_bar.update.call_count == 1
+        progress_bar.update.assert_called_once()
 
     def test_get_details_with_table_name(self):
         """Test the ``get_details`` method.