Fix Definition.get_definitions

This commit is contained in:
Takafumi Arakaki
2013-02-13 23:26:17 +01:00
parent 0f67b3e0c2
commit 08f45d1f95
2 changed files with 15 additions and 1 deletions

View File

@@ -364,7 +364,7 @@ class Definition(BaseDefinition):
:rtype: list of Definition
"""
return get_definitions(self._parser.scope)
return get_definitions(self.definition)
def get_definitions(scope):

View File

@@ -425,6 +425,20 @@ class TestGetDefinitions(TestBase):
assert definitions[1].names == ['b']
assert definitions[2].names == []
def test_nested_definitions(self):
definitions = api.get_definitions("""
class Class:
def f():
pass
def g():
pass
""")
assert len(definitions) == 1
assert definitions[0].names == ['Class']
subdefinitions = definitions[0].get_definitions()
assert subdefinitions[0].names == ['f']
assert subdefinitions[1].names == ['g']
class TestSpeed(TestBase):
def _check_speed(time_per_run, number=4, run_warm=True):