Fix some stuff to make parso work again.

This commit is contained in:
Dave Halter
2017-06-02 00:00:31 +02:00
parent af26cc9f05
commit e0485b032e
6 changed files with 11 additions and 9 deletions

View File

@@ -12,6 +12,6 @@ import pytest
])
def test_equals(source):
script = Script(source)
node = script._get_module_node().children[0].children[0]
node = script._get_module_node().children[0]
first, = script._get_module().eval_node(node)
assert isinstance(first, CompiledObject) and first.obj is True

View File

@@ -11,7 +11,7 @@ from jedi import Script
def test_paths_from_assignment():
def paths(src):
script = Script(src)
expr_stmt = script._get_module_node().children[0].children[0]
expr_stmt = script._get_module_node().children[0]
return set(sys_path._paths_from_assignment(script._get_module(), expr_stmt))
assert paths('sys.path[0:0] = ["a"]') == set(['a'])

View File

@@ -11,8 +11,10 @@ import pytest
class TestCallAndName():
def get_call(self, source):
# Get the simple_stmt and then the first one.
simple_stmt = parse(source).children[0]
return simple_stmt.children[0]
node = parse(source).children[0]
if node.type == 'simple_stmt':
return node.children[0]
return node
def test_name_and_call_positions(self):
name = self.get_call('name\nsomething_else')