diff nagslang/tests/test_yamlish.py @ 114:d1a96fc62d26

Fallback to pytest's skip exception
author Stefano Rivera <stefano@rivera.za.net>
date Mon, 02 Sep 2013 15:00:28 +0200
parents c28f2fc2bb05
children b75de48a618c
line wrap: on
line diff
--- a/nagslang/tests/test_yamlish.py	Mon Sep 02 14:51:50 2013 +0200
+++ b/nagslang/tests/test_yamlish.py	Mon Sep 02 15:00:28 2013 +0200
@@ -1,11 +1,17 @@
-from unittest import TestCase, SkipTest
+from unittest import TestCase
+
+from nagslang.yamlish import load_s, dump_s
 
 try:
     import yaml
 except ImportError:
     yaml = None  # pyflakes:ignore
 
-from nagslang.yamlish import load_s, dump_s
+try:
+    from unittest import SkipTest
+except ImportError:
+    from pytest import skip
+    SkipTest = skip.Exception  # pyflakes:ignore
 
 
 class TestParse(TestCase):