VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-06 14:04:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make py__iter__ work as well for Interpreter

Browse Source
This commit is contained in:
Dave Halter
2021-01-01 17:58:31 +01:00
parent 971913be35
commit 1ccc63e83d
2 changed files with 31 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
test/test_api/test_interpreter.py
View File

@@ -600,16 +600,31 @@ def test_dict_getitem(code, types):
@pytest.mark.parametrize('class_is_findable', [False, True])
def test__getitem__(class_is_findable):
class GetitemCls:
@pytest.mark.parametrize(
'code, expected', [
('DunderCls()[0]', 'int'),
('next(DunderCls())', 'float'),
('for x in DunderCls(): x', 'str'),
]
)
def test_dunders(class_is_findable, code, expected):
from typing import Iterator
class DunderCls:
def __getitem__(self, key) -> int:
pass
if not class_is_findable:
GetitemCls.__name__ = 'asdf'
def __iter__(self, key) -> Iterator[str]:
pass
n, = jedi.Interpreter('GetitemCls()[0]', [locals()]).infer()
assert n.name == 'int'
def __next__(self, key) -> float:
pass
if not class_is_findable:
DunderCls.__name__ = 'asdf'
n, = jedi.Interpreter(code, [locals()]).infer()
assert n.name == expected
def foo():