mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 22:14:27 +08:00
122 lines
3.6 KiB
Python
122 lines
3.6 KiB
Python
from jedi._compatibility import u
|
|
from jedi.parser import Parser
|
|
from jedi.parser.user_context import UserContextParser
|
|
from jedi.parser import representation as pr
|
|
from textwrap import dedent
|
|
|
|
|
|
def test_user_statement_on_import():
|
|
"""github #285"""
|
|
s = u("from datetime import (\n"
|
|
" time)")
|
|
|
|
for pos in [(2, 1), (2, 4)]:
|
|
p = UserContextParser(s, None, pos, None).user_stmt()
|
|
assert isinstance(p, pr.Import)
|
|
assert p.defunct is False
|
|
assert [str(n) for n in p.get_defined_names()] == ['time']
|
|
|
|
|
|
class TestCallAndName():
|
|
def get_call(self, source):
|
|
stmt = Parser(u(source), no_docstr=True).module.statements[0]
|
|
return stmt.expression_list()[0]
|
|
|
|
def test_name_and_call_positions(self):
|
|
call = self.get_call('name\nsomething_else')
|
|
assert str(call.name) == 'name'
|
|
assert call.name.start_pos == call.start_pos == (1, 0)
|
|
assert call.name.end_pos == call.end_pos == (1, 4)
|
|
|
|
call = self.get_call('1.0\n')
|
|
assert call.value == 1.0
|
|
assert call.start_pos == (1, 0)
|
|
assert call.end_pos == (1, 3)
|
|
|
|
def test_call_type(self):
|
|
call = self.get_call('hello')
|
|
assert isinstance(call, pr.Call)
|
|
assert type(call.name) == pr.Name
|
|
|
|
def test_literal_type(self):
|
|
literal = self.get_call('1.0')
|
|
assert isinstance(literal, pr.Literal)
|
|
assert type(literal.value) == float
|
|
|
|
literal = self.get_call('1')
|
|
assert isinstance(literal, pr.Literal)
|
|
assert type(literal.value) == int
|
|
|
|
literal = self.get_call('"hello"')
|
|
assert isinstance(literal, pr.Literal)
|
|
assert literal.value == 'hello'
|
|
|
|
|
|
class TestSubscopes():
|
|
def get_sub(self, source):
|
|
return Parser(u(source)).module.subscopes[0]
|
|
|
|
def test_subscope_names(self):
|
|
name = self.get_sub('class Foo: pass').name
|
|
assert name.start_pos == (1, len('class '))
|
|
assert name.end_pos == (1, len('class Foo'))
|
|
assert str(name) == 'Foo'
|
|
|
|
name = self.get_sub('def foo(): pass').name
|
|
assert name.start_pos == (1, len('def '))
|
|
assert name.end_pos == (1, len('def foo'))
|
|
assert str(name) == 'foo'
|
|
|
|
|
|
class TestImports():
|
|
def get_import(self, source):
|
|
return Parser(source).module.imports[0]
|
|
|
|
def test_import_names(self):
|
|
imp = self.get_import(u('import math\n'))
|
|
names = imp.get_defined_names()
|
|
assert len(names) == 1
|
|
assert str(names[0]) == 'math'
|
|
assert names[0].start_pos == (1, len('import '))
|
|
assert names[0].end_pos == (1, len('import math'))
|
|
|
|
assert imp.start_pos == (1, 0)
|
|
assert imp.end_pos == (1, len('import math'))
|
|
|
|
|
|
def test_module():
|
|
module = Parser(u('asdf'), 'example.py', no_docstr=True).module
|
|
name = module.name
|
|
assert str(name) == 'example'
|
|
assert name.start_pos == (0, 0)
|
|
assert name.end_pos == (0, 7)
|
|
|
|
module = Parser(u('asdf'), no_docstr=True).module
|
|
name = module.name
|
|
assert str(name) == ''
|
|
assert name.start_pos == (0, 0)
|
|
assert name.end_pos == (0, 0)
|
|
|
|
|
|
def test_end_pos():
|
|
s = u(dedent('''
|
|
x = ['a', 'b', 'c']
|
|
def func():
|
|
y = None
|
|
'''))
|
|
parser = Parser(s)
|
|
scope = parser.module.subscopes[0]
|
|
assert scope.start_pos == (3, 0)
|
|
assert scope.end_pos == (5, 0)
|
|
|
|
|
|
def test_carriage_return_statements():
|
|
source = u(dedent('''
|
|
foo = 'ns1!'
|
|
|
|
# this is a namespace package
|
|
'''))
|
|
source = source.replace('\n', '\r\n')
|
|
stmt = Parser(source).module.statements[0]
|
|
assert '#' not in stmt.get_code()
|