forked from VimPlug/jedi
38 lines
813 B
Python
38 lines
813 B
Python
from textwrap import dedent
|
|
|
|
import jedi
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.skipif('sys.version_info[0] < 3')
|
|
def test_simple_annotations():
|
|
"""
|
|
Annotations only exist in Python 3.
|
|
If annotations adhere to PEP-0484, we use them (they override inference),
|
|
else they are parsed but ignored
|
|
"""
|
|
|
|
source = dedent("""\
|
|
def annot(a:3):
|
|
return a
|
|
|
|
annot('')""")
|
|
|
|
assert [d.name for d in jedi.Script(source, ).goto_definitions()] == ['str']
|
|
|
|
source = dedent("""\
|
|
|
|
def annot_ret(a:3) -> 3:
|
|
return a
|
|
|
|
annot_ret('')""")
|
|
assert [d.name for d in jedi.Script(source, ).goto_definitions()] == ['str']
|
|
|
|
source = dedent("""\
|
|
def annot(a:int):
|
|
return a
|
|
|
|
annot('')""")
|
|
|
|
assert [d.name for d in jedi.Script(source, ).goto_definitions()] == ['int']
|