Method assertAstMatches

dali/test/python/autograph/pyct/test_parser.py:31–42

(self, actual_node, expected_node_src, expr=True)

class ParserTest(unittest.TestCase):
    def assertAstMatches(self, actual_node, expected_node_src, expr=True):
        if expr:
            # Ensure multi-line expressions parse.
            expected_node = gast.parse("({})".format(expected_node_src)).body[0]
            expected_node = expected_node.value
        else:
            expected_node = gast.parse(expected_node_src).body[0]

        msg = "AST did not match expected:\n{}\nActual:\n{}".format(
            pretty_printer.fmt(expected_node), pretty_printer.fmt(actual_node)
        )
        self.assertTrue(ast_util.matches(actual_node, expected_node), msg)

    def test_parse_entity(self):
        def f(x):