diff options
Diffstat (limited to 'test/test_reader.py')
-rw-r--r-- | test/test_reader.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/test_reader.py b/test/test_reader.py index 0676b30..a34104c 100644 --- a/test/test_reader.py +++ b/test/test_reader.py @@ -3,7 +3,10 @@ from __future__ import absolute_import, print_function, unicode_literals import yaml import yamlish import test -import unittest2 as unittest +try: + import unittest2 as unittest +except ImportError: + import unittest test_data_list = [ { |