shithub: pokecrystal

Info  •  Files  •  Log  •  Branches

Adding an exception handler around the unittest import.

--- a/extras/crystal.py

+++ b/extras/crystal.py

@@ -11,7 +11,10 @@

 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