From 5b8b3599be702a05d381254bd6dbe6463913c0b0 Mon Sep 17 00:00:00 2001 From: ZyX Date: Sat, 6 Apr 2013 19:01:59 +0400 Subject: [PATCH] Fix tests on python-2.6 --- tests/__init__.py | 2 ++ tests/test_lib.py | 3 +-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/__init__.py b/tests/__init__.py index 13b41550..2492b045 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -2,5 +2,7 @@ import sys if sys.version_info < (2, 7): from unittest2 import TestCase, main # NOQA + from unittest2.case import SkipTest # NOQA else: from unittest import TestCase, main # NOQA + from unittest.case import SkipTest # NOQA diff --git a/tests/test_lib.py b/tests/test_lib.py index f903b323..654f4905 100644 --- a/tests/test_lib.py +++ b/tests/test_lib.py @@ -5,9 +5,8 @@ from powerline.lib.vcs import guess from subprocess import call, PIPE import os import sys -from unittest.case import SkipTest from functools import partial -from tests import TestCase +from tests import TestCase, SkipTest class TestLib(TestCase):