summaryrefslogtreecommitdiff
path: root/tests/test_bio_file.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_bio_file.py')
-rw-r--r--tests/test_bio_file.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/test_bio_file.py b/tests/test_bio_file.py
index e118386..506bf28 100644
--- a/tests/test_bio_file.py
+++ b/tests/test_bio_file.py
@@ -4,10 +4,14 @@
Copyright (c) 1999-2002 Ng Pheng Siong. All rights reserved."""
-import unittest
+import os, sys
+try:
+ import unittest2 as unittest
+except ImportError:
+ import unittest
+
import M2Crypto
from M2Crypto.BIO import File, openfile
-import os, sys
class FileTestCase(unittest.TestCase):