tests: fix missing SkipTest imports

astanin · astanin · commit 5105121ff39f · 2015-08-20T12:31:31.000+02:00
diff --git a/test/common.py b/test/common.py
@@ -1,3 +1,18 @@
+try:
+    from nose.plugins.skip import SkipTest
+except ImportError:
+    try:
+        from unittest.case import SkipTest  # Python >= 2.7
+    except ImportError:
+        try:
+            from unittest2.case import SkipTest  # Python < 2.7
+        except ImportError:
+            class SkipTest(Exception):
+                """Raise this exception to mark a test as skipped.
+                """
+                pass
+
+
 try:
     from nose.tools import assert_equal, assert_in, assert_raises
 
diff --git a/test/test_api.py b/test/test_api.py
@@ -6,7 +6,7 @@
 from __future__ import unicode_literals
 from tabulate import tabulate, tabulate_formats, simple_separated_format
 from platform import python_version_tuple
-from nose.plugins.skip import SkipTest
+from common import SkipTest
 
 
 try:
diff --git a/test/test_input.py b/test/test_input.py
@@ -5,8 +5,7 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 from tabulate import tabulate
-from common import assert_equal, assert_in, assert_raises
-from nose.plugins.skip import SkipTest
+from common import assert_equal, assert_in, assert_raises, SkipTest
 
 
 def test_iterable_of_iterables():
diff --git a/test/test_output.py b/test/test_output.py
@@ -5,7 +5,7 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 from tabulate import tabulate, simple_separated_format
-from common import assert_equal, assert_raises
+from common import assert_equal, assert_raises, SkipTest
 
 
 # _test_table shows
diff --git a/test/test_regression.py b/test/test_regression.py
@@ -5,7 +5,7 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 from tabulate import tabulate, _text_type, _long_type
-from common import assert_equal, assert_in
+from common import assert_equal, assert_in, SkipTest
 
 
 def test_ansi_color_in_table_cells():
