diff options
Diffstat (limited to 'test/__init__.py')
| -rw-r--r-- | test/__init__.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/__init__.py b/test/__init__.py index 83cdd6d..05ba3b2 100644 --- a/test/__init__.py +++ b/test/__init__.py @@ -2,7 +2,10 @@ from __future__ import absolute_import, print_function, unicode_literals import logging import yamlish -import unittest2 as unittest +try: + import unittest2 as unittest +except ImportError: + import unittest import yaml import tempfile import textwrap |