diff options
Diffstat (limited to 'tests/test_declarations.py')
| -rw-r--r-- | tests/test_declarations.py | 5 |
1 files changed, 4 insertions, 1 deletions
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 |