diff --git a/test/test_parser.py b/test/test_parser.py index a23bea7..31a876d 100644 --- a/test/test_parser.py +++ b/test/test_parser.py @@ -142,23 +142,15 @@ def test_started_lambda_stmt(): assert m.children[0].type == 'error_node' -def test_python2_octal(): - module = parse('0660') +def test_python2_octal(each_version): + module = parse('0660', version=each_version) first = module.children[0] - if py_version >= 30: + if each_version.startswith('3'): assert first.type == 'error_node' else: assert first.type == 'number' -def test_python3_octal(): - module = parse('0o660') - if py_version >= 30: - assert module.children[0].type == 'number' - else: - assert module.children[0].type == 'error_node' - - @pytest.mark.parametrize('code', ['foo "', 'foo """\n', 'foo """\nbar']) def test_open_string_literal(code): """ diff --git a/test/test_parser_tree.py b/test/test_parser_tree.py index d39c347..2bf546f 100644 --- a/test/test_parser_tree.py +++ b/test/test_parser_tree.py @@ -26,7 +26,7 @@ class TestsFunctionAndLambdaParsing(object): @pytest.fixture(params=FIXTURES) def node(self, request): - parsed = parse(dedent(request.param[0])) + parsed = parse(dedent(request.param[0]), version='3.5') request.keywords['expected'] = request.param[1] child = parsed.children[0] if child.type == 'simple_stmt': @@ -61,10 +61,11 @@ class TestsFunctionAndLambdaParsing(object): assert node.annotation.value == expected_annotation -def test_end_pos_line(): +def test_end_pos_line(each_version): # jedi issue #150 s = "x()\nx( )\nx( )\nx ( )\n" - module = parse(s) + + module = parse(s, version=each_version) for i, simple_stmt in enumerate(module.children[:-1]): expr_stmt = simple_stmt.children[0] assert expr_stmt.end_pos == (i + 1, i + 3)