def dump_nodetype(self):
if not self.is_debug:
return
for node, ty in self.nodetype.items():
print("{} : \x1b[36m{}\x1b[39m".format(
utils.node_description(node), ty))
print()
def generate_assertion(type_table_name, id2type, id2node, ofile=None):
for i, t in sorted(id2type.items()):
node = id2node[i]
comment = "\t# " + node_description(node)
output = " self.assertEqual(str({}[{}]), \"{}\"){}".format( \
type_table_name, i, t, comment)
if ofile is None:
print(output)
else:
ofile.write(output + '\n')
def generate_assertion(type_table_name, id2type, id2node, ofile=None):
for i, t in sorted(id2type.items()):
node = id2node[i]
if not is_expr(node):
if not isinstance(node, gast.FunctionDef):
continue
output = " # === function {} ===".format(node.name)
else:
comment = "\t# " + node_description(node)
output = " self.assertEqual(str({}[{}]), \"{}\"){}".format( \
type_table_name, i, t, comment)
if ofile is None:
print(output)
else:
ofile.write(output + '\n')
def dump_stack(self):
print("=== stack ===\x1b[32m")
for node in self.stack:
print(utils.node_description(node))
print("\x1b[39m")
def dump_one_node(self, node):
if node not in self.nodetype.keys():
return
print("{} : \x1b[36m{}\x1b[39m".format(utils.node_description(node),
self.nodetype[node]))