diff --git a/tests/utils_tests/test_lazyobject.py b/tests/utils_tests/test_lazyobject.py index 134ae77750..a95335413e 100644 --- a/tests/utils_tests/test_lazyobject.py +++ b/tests/utils_tests/test_lazyobject.py @@ -1,9 +1,10 @@ import copy import pickle import sys +import unittest import warnings -from unittest import TestCase +from django.test import TestCase from django.utils.functional import LazyObject, SimpleLazyObject, empty from .models import Category, CategoryInfo @@ -20,7 +21,7 @@ class Foo: return self.foo == other.foo -class LazyObjectTestCase(TestCase): +class LazyObjectTestCase(unittest.TestCase): def lazy_wrap(self, wrapped_object): """ Wrap the given object into a LazyObject