Пример #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))
Пример #2
0
    def test_non_docstring_node(self):
        node = transformers.ParentNodeTransformer().visit(ast.parse(''))

        self.assertFalse(utils.is_docstring(node))
Пример #3
0
    def test_non_docstring_node(self):
        node = transformers.ParentNodeTransformer().visit(ast.parse(''))

        self.assertFalse(utils.is_docstring(node))
Пример #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))
Пример #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))
Пример #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))
Пример #7
0
    def test_non_docstring_node(self):
        node = transformers.ParentChildNodeTransformer().visit(ast.parse(''))

        assert not utils.is_docstring(node)
Пример #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)
Пример #9
0
    def test_module_docstring_node(self):
        node = transformers.ParentChildNodeTransformer().visit(ast.parse('"""doc"""'))

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