/ examples / is_docstring.py
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))