Merge branch '21721-refresh-python-deps'

[arvados.git] / sdk / python / tests / test_sdk.py

diff --git a/sdk/python/tests/test_sdk.py b/sdk/python/tests/test_sdk.py
index 8c56180059d4b79991120d4bf3842c03c3869d6b..4ef81c53d8448222a7ca5b1c387ebb388d78faa3 100644 (file)
--- a/sdk/python/tests/test_sdk.py
+++ b/sdk/python/tests/test_sdk.py
@@ -1,7 +1,12 @@
-import mock
+# Copyright (C) The Arvados Authors. All rights reserved.
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import os
 import unittest
 
+from unittest import mock
+
 import arvados
 import arvados.collection
 
@@ -39,5 +44,5 @@ class TestSDK(unittest.TestCase):
 
         # Because one_task_per_input_file normalizes this collection,
         # it should now create only one job task and not three.
-        arvados.job_setup.one_task_per_input_file(and_end_task=False, arvapi=mock_api)
+        arvados.job_setup.one_task_per_input_file(and_end_task=False, api_client=mock_api)
         mock_api.job_tasks().create().execute.assert_called_once_with()