VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

Evaluate annotations for dataclasses when infer is called on param

Browse Source
This commit is contained in:
Dave Halter
2019-07-19 11:42:08 +02:00
parent 93c52f615a
commit 5722a3458e
2 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
test/test_evaluate/test_signature.py
View File

@@ -86,12 +86,16 @@ def test_dataclass_signature(Script, skip_pre_python37, start, start_params):
code = dedent('''
name: str
foo = 3
unit_price: float
quantity_on_hand: int = 0
price: float
quantity: int = 0.0
X(''')
code = 'from dataclasses import dataclass\n' + start + code
sig, = Script(code).call_signatures()
assert [p.name for p in sig.params] == start_params + ['name', 'unit_price', 'quantity_on_hand']
assert [p.name for p in sig.params] == start_params + ['name', 'price', 'quantity']
quantity, = sig.params[-1].infer()
assert quantity.name == 'int'
price, = sig.params[-2].infer()
assert price.name == 'float'