diff --git a/test/completion/basic.py b/test/completion/basic.py index 844b6e9f..6cf7028a 100644 --- a/test/completion/basic.py +++ b/test/completion/basic.py @@ -177,6 +177,13 @@ left, right = (i for i in (1, '')) #? int() left +gen = (i for i in (1,)) + +#? int() +next(gen) +#? +gen[0] + # ----------------- # ternary operator # ----------------- diff --git a/test/completion/generators.py b/test/completion/generators.py index c5677877..5926245f 100644 --- a/test/completion/generators.py +++ b/test/completion/generators.py @@ -110,6 +110,18 @@ a #? str() float() b + +def simple(): + yield 1 + yield '' + +a, b = simple() +#? int() +a +#? str() +b + + # ----------------- # More complicated access # -----------------