Fix quite a few more tests. Only about a fifth failing now

jedi/plugins/flask.py

@@ -3,7 +3,7 @@ def import_module(callback):
     Handle "magic" Flask extension imports:
     ``flask.ext.foo`` is really ``flask_foo`` or ``flaskext.foo``.
     """
-    def wrapper(inference_state, import_names, module_value, *args, **kwargs):
+    def wrapper(inference_state, import_names, module_context, *args, **kwargs):
         if len(import_names) == 3 and import_names[:2] == ('flask', 'ext'):
             # New style.
             ipath = (u'flask_' + import_names[2]),
@@ -17,5 +17,5 @@ def import_module(callback):
                 next(iter(value_set)),
                 *args, **kwargs
             )
-        return callback(inference_state, import_names, module_value, *args, **kwargs)
+        return callback(inference_state, import_names, module_context, *args, **kwargs)
     return wrapper