Tests: added python 2.7 fallback for mock

anexia · Mar 13, 2017 · b202608 · b202608
1 parent 1bcf1d8
commit b202608
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test.py b/tests/test.py
@@ -4,7 +4,11 @@
 from rest_framework import status
 from rest_framework.test import APITestCase
 from django_rest_multitokenauth.models import MultiToken
-from unittest.mock import patch
+try:
+    from unittest.mock import patch
+except:
+    # Python 2.7 fallback
+    from mock import patch
 
 # try getting reverse from django.urls
 try: