1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
|
--- a/docs/sample.py
+++ b/docs/sample.py
@@ -1,4 +1,4 @@
-import unittest2
+import unittest
import deprecation
@@ -70,7 +70,7 @@
return 1
-class Tests(unittest2.TestCase):
+class Tests(unittest.TestCase):
@deprecation.fail_if_not_removed
def test_won(self):
--- a/tests/test_deprecation.py
+++ b/tests/test_deprecation.py
@@ -12,14 +12,14 @@
# As we unfortunately support Python 2.7, it lacks TestCase.subTest which
# is in 3.4+ or in unittest2
-import unittest2
+import unittest
import warnings
import deprecation
from datetime import date
-class Test_deprecated(unittest2.TestCase):
+class Test_deprecated(unittest.TestCase):
def test_args_set_on_base_class(self):
args = (1, 2, 3, 4)
@@ -230,7 +230,7 @@
self.assertEqual(sot.method(), ret_val)
-class Test_fail_if_not_removed(unittest2.TestCase):
+class Test_fail_if_not_removed(unittest.TestCase):
@deprecation.deprecated(deprecated_in="1.0", current_version="2.0")
def _deprecated_method(self):
@@ -259,7 +259,7 @@
except AssertionError:
self.fail("A DeprecatedWarning shouldn't cause a failure")
- @unittest2.expectedFailure
+ @unittest.expectedFailure
@deprecation.fail_if_not_removed
def test_literal_UnsupportedWarning(self):
self._unsupported_method()
|