diff options
Diffstat (limited to 'tests/dbshell')
| -rw-r--r-- | tests/dbshell/test_postgresql_psycopg2.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/dbshell/test_postgresql_psycopg2.py b/tests/dbshell/test_postgresql_psycopg2.py index f0848ac7b8..26090fb7f1 100644 --- a/tests/dbshell/test_postgresql_psycopg2.py +++ b/tests/dbshell/test_postgresql_psycopg2.py @@ -1,7 +1,8 @@ import os +from unittest import mock from django.db.backends.postgresql.client import DatabaseClient -from django.test import SimpleTestCase, mock +from django.test import SimpleTestCase class PostgreSqlDbshellCommandTestCase(SimpleTestCase): |