Ejemplo n.º 1
0
    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))
Ejemplo n.º 2
0
    def test_non_docstring_node(self):
        node = transformers.ParentNodeTransformer().visit(ast.parse(''))

        self.assertFalse(utils.is_docstring(node))
Ejemplo n.º 3
0
    def test_non_docstring_node(self):
        node = transformers.ParentNodeTransformer().visit(ast.parse(''))

        self.assertFalse(utils.is_docstring(node))
Ejemplo n.º 4
0
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))
Ejemplo n.º 5
0
    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))
Ejemplo n.º 6
0
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))
Ejemplo n.º 7
0
    def test_non_docstring_node(self):
        node = transformers.ParentChildNodeTransformer().visit(ast.parse(''))

        assert not utils.is_docstring(node)
Ejemplo n.º 8
0
    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)
Ejemplo n.º 9
0
    def test_module_docstring_node(self):
        node = transformers.ParentChildNodeTransformer().visit(ast.parse('"""doc"""'))

        assert utils.is_docstring(node.body[0].value)