def testNameKwargs(self): expected_string = """def testFunc(a=b, c=d): pass""" expected_node = GetNodeFromInput(expected_string).args test_node = create_node.arguments( keys=['a', 'c'], values=[create_node.Name('b'), create_node.Name('d')]) self.assertNodesEqual(expected_node, test_node)
def testVararg(self): expected_string = """def testFunc(*args): pass""" expected_node = GetNodeFromInput(expected_string).args test_node = create_node.arguments(vararg_name='args') self.assertNodesEqual(expected_node, test_node)
def testArgs(self): expected_string = """def testFunc(a, b): pass""" expected_node = GetNodeFromInput(expected_string).args test_node = create_node.arguments(args=('a', 'b')) self.assertNodesEqual(expected_node, test_node)
def testEmpty(self): expected_string = """def testFunc(): pass""" expected_node = GetNodeFromInput(expected_string).args test_node = create_node.arguments() self.assertNodesEqual(expected_node, test_node)
def testFunctionDefWithKwarg(self): expected_string = """def testFunc(**kwargs): pass""" expected_node = GetNodeFromInput(expected_string).args test_node = create_node.arguments(kwarg_name='kwargs') self.assertNodesEqual(expected_node, test_node)