Example #1
0
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)
Example #2
0
 def setup_transformer(self):
     self.transformer = AstToCodeTransformer()