diff options
Diffstat (limited to 'tests/check_framework/test_database.py')
| -rw-r--r-- | tests/check_framework/test_database.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/check_framework/test_database.py b/tests/check_framework/test_database.py index 594f4ccee6..80824ab8e0 100644 --- a/tests/check_framework/test_database.py +++ b/tests/check_framework/test_database.py @@ -1,9 +1,10 @@ import unittest +from unittest import mock from django.core.checks import Tags, run_checks from django.core.checks.registry import CheckRegistry from django.db import connection -from django.test import TestCase, mock +from django.test import TestCase class DatabaseCheckTests(TestCase): |