Replace 'called_with' with 'assert_called_with'.

jaraco · Feb 19, 2023 · 301af23 · 301af23
1 parent 58f9899
commit 301af23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_functools.py b/test_functools.py
@@ -214,7 +214,7 @@ def test_cleanup_called_on_exception(self):
         self.set_to_fail(times=calls)
         retry_call(self.attempt, retries=calls, cleanup=cleanup, trap=Exception)
         assert cleanup.call_count == calls
-        assert cleanup.called_with()
+        cleanup.assert_called_with()
 
     def test_infinite_retries(self):
         self.set_to_fail(times=999)