Use unittest2 for tests.unit.test_types

2014-03-04 15:22:57 -06:00
parent bf07af6d7f
commit 33b69c06e1
1 changed files with 5 additions and 1 deletions
--- a/tests/unit/test_types.py
+++ b/tests/unit/test_types.py
@@ -1,5 +1,9 @@
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest  # noqa
+
 from binascii import unhexlify
-import unittest
 import datetime
 import cassandra
 from cassandra.cqltypes import (BooleanType, lookup_casstype_simple, lookup_casstype,