test_fatoptimizer.py 文件源码

def test_transformer_call_visitor(self):
        class BuggyTransformer(fatoptimizer.tools.NodeTransformer):
            def visit_Module(self, node):
                # visit_Module() calls indirectly visit_BinOp(),
                # but the exception must only be wrapped once
                self.generic_visit(node)

            def visit_BinOp(self, node):
                raise Exception("bug")

        visitor = BuggyTransformer("<string>")
        self.check_call_visitor(visitor)