python-freezegun/remove_dependency_on_mock.patch

28 lines
517 B
Diff
Raw Normal View History

--- a/tests/test_ticking.py
+++ b/tests/test_ticking.py
@@ -1,7 +1,10 @@
import datetime
import time
-import mock
import pytest
+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,6 +1,9 @@
from unittest import SkipTest
-import mock
+try:
+ import mock
+except ImportError:
+ import unittest.mock as mock
from freezegun import api
from tests import utils