5 files changed, 21 insertions, 5 deletions
diff --git a/tests/test_base.py b/tests/test_base.py
index a548b56..0788f82 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -20,7 +20,11 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
+
 import warnings
 
 from factory import base
diff --git a/tests/test_containers.py b/tests/test_containers.py
index 6e58573..a2a369d 100644
--- a/tests/test_containers.py
+++ b/tests/test_containers.py
@@ -20,7 +20,10 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from factory import base
 from factory import containers
diff --git a/tests/test_declarations.py b/tests/test_declarations.py
index 7215a54..3960670 100644
--- a/tests/test_declarations.py
+++ b/tests/test_declarations.py
@@ -20,7 +20,10 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from factory.declarations import deepgetattr, OrderedDeclaration, Sequence
 
diff --git a/tests/test_using.py b/tests/test_using.py
index e6df3c2..fcd3aab 100644
--- a/tests/test_using.py
+++ b/tests/test_using.py
@@ -20,7 +20,10 @@
 # THE SOFTWARE.
 """Tests using factory."""
 
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 import factory
 
diff --git a/tests/test_utils.py b/tests/test_utils.py
index 543a6c0..c3047d3 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -21,7 +21,10 @@
 # THE SOFTWARE.
 
 
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 from factory import utils