Skip tests in Python 2 when using Python 3 isms

cisagov · Oct 21, 2017 · 905b91f · 905b91f
1 parent 760dc18
commit 905b91f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -11,12 +11,21 @@ def test_without_filename(self):
         with smart_open() as fh:
             self.assertIs(fh, sys.stdout)
 
+    @unittest.skipIf(sys.version_info[0] < 3, 'Python 3 version of test')
     def test_with_empty_filename(self):
         """Should raise a `FileNotFoundError`"""
         with self.assertRaises(FileNotFoundError):
             with smart_open(''):
                 pass
 
+    @unittest.skipIf(sys.version_info[0] >= 3, 'Python 2 version of test')
+    def test_with_empty_filename_python2(self):
+        """Should raise a `FileNotFoundError`"""
+        with self.assertRaises(IOError):
+            with smart_open(''):
+                pass
+
+    @unittest.skipIf(sys.version_info[0] < 3, 'Python 3 version of test')
     def test_with_real_filename(self):
         test_data = 'This is the test data'