Fix import Mock

This commit is contained in:
Carlos Martinez 2017-10-26 12:22:12 -05:00
parent 1faa41c63c
commit e7600f4301

View File

@ -1,7 +1,7 @@
import datetime import datetime
import sys
import pytest import pytest
from unittest.mock import Mock
from django.db import models from django.db import models
from django.utils.functional import SimpleLazyObject from django.utils.functional import SimpleLazyObject
from py.test import raises from py.test import raises
@ -22,6 +22,11 @@ from ..rest_framework.mutation import SerializerMutation
pytestmark = pytest.mark.django_db pytestmark = pytest.mark.django_db
if sys.version_info < (3, 0):
from unittest.mock import Mock
else:
from mock import Mock
class MockUserContext(object): class MockUserContext(object):