class TestCrisAstToCodeTransformer: def log(self, title, txt): logging.info("TestCrisAstToCodeTransformer::" + title + ":\n" + txt + "\n\n") def setup_transformer(self): self.transformer = AstToCodeTransformer() @pytest.mark.skipif(True, reason='Too many logs.') def test_filter_process(self, cls_gen): self.setup_transformer() for simple_cls in cls_gen.generate(1000, 10, 10): original_source = simple_cls.get_source_code() ast_node = simple_cls.get_ast_node() source_code = self.transformer.transform(ast_node) new_ast_node = ast.parse(source_code).body[0] self.log("Original Source", original_source) self.log("New Source", source_code) self.log("Original Dump", astpp.dump(ast_node)) self.log("New Source", astpp.dump(new_ast_node)) assert ast.dump(ast_node) == ast.dump(new_ast_node)
def setup_transformer(self): self.transformer = AstToCodeTransformer()