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