Exemplo n.º 1
0
    def testBasicIfElif(self):
        expected_string = """if True:
  pass
elif False:
  pass
"""
        expected_node = GetNodeFromInput(expected_string)
        test_node = create_node.If(
            create_node.Name('True'),
            orelse=[create_node.If(create_node.Name('False'))])
        self.assertNodesEqual(expected_node, test_node)
Exemplo n.º 2
0
    def testIfAndOthersInElse(self):
        expected_string = """if True:
  pass
else:
  if False:
    pass
  True
"""
        expected_node = GetNodeFromInput(expected_string)
        test_node = create_node.If(
            create_node.Name('True'),
            orelse=[
                create_node.If(create_node.Name('False')),
                create_node.Expr(create_node.Name('True'))
            ])
        self.assertNodesEqual(expected_node, test_node)
Exemplo n.º 3
0
 def testBasicIf(self):
     expected_string = """if True:\n  pass"""
     expected_node = GetNodeFromInput(expected_string)
     test_node = create_node.If(create_node.Name('True'))
     self.assertNodesEqual(expected_node, test_node)