diff options
-rwxr-xr-x | tests/test_base.py | 6 | ||||
-rwxr-xr-x | tests/test_match.py | 1 | ||||
-rw-r--r-- | tests/test_npm.py | 1 | ||||
-rwxr-xr-x | tests/test_parsing.py | 2 |
4 files changed, 10 insertions, 0 deletions
diff --git a/tests/test_base.py b/tests/test_base.py index d0a0e81..4a844c3 100755 --- a/tests/test_base.py +++ b/tests/test_base.py @@ -16,6 +16,7 @@ class TopLevelTestCase(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield @@ -98,6 +99,7 @@ class TopLevelTestCase(unittest.TestCase): class VersionTestCase(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield @@ -383,6 +385,7 @@ class VersionTestCase(unittest.TestCase): class SpecItemTestCase(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield @@ -562,6 +565,7 @@ class SpecItemTestCase(unittest.TestCase): class CoerceTestCase(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield @@ -592,12 +596,14 @@ class CoerceTestCase(unittest.TestCase): class SpecTestCase(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield def assertCountEqual(self, a, b): import collections + self.assertEqual( collections.Counter(a), collections.Counter(b), diff --git a/tests/test_match.py b/tests/test_match.py index f807828..6851eb2 100755 --- a/tests/test_match.py +++ b/tests/test_match.py @@ -12,6 +12,7 @@ import semantic_version class MatchTestCase(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield diff --git a/tests/test_npm.py b/tests/test_npm.py index da3f5ba..2102cd8 100644 --- a/tests/test_npm.py +++ b/tests/test_npm.py @@ -14,6 +14,7 @@ from semantic_version import base class NpmSpecTests(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield diff --git a/tests/test_parsing.py b/tests/test_parsing.py index 01b7b5e..45d22c6 100755 --- a/tests/test_parsing.py +++ b/tests/test_parsing.py @@ -13,6 +13,7 @@ import semantic_version class ParsingTestCase(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield @@ -75,6 +76,7 @@ class ParsingTestCase(unittest.TestCase): class ComparisonTestCase(unittest.TestCase): if sys.version_info[0] <= 2: import contextlib + @contextlib.contextmanager def subTest(self, **kwargs): yield |