tests: replace mock with stdlib unittest.mock

4 files changed, 6 insertions, 6 deletions

diff --git a/yoyo/tests/test_backends.py b/yoyo/tests/test_backends.py
index f89861a..abf8d47 100644
--- a/yoyo/tests/test_backends.py
+++ b/yoyo/tests/test_backends.py
@@ -2,9 +2,9 @@ from tempfile import NamedTemporaryFile
 from threading import Thread
 import time
 
-from mock import Mock
-from mock import call
-from mock import patch
+from unittest.mock import Mock
+from unittest.mock import call
+from unittest.mock import patch
 import pytest
 
 from yoyo import backends
diff --git a/yoyo/tests/test_cli_script.py b/yoyo/tests/test_cli_script.py
index 09152aa..b385e1c 100644
--- a/yoyo/tests/test_cli_script.py
+++ b/yoyo/tests/test_cli_script.py
@@ -24,7 +24,7 @@ import re
 import sys
 import textwrap
 
-from mock import Mock, patch, call
+from unittest.mock import Mock, patch, call
 import freezegun
 import pytest
 import tms
diff --git a/yoyo/tests/test_connections.py b/yoyo/tests/test_connections.py
index ca1d27b..3877181 100644
--- a/yoyo/tests/test_connections.py
+++ b/yoyo/tests/test_connections.py
@@ -14,7 +14,7 @@
 
 import sys
 
-from mock import patch, call, MagicMock
+from unittest.mock import patch, call, MagicMock
 import pytest
 
 from yoyo.connections import parse_uri, BadConnectionURI
diff --git a/yoyo/tests/test_migrations.py b/yoyo/tests/test_migrations.py
index 09f94ba..0eea603 100644
--- a/yoyo/tests/test_migrations.py
+++ b/yoyo/tests/test_migrations.py
@@ -14,7 +14,7 @@
 
 from datetime import datetime
 from datetime import timedelta
-from mock import Mock, patch
+from unittest.mock import Mock, patch
 import io
 import os