fix Python 2.6 build

thomasballinger · thomasballinger · commit 19776ac25a89 · 2015-12-03T11:42:54.000-05:00
diff --git a/bpython/test/test_curtsies_repl.py b/bpython/test/test_curtsies_repl.py
@@ -6,11 +6,11 @@
 import sys
 import tempfile
 import io
-import importlib
 from functools import partial
 from contextlib import contextmanager
 from six.moves import StringIO
 
+
 from bpython.curtsiesfrontend import repl as curtsiesrepl
 from bpython.curtsiesfrontend import interpreter
 from bpython.curtsiesfrontend import events as bpythonevents
@@ -21,6 +21,11 @@
 from bpython.test import (FixLanguageTestCase as TestCase, MagicIterMock, mock,
                           builtin_target, unittest)
 
+if py3:
+    from importlib import invalidate_caches
+else:
+    def invalidate_caches():
+        """Does not exist before Python 3.3"""
 
 def setup_config(conf):
     config_struct = config.Struct()
@@ -292,8 +297,7 @@ def setUp(self):
         self.open = partial(io.open, mode='wt', encoding='utf-8')
         self.dont_write_bytecode = sys.dont_write_bytecode
         sys.dont_write_bytecode = True
-        if py3:
-            importlib.invalidate_caches()
+        invalidate_caches()
 
     def tearDown(self):
         sys.dont_write_bytecode = self.dont_write_bytecode
