use unittest.mock instead of mock

docs/requirements.txt

@@ -4,6 +4,5 @@
 Sphinx==3.5.4
 furo==2021.4.11b34
 myst-parser==0.13.7
-mock==4.0.3
 moto==2.0.5
 sphinx-copybutton==0.3.1

papermill/tests/test_autosave.py

@@ -3,8 +3,7 @@
 import tempfile
 import time
 import unittest
-
-from mock import patch
+from unittest.mock import patch
 
 from . import get_notebook_path
 

papermill/tests/test_cli.py

@@ -13,12 +13,11 @@
 import nbformat
 from jupyter_client import kernelspec
 import unittest
+from unittest.mock import patch
 
 import pytest
 from click.testing import CliRunner
 
-from mock import patch
-
 from . import get_notebook_path, kernel_name
 from .. import cli
 from ..cli import papermill, _is_int, _is_float, _resolve_type

papermill/tests/test_clientwrap.py

@@ -1,7 +1,7 @@
 import nbformat
 import unittest
 
-from mock import call, patch
+from unittest.mock import call, patch
 
 from . import get_notebook_path
 

papermill/tests/test_engines.py

@@ -3,7 +3,7 @@
 import unittest
 
 from abc import ABCMeta
-from mock import Mock, patch, call
+from unittest.mock import Mock, patch, call
 from nbformat.notebooknode import NotebookNode
 
 from . import get_notebook_path

papermill/tests/test_execute.py

@@ -3,17 +3,13 @@
 import shutil
 import tempfile
 import unittest
+from unittest.mock import patch
 
 from functools import partial
 from pathlib import Path
 
 from nbformat import validate
 
-try:
-    from unittest.mock import patch
-except ImportError:
-    from mock import patch
-
 from .. import engines
 from ..log import logger
 from ..iorw import load_notebook_node

papermill/tests/test_gcs.py

@@ -1,5 +1,4 @@
 import unittest
-
 from unittest.mock import patch
 
 from ..exceptions import PapermillRateLimitException

requirements/dev.txt

@@ -6,7 +6,6 @@ google_compute_engine # Need this because boto has issues with dynamic package l
 ipython>=5.0
 ipywidgets
 notebook
-mock
 moto
 pytest>=4.1
 pytest-cov>=2.6.1