diff --git a/tests/regressiontests/utils/simplelazyobject.py b/tests/regressiontests/utils/simplelazyobject.py index e84752a5cc..c63c948f30 100644 --- a/tests/regressiontests/utils/simplelazyobject.py +++ b/tests/regressiontests/utils/simplelazyobject.py @@ -1,5 +1,5 @@ import copy -import unittest +from django.utils.unittest import TestCase from django.utils.functional import SimpleLazyObject, empty @@ -25,7 +25,7 @@ class _ComplexObject(object): complex_object = lambda: _ComplexObject("joe") -class TestUtilsSimpleLazyObject(unittest.TestCase): +class TestUtilsSimpleLazyObject(TestCase): """ Tests for SimpleLazyObject """ @@ -95,4 +95,4 @@ class TestUtilsSimpleLazyObject(unittest.TestCase): x = SimpleLazyObject(lambda: 3) self.assertTrue(x) x = SimpleLazyObject(lambda: 0) - self.assertFalse(x) \ No newline at end of file + self.assertFalse(x)