Method assertASTTransformation

Lib/test/test_ast/test_ast.py:3071–3079

(self, transformer_class,
                                code, expected_code)

class NodeTransformerTests(ASTTestMixin, unittest.TestCase):
    def assertASTTransformation(self, transformer_class,
                                code, expected_code):
        initial_ast = ast.parse(dedent(code))
        expected_ast = ast.parse(dedent(expected_code))

        transformer = transformer_class()
        result_ast = ast.fix_missing_locations(transformer.visit(initial_ast))

        self.assertASTEqual(result_ast, expected_ast)

    def test_node_remove_single(self):
        code = 'def func(arg) -> SomeType: ...'