assert_called_once

sdv-dev · Nov 2, 2023 · 65b3a4f · 65b3a4f
1 parent 39007d1
commit 65b3a4f
Showing 1 changed file with 2 additions and 2 deletions.
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.