tests: python2 compatitibility. Need mock library to run tests

ACRIOS-Systems · Mar 10, 2015 · c66cc1e · c66cc1e
1 parent db502ec
commit c66cc1e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tests/__init__.pyc b/tests/__init__.pyc
diff --git a/tests/test_gpio.py b/tests/test_gpio.py
@@ -1,6 +1,9 @@
 # -*- coding: utf-8 -*-
 from unittest import TestCase
-from unittest.mock import mock_open, patch
+try:
+    from unittest.mock import mock_open, patch
+except ImportError:
+    from mock import mock_open, patch
 import sys
 import os
 pjoin = os.path.join