tests: Use unittest2 when available

Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
2018-07-18 12:53:00 +02:00 · 2018-07-18 12:53:00 +02:00 · 0b9652f2d6
parent 6080b45178
commit 0b9652f2d6
1 changed files with 4 additions and 1 deletions
--- a/tests/test_initphase.py
+++ b/tests/test_initphase.py
@ -2,7 +2,10 @@
 # -*- coding: utf-8 -*-


-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 import mock

 import os