diff options
Diffstat (limited to 'tests/schema/tests.py')
| -rw-r--r-- | tests/schema/tests.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/schema/tests.py b/tests/schema/tests.py index 73c1371cfe..4f62333dba 100644 --- a/tests/schema/tests.py +++ b/tests/schema/tests.py @@ -2,6 +2,7 @@ import datetime import itertools import unittest from copy import copy +from unittest import mock from django.db import ( DatabaseError, IntegrityError, OperationalError, connection, @@ -19,7 +20,7 @@ from django.db.models.fields.related import ( from django.db.models.indexes import Index from django.db.transaction import TransactionManagementError, atomic from django.test import ( - TransactionTestCase, mock, skipIfDBFeature, skipUnlessDBFeature, + TransactionTestCase, skipIfDBFeature, skipUnlessDBFeature, ) from django.test.utils import CaptureQueriesContext from django.utils import timezone |