forked from VimPlug/jedi
Create a name for the generators.
This commit is contained in:
@@ -158,6 +158,10 @@ class GeneratorMixin(object):
|
|||||||
gen_obj = compiled.get_special_object(self.evaluator, 'GENERATOR_OBJECT')
|
gen_obj = compiled.get_special_object(self.evaluator, 'GENERATOR_OBJECT')
|
||||||
return gen_obj.py__class__()
|
return gen_obj.py__class__()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self):
|
||||||
|
return compiled.CompiledContextName(self, 'generator')
|
||||||
|
|
||||||
|
|
||||||
class Generator(GeneratorMixin, context.Context):
|
class Generator(GeneratorMixin, context.Context):
|
||||||
"""Handling of `yield` functions."""
|
"""Handling of `yield` functions."""
|
||||||
|
|||||||
@@ -119,6 +119,11 @@ def test_goto_definitions_on_non_name():
|
|||||||
assert api.Script('import x', column=0).goto_definitions() == []
|
assert api.Script('import x', column=0).goto_definitions() == []
|
||||||
|
|
||||||
|
|
||||||
|
def test_goto_definitions_on_generator():
|
||||||
|
def_, = api.Script('def x(): yield 1\ny=x()\ny').goto_definitions()
|
||||||
|
assert def_.name == 'generator'
|
||||||
|
|
||||||
|
|
||||||
def test_goto_definition_not_multiple():
|
def test_goto_definition_not_multiple():
|
||||||
"""
|
"""
|
||||||
There should be only one Definition result if it leads back to the same
|
There should be only one Definition result if it leads back to the same
|
||||||
|
|||||||
Reference in New Issue
Block a user