def testDeleteInvalid(self): expected_string = 'del (a, b)' expected_node = GetNodeFromInput(expected_string).targets[0] test_node = create_node.Tuple(create_node.Name('a'), create_node.Name('b'), ctx_type=create_node.CtxEnum.DEL) self.assertNodesEqual(expected_node, test_node)
def testTupleStore(self): expected_string = '(a, b) = ["c", "d"]' expected_node = GetNodeFromInput(expected_string).targets[0] test_node = create_node.Tuple(create_node.Name('a'), create_node.Name('b'), ctx_type=create_node.CtxEnum.STORE) self.assertNodesEqual(expected_node, test_node)
def testTupleWithStrings(self): expected_string = 'a = (b,c)' expected_node = GetNodeFromInput(expected_string).value test_node = create_node.Tuple('b', 'c', ctx_type=create_node.CtxEnum.LOAD) self.assertNodesEqual(expected_node, test_node)
def testAssignWithTuple(self): expected_string = '(a, c) = "b"' expected_node = GetNodeFromInput(expected_string) test_node = create_node.Assign( create_node.Tuple('a', 'c', ctx_type=create_node.CtxEnum.STORE), create_node.Str('b')) self.assertNodesEqual(expected_node, test_node)
def testWithAsTuple(self): expected_string = 'with a as (b, c):\n pass\n' expected_node = GetNodeFromInput(expected_string) test_node = create_node.With(create_node.Name('a'), as_part=create_node.Tuple( create_node.Name('b'), create_node.Name('c'))) self.assertNodesEqual(expected_node, test_node)
def testTupleOverridesInnerCtx(self): expected_string = 'a = (b, c)' expected_node = GetNodeFromInput(expected_string).value test_node = create_node.Tuple( create_node.Name('b', ctx_type=create_node.CtxEnum.DEL), create_node.Name('c', ctx_type=create_node.CtxEnum.STORE), ctx_type=create_node.CtxEnum.LOAD) self.assertNodesEqual(expected_node, test_node)
def testTupleLoad(self): expected_string = 'a = ("b",)' expected_node = GetNodeFromInput(expected_string).value test_node = create_node.Tuple(create_node.Str('b'), ctx_type=create_node.CtxEnum.LOAD) self.assertNodesEqual(expected_node, test_node)