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/test/test_evaluate/test_gradual/test_typeshed.py
+++ b/test/test_evaluate/test_gradual/test_typeshed.py
@@ -190,7 +190,7 @@ def test_goto_stubs(Script):
        'import os; os.walk'
        'from collections import Counter; Counter'
    ])
-def test_goto_stubs_on_itself(Script):
+def test_goto_stubs_on_itself(Script, code):
    """
    If goto_stubs is used on an identifier in e.g. the stdlib, we should goto
    the stub of it.