diff options
-rw-r--r-- | pylibchorus/tests/chorus_client_tests.py | 6 | ||||
-rw-r--r-- | setup.py | 3 |
2 files changed, 8 insertions, 1 deletions
diff --git a/pylibchorus/tests/chorus_client_tests.py b/pylibchorus/tests/chorus_client_tests.py index c36edcd..5103b70 100644 --- a/pylibchorus/tests/chorus_client_tests.py +++ b/pylibchorus/tests/chorus_client_tests.py @@ -8,7 +8,11 @@ from pylibchorus.chorus_api import _check_login_ from pylibchorus.chorus_api import _create_workfile_ from pylibchorus.chorus_api import _update_workfile_version_ from pylibchorus.chorus_api import _delete_workfile_ -import unittest +import sys +if sys.version_info[0] == 2 and sys.version_info[1] == 6: + import unittest2 as unittest +else: + import unittest LOG = logging.getLogger(__name__) @@ -39,6 +39,9 @@ setup( 'requests', 'setuptools', ], + tests_require=[ + 'unittest2', + ], classifiers=[ 'Development Status :: 2 - Pre-Alpha', 'Environment :: Console', |