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)