called_once_with => assert_called_once_with

Author: dwhswenson

paths_cli/tests/compiling/test_root_compiler.py

@@ -247,8 +247,8 @@ def test_register_plugins_unit(foo_compiler_plugin, foo_baz_builder_plugin):
     with patch(BASE + "_register_builder_plugin", Mock()) as builder, \
             patch(BASE + "_register_compiler_plugin", Mock()) as compiler:
         register_plugins([foo_baz_builder_plugin, foo_compiler_plugin])
-        assert builder.called_once_with(foo_baz_builder_plugin)
-        assert compiler.called_once_with(foo_compiler_plugin)
+        builder.assert_called_once_with(foo_baz_builder_plugin)
+        compiler.assert_called_once_with(foo_compiler_plugin)
 
 
 def test_register_plugins_integration(foo_compiler_plugin,

paths_cli/tests/wizard/test_plugin_classes.py

@@ -38,7 +38,7 @@ def test_call(self):
         with mock.patch(patch_loc, mock_load):
             result = loader(wiz)
 
-        assert mock_load.called_once_with(wiz, "foos", "foo")
+        mock_load.assert_called_once_with(wiz, "foos", "foo")
         assert result == "some_object"