diff --git a/test/completion/pep0484.py b/test/completion/pep0484.py index fc08460f..b944cd1d 100644 --- a/test/completion/pep0484.py +++ b/test/completion/pep0484.py @@ -86,8 +86,6 @@ def annotation_forward_reference(b: "B") -> "B": #? B() b -#? B() -annotation_forward_reference(1) #? ["test_element"] annotation_forward_reference(1).t @@ -95,6 +93,9 @@ class B: test_element = 1 pass +#? B() +annotation_forward_reference(1) + class SelfReference: test_element = 1 diff --git a/test/run.py b/test/run.py index 6f487403..59cb18f7 100755 --- a/test/run.py +++ b/test/run.py @@ -184,7 +184,8 @@ class IntegrationTestCase(object): def definition(correct, correct_start, path): should_be = set() for match in re.finditer('(?:[^ ]+)', correct): - parser = Parser(load_grammar(), match.string, start_symbol='eval_input') + string = match.group(0) + parser = Parser(load_grammar(), string, start_symbol='eval_input') parser.position_modifier.line = self.line_nr element = parser.get_parsed_node() element.parent = script._parser.user_scope()