diff options
Diffstat (limited to 'test/test_output.py')
| -rw-r--r-- | test/test_output.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/test_output.py b/test/test_output.py index 888f812..1105b84 100644 --- a/test/test_output.py +++ b/test/test_output.py @@ -5,7 +5,10 @@ Test general output functionality. Without much stress on the format itself. """ from __future__ import absolute_import, print_function, unicode_literals -import unittest2 as unittest +try: + import unittest2 as unittest +except ImportError: + import unittest import yamlish import yaml import logging |