mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 14:04:26 +08:00
97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
from jedi.parser import Parser
|
|
from jedi.parser.user_context import UserContextParser
|
|
from jedi.parser import representation as pr
|
|
|
|
|
|
def test_user_statement_on_import():
|
|
"""github #285"""
|
|
s = "from datetime import (\n" \
|
|
" time)"
|
|
|
|
for pos in [(2, 1), (2, 4)]:
|
|
u = UserContextParser(s, None, pos, None).user_stmt()
|
|
assert isinstance(u, pr.Import)
|
|
assert u.defunct is False
|
|
assert [str(n) for n in u.get_defined_names()] == ['time']
|
|
|
|
|
|
class TestCallAndName():
|
|
def get_call(self, source):
|
|
stmt = Parser(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(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('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('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, 0)
|
|
|
|
module = Parser('asdf', no_docstr=True).module
|
|
name = module.name
|
|
assert str(name) == ''
|
|
assert name.start_pos == (0, 0)
|
|
assert name.end_pos == (0, 0)
|