Fix for the Python3 tests, restoring modules as it expects them.

aigamedev · May 7, 2015 · 97eb138 · 97eb138
1 parent 7085c25
commit 97eb138
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -44,6 +44,7 @@ htmlcov/
 .tox/
 .coverage
 .coveragerc
+build
 .coverage.*
 .cache
 .settings

diff --git a/sknn/tests/test_backend.py b/sknn/tests/test_backend.py
@@ -14,9 +14,11 @@ class TestBackendPseudoModule(unittest.TestCase):
     def setUp(self):
         if 'THEANO_FLAGS' in os.environ:
             del os.environ['THEANO_FLAGS']
-
+
+        import theano
+
         self.removed = {}
-        for name in sys.modules.keys():
+        for name in list(sys.modules.keys()):
             if name.startswith('theano'):
                 self.removed[name] = sys.modules[name]
                 del sys.modules[name]