From 7aba69145dcb436539a7798086748b73a39121e5 Mon Sep 17 00:00:00 2001 From: Tim Graham Date: Thu, 19 Jan 2017 12:16:04 -0500 Subject: Refs #23919 -- Removed django.test.mock Python 2 compatibility shim. --- tests/check_framework/test_database.py | 3 ++- tests/check_framework/test_multi_db.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) (limited to 'tests/check_framework') 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): diff --git a/tests/check_framework/test_multi_db.py b/tests/check_framework/test_multi_db.py index 700553f47b..3080b52fda 100644 --- a/tests/check_framework/test_multi_db.py +++ b/tests/check_framework/test_multi_db.py @@ -1,5 +1,7 @@ +from unittest import mock + from django.db import connections, models -from django.test import TestCase, mock +from django.test import TestCase from django.test.utils import isolate_apps, override_settings -- cgit v1.3