Restore Python2 support.rbarrois/restore-py2

1 files changed, 13 insertions, 0 deletions

diff --git a/tests/test_parsing.py b/tests/test_parsing.py
index 0da679f..01b7b5e 100755
--- a/tests/test_parsing.py
+++ b/tests/test_parsing.py
@@ -5,11 +5,18 @@
 
 import itertools
 import unittest
+import sys
 
 import semantic_version
 
 
 class ParsingTestCase(unittest.TestCase):
+    if sys.version_info[0] <= 2:
+        import contextlib
+        @contextlib.contextmanager
+        def subTest(self, **kwargs):
+            yield
+
     invalids = [
         None,
         '',
@@ -66,6 +73,12 @@ class ParsingTestCase(unittest.TestCase):
 
 
 class ComparisonTestCase(unittest.TestCase):
+    if sys.version_info[0] <= 2:
+        import contextlib
+        @contextlib.contextmanager
+        def subTest(self, **kwargs):
+            yield
+
     order = [
         '1.0.0-alpha',
         '1.0.0-alpha.1',