Fix some tests because of stub_context changes

2019-03-16 01:06:13 +01:00
parent fce37fa0e3
commit e2fea0a5de
4 changed files with 16 additions and 19 deletions
--- a/jedi/evaluate/context/function.py
+++ b/jedi/evaluate/context/function.py
@@ -336,7 +336,10 @@ class FunctionExecutionContext(TreeContext):
                # Only the first generic is relevant.
                generics = (return_contexts.py__class__(), NO_CONTEXTS, NO_CONTEXTS)
                return ContextSet(
-                    AnnotatedSubClass(getattr(c, 'stub_context', c), generics) for c in async_classes
+                    AnnotatedSubClass(
+                        c if c.stub_context is None else c.stub_context,
+                        generics
+                    ) for c in async_classes
                ).execute_annotation()
        else:
            if is_generator: