From 85ee624293900c78e9b02f1fcb4aa00c5b2140a5 Mon Sep 17 00:00:00 2001 From: Bryan Bishop Date: Sat, 27 Aug 2016 19:14:32 -0500 Subject: import unittest.mock as mock --- tests/integration/tests.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'tests/integration/tests.py') 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" -- cgit v1.2.3