Fix test cases for Python 2

thombashi · Jun 16, 2019 · 8bf019a · 8bf019a
1 parent d36f0f7
commit 8bf019a
Showing 1 changed file with 17 additions and 14 deletions.
diff --git a/test/test_argparse.py b/test/test_argparse.py
@@ -2,6 +2,7 @@
 
 from __future__ import unicode_literals
 
+import sys
 from argparse import ArgumentError, ArgumentParser, ArgumentTypeError
 
 import pytest
@@ -23,15 +24,16 @@ def test_exception(self, value):
         parser = ArgumentParser()
         parser.add_argument("filename", type=filename)
 
-        try:
-            parser.parse_args([value])
-        except SystemExit as e:
-            assert isinstance(e.__context__, ArgumentError)
-        else:
-            raise RuntimeError()
+        if sys.version_info[0] >= 3:
+            try:
+                parser.parse_args([value])
+            except SystemExit as e:
+                assert isinstance(e.__context__, ArgumentError)
+            else:
+                raise RuntimeError()
 
-        with pytest.raises(ArgumentTypeError):
-            filename(value)
+            with pytest.raises(ArgumentTypeError):
+                filename(value)
 
 
 class Test_argparse_filepath_validator(object):
@@ -48,12 +50,13 @@ def test_exception(self, value):
         parser = ArgumentParser()
         parser.add_argument("filepath", type=filepath)
 
-        try:
-            parser.parse_args([value])
-        except SystemExit as e:
-            assert isinstance(e.__context__, ArgumentError)
-        else:
-            raise RuntimeError()
+        if sys.version_info[0] >= 3:
+            try:
+                parser.parse_args([value])
+            except SystemExit as e:
+                assert isinstance(e.__context__, ArgumentError)
+            else:
+                raise RuntimeError()
 
         with pytest.raises(ArgumentTypeError):
             filepath(value)