diff options
Diffstat (limited to 'tests/integration/tests.py')
| -rw-r--r-- | tests/integration/tests.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/integration/tests.py b/tests/integration/tests.py index 79fadda..024feab 100644 --- a/tests/integration/tests.py +++ b/tests/integration/tests.py @@ -99,7 +99,11 @@ from pokemontools.crystal import ( import pokemontools.wram import unittest -import mock + +try: + import unittest.mock as mock +except ImportError: + import mock class BasicTestCase(unittest.TestCase): "this is where i cram all of my unit tests together" |