test_fatoptimizer.py 文件源码

def test_return_logger(self):
        self.config.logger = io.StringIO()

        def get_logs():
            logger = self.config.logger
            logger.seek(0)
            return logger.readlines()

        self.check_optimize("""
            def func():
                x = 1
                return 1
                return 2
        """, """
            def func():
                x = 1
                return 1
        """)

        self.assertEqual(get_logs(),
                         ['<string>:4: fatoptimizer: Remove unreachable code: '
                          'Return(value=Constant(value=2))\n'])