Attribute docstrings work now, fixes #138

This commit is contained in:
Dave Halter
2019-12-22 02:05:02 +01:00
parent 4161bfc7f2
commit fc785ce6ea
3 changed files with 42 additions and 3 deletions

View File

@@ -52,3 +52,26 @@ def test_lambda(Script):
)
def test_help_no_returns(Script, code, kwargs):
assert not Script(code).help(**kwargs)
def test_attribute_docstrings(goto_or_help):
code = dedent('''\
class X:
"ha"
x = 3
""" Yeah """
y = 5
"f g "
z = lambda: 1
''')
d, = goto_or_help(code + 'X.x')
assert d.docstring() == 'Yeah '
d, = goto_or_help(code + 'X().x')
assert d.docstring() == 'Yeah '
d, = goto_or_help(code + 'X.y')
assert d.docstring() == 'f g '
d, = goto_or_help(code + 'X.z')
assert d.docstring() == ''