remove mock.assert_called_once()

Author: jlwalke2

tests/unit/test_core.py

@@ -17,7 +17,7 @@ def test_request():
 
         resp = request('GET', 'example.com', session=mock_sess)
 
-    mock_sess.request.assert_called_once()
+    assert mock_sess.request.call_count == 1
     assert isinstance(resp, RestObj)
     assert hasattr(resp, '_headers')
 
@@ -41,7 +41,7 @@ def test_list_items():
 
         resp = list_items()
 
-    request.assert_called_once()
+    assert request.call_count == 1
     assert [RestObj()] == resp
 
 
@@ -56,7 +56,7 @@ def test_get_item_by_dict():
         resp = get_item(target)
 
     assert target == resp
-    request.assert_not_called()
+    assert request.call_count == 0
 
 
 def test_get_item_by_name():
@@ -86,8 +86,8 @@ def test_get_item_by_id():
             request.return_value = target
             resp = get_item(12345)
 
-    is_uuid.assert_called_once()
-    request.assert_called_once()
+    assert is_uuid.call_count == 1
+    assert request.call_count == 1
     assert target == resp
 
 
@@ -107,7 +107,7 @@ def test_update_item():
 
         target._headers = {'etag': 'abcd'}
         resp = update_item(target)
-    request.assert_called_once()
+    assert request.call_count == 1
     assert ('put', '/widget/12345') == request.call_args[0]
     assert target == resp
 

tests/unit/test_microanalytic_score.py

@@ -27,7 +27,7 @@ def test_create_python_module():
                             "    return out1, out2"))
         create_module(source=source)
 
-        post.assert_called_once()
+        assert post.call_count == 1
         json = post.call_args[1].get('json', {})
         assert 'text/x-python' == json['type']
         assert 'public' == json['scope']

tests/unit/test_model_management.py

@@ -56,7 +56,7 @@ def test_create_performance_definition():
                 get_project.return_value['predictionVariable'] = 'predicted'
                 _ = mm.create_performance_definition('model', 'TestLibrary', 'TestData')
 
-            post.assert_called_once()
+            assert post.call_count == 1
             url, data = post.call_args
 
             assert PROJECT['id'] == data['json']['projectId']

tests/unit/test_model_repository.py

@@ -46,7 +46,7 @@ def test_create_model():
                                 tool=target['tool'],
                                 is_champion=target['champion'],
                                 properties=dict(custom1=123, custom2='somevalue'))
-            post.assert_called_once()
+            assert post.call_count == 1
             url, data = post.call_args
 
             # dict isn't guaranteed to preserve order
@@ -69,7 +69,7 @@ def test_create_model():
             get_project.return_value = {'id': PROJECT_ID}
             _ = mr.create_model(copy.deepcopy(target), PROJECT_NAME, description='Updated Model')
             target['description'] = 'Updated Model'
-            post.assert_called_once()
+            assert post.call_count == 1
             url, data = post.call_args
 
             # dicts don't preserve order so property order may not match