Restore Python2 support.rbarrois/restore-py2

1 files changed, 7 insertions, 0 deletions

diff --git a/tests/test_match.py b/tests/test_match.py
index 4bf7162..f807828 100755
--- a/tests/test_match.py
+++ b/tests/test_match.py
@@ -4,11 +4,18 @@
 # This code is distributed under the two-clause BSD License.
 
 import unittest
+import sys
 
 import semantic_version
 
 
 class MatchTestCase(unittest.TestCase):
+    if sys.version_info[0] <= 2:
+        import contextlib
+        @contextlib.contextmanager
+        def subTest(self, **kwargs):
+            yield
+
     invalid_specs = [
         '',
         '!0.1',