def test_class_docstring_node(self): node = transformers.ParentNodeTransformer().visit(ast.parse('class X:\n\t"""doc"""')) self.assertTrue(utils.is_docstring(node.body[0].body[0].value))
def test_non_docstring_node(self): node = transformers.ParentNodeTransformer().visit(ast.parse('')) self.assertFalse(utils.is_docstring(node))
import ast from astmonkey import utils, transformers node = ast.parse('def foo(x):\n\t"""doc"""') node = transformers.ParentChildNodeTransformer().visit(node) docstring_node = node.body[0].body[0].value assert(not utils.is_docstring(node)) assert(utils.is_docstring(docstring_node))
def test_class_docstring_node(self): node = transformers.ParentNodeTransformer().visit( ast.parse('class X:\n\t"""doc"""')) self.assertTrue(utils.is_docstring(node.body[0].body[0].value))
import ast from astmonkey import utils, transformers node = ast.parse('def foo(x):\n\t"""doc"""') node = transformers.ParentNodeTransformer().visit(node) docstring_node = node.body[0].body[0].value assert(not utils.is_docstring(node)) assert(utils.is_docstring(docstring_node))
def test_non_docstring_node(self): node = transformers.ParentChildNodeTransformer().visit(ast.parse('')) assert not utils.is_docstring(node)
def test_function_docstring_node(self): node = transformers.ParentChildNodeTransformer().visit(ast.parse('def foo():\n\t"""doc"""')) assert utils.is_docstring(node.body[0].body[0].value)
def test_module_docstring_node(self): node = transformers.ParentChildNodeTransformer().visit(ast.parse('"""doc"""')) assert utils.is_docstring(node.body[0].value)