diff --git a/test/test_inference/test_gradual/test_stubs.py b/test/test_inference/test_gradual/test_stubs.py index 0782469c..c811b01f 100644 --- a/test/test_inference/test_gradual/test_stubs.py +++ b/test/test_inference/test_gradual/test_stubs.py @@ -16,7 +16,7 @@ from test.helpers import root_dir ] ) @pytest.mark.parametrize( - ('code', 'full_name', 'has_stub', 'has_python', 'goto_changes'), [ + ('code', 'full_name', 'has_stub', 'has_python', 'options'), [ ['import os; os.walk', 'os.walk', True, True, {}], ['from collections import Counter', 'collections.Counter', True, True, {}], ['from collections', 'collections', True, True, {}], @@ -24,7 +24,8 @@ from test.helpers import root_dir ['from collections import Counter; Counter()', 'collections.Counter', True, True, {}], ['from collections import Counter; Counter.most_common', 'collections.Counter.most_common', True, True, {}], - ['from collections import deque', 'collections.deque', True, False, {'has_python': True}], + ['from collections import deque', 'collections.deque', True, False, + {'goto_has_python': True}], ['from keyword import kwlist; kwlist', 'typing.Sequence', True, True, {'full_name': 'keyword.kwlist'}], @@ -41,7 +42,7 @@ from test.helpers import root_dir ['import stub_only', 'stub_only', True, False, {}], ]) def test_infer_and_goto(Script, code, full_name, has_stub, has_python, way, - kwargs, type_, goto_changes, environment): + kwargs, type_, options, environment): if environment.version_info < (3, 5): # We just don't care about much of the detailed Python 2 failures # anymore, because its end-of-life soon. (same for 3.4) @@ -58,8 +59,8 @@ def test_infer_and_goto(Script, code, full_name, has_stub, has_python, way, only_stubs = kwargs['only_stubs'] if type_ == 'goto': - full_name = goto_changes.get('full_name', full_name) - has_python = goto_changes.get('has_python', has_python) + full_name = options.get('goto_full_name', full_name) + has_python = options.get('goto_has_python', has_python) if way == 'direct': if type_ == 'goto':