Adding an exception handler around the unittest import.

original-commit-id: 4de0c12b438cf99e90070e723508ce636278e063

1 files changed, 4 insertions, 1 deletions

diff --git a/crystal.py b/crystal.py
index ef03ec1..d303c9d 100644
--- a/crystal.py
+++ b/crystal.py
@@ -11,7 +11,10 @@ from bisect import bisect_left, bisect_right
 from itertools import izip
 
 #for testing all this crap
-import unittest2 as unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 if not hasattr(json, "dumps"):
     json.dumps = json.write