Make the package compatible with python 2.70.9

There is actually unittest module now, so we have to work for the
possibility that unittest2 is not available.

Fixes #49.

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