is_docstring.py
1 import ast 2 from astmonkey import utils, transformers 3 4 node = ast.parse('def foo(x):\n\t"""doc"""') 5 node = transformers.ParentChildNodeTransformer().visit(node) 6 7 docstring_node = node.body[0].body[0].value 8 assert(not utils.is_docstring(node)) 9 assert(utils.is_docstring(docstring_node))