test_owpredictions - Replace mock.called_with with assert_called_once

biolab · Mar 11, 2024 · f340648 · f340648
1 parent 6da11b8
commit f340648
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Orange/widgets/evaluate/tests/test_owpredictions.py b/Orange/widgets/evaluate/tests/test_owpredictions.py
@@ -633,9 +633,8 @@ def test_unregister_prediction_model(self):
         self.send_signal(self.widget.Inputs.predictors, log_reg_iris)
         self.send_signal(self.widget.Inputs.data, self.iris)
         self.widget.selection_store.unregister = Mock()
-        prev_model = self.widget.predictionsview.model()
         self.send_signal(self.widget.Inputs.predictors, log_reg_iris)
-        self.widget.selection_store.unregister.called_with(prev_model)
+        self.widget.selection_store.unregister.assert_called_once()
 
     def test_multi_inputs(self):
         w = self.widget