summaryrefslogtreecommitdiff
path: root/src/isodate/tests/test_pickle.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/isodate/tests/test_pickle.py')
-rw-r--r--src/isodate/tests/test_pickle.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/src/isodate/tests/test_pickle.py b/src/isodate/tests/test_pickle.py
index 4b824e7..0d8c184 100644
--- a/src/isodate/tests/test_pickle.py
+++ b/src/isodate/tests/test_pickle.py
@@ -1,5 +1,8 @@
+try:
+ import cPickle as pickle
+except ImportError:
+ import pickle
import unittest
-import cPickle as pickle
import isodate
@@ -31,7 +34,7 @@ class TestPickle(unittest.TestCase):
pikl = pickle.dumps(dur, proto)
if dur != pickle.loads(pikl):
raise Exception("not equal")
- except Exception, e:
+ except Exception as e:
failed.append("pickle proto %d failed (%s)" % (proto, repr(e)))
self.assertEqual(len(failed), 0, "pickle protos failed: %s" %
str(failed))