python-freezegun/remove_dependency_on_mock.patch

25 lines
474 B
Diff
Raw Normal View History

--- a/tests/test_ticking.py
+++ b/tests/test_ticking.py
@@ -1,6 +1,9 @@
import datetime
import time
-import mock
+try:
+ import mock
+except ImportError:
+ import unittest.mock as mock
from freezegun import freeze_time
from tests import utils
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -1,4 +1,7 @@
-import mock
+try:
+ import mock
+except ImportError:
+ import unittest.mock as mock
from nose.plugins import skip
from freezegun import api