File: test_annotations.py

package info (click to toggle)
python-jedi 0.10.0~git1%2Bf05c071-1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 2,064 kB
  • ctags: 3,014
  • sloc: python: 16,997; makefile: 149; ansic: 13
file content (60 lines) | stat: -rw-r--r-- 1,459 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
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']


@pytest.mark.skipif('sys.version_info[0] < 3')
@pytest.mark.parametrize('reference', [
    'assert 1',
    '1',
    'def x(): pass',
    '1, 2',
    r'1\n'
])
def test_illegal_forward_references(reference):
    source = 'def foo(bar: "%s"): bar' % reference

    assert not jedi.Script(source).goto_definitions()


@pytest.mark.skipif('sys.version_info[0] < 3')
def test_lambda_forward_references():
    source = 'def foo(bar: "lambda: 3"): bar'

    # For now just receiving the 3 is ok. I'm doubting that this is what we
    # want. We also execute functions. Should we only execute classes?
    assert jedi.Script(source).goto_definitions()