test_fatoptimizer.py 文件源码

def test_get_keywords(self):
        tree = compile_ast('func()')
        keywords = fatoptimizer.tools.get_keywords(tree.body[0].value)
        self.assertFalse(keywords)

        tree = compile_ast('func(x=1, y=2)')
        keywords = fatoptimizer.tools.get_keywords(tree.body[0].value)
        self.assertEqual(len(keywords), 2)
        self.assertIsInstance(keywords[0], ast.keyword)
        self.assertEqual(keywords[0].arg, 'x')
        self.assertIsInstance(keywords[1], ast.keyword)
        self.assertEqual(keywords[1].arg, 'y')

        tree = compile_ast('func(arg, *varargs, **kwargs)')
        keywords = fatoptimizer.tools.get_keywords(tree.body[0].value)
        self.assertEqual(len(keywords), 1)
        self.assertIsInstance(keywords[0], ast.keyword)
        self.assertIsNone(keywords[0].arg)

        tree = compile_ast('func()')
        with self.assertRaises(ValueError):
            fatoptimizer.tools.get_keywords(tree)