9 examples of 'python dict comprehension if else' in Python

332
def visitIf(self, n):
333
    if not n.else_:
334
        return
335
    visits = []
336
    for test, code in n.tests:
337
        visits.append(walk(code, Visitor(self.defines, self.uses)))
338
    visits.append(walk(n.else_, Visitor(self.defines, self.uses)))
339
    # compute the intersection of defines
340
    self.defines = intersect([v.defines for v in visits])
341
    # compute the union of uses, perserving first occurances
342
    union = {}
343
    visits.reverse()
344
    for v in visits:
345
        union.update(v.uses)
346
    union.update(self.uses)
347
    self.uses = union

1042
@utils.check_messages("using-constant-test", "missing-parentheses-for-call-in-test")
1043
def visit_comprehension(self, node):
1044
    if node.ifs:
1045
        for if_test in node.ifs:
1046
            self._check_using_constant_test(node, if_test)

1276
def visit_ComprehensionNode(self, node):
1277
    if node.expr_scope:
1278
        self.env_stack.append(self.env)
1279
        self.env = node.expr_scope
1280
    # Skip append node here
1281
    self._visit(node.loop)
1282
    if node.expr_scope:
1283
        self.env = self.env_stack.pop()
1284
    return node

79
def visit_If(self, n: ca.If) -> str:
80
    n2 = n
81
    if (
82
        n.iffalse
83
        and isinstance(n.iffalse, ca.Compound)
84
        and n.iffalse.block_items
85
        and len(n.iffalse.block_items) == 1
86
        and isinstance(n.iffalse.block_items[0], ca.If)
87
    ):
88
        n2 = ca.If(cond=n.cond, iftrue=n.iftrue, iffalse=n.iffalse.block_items[0])
89
    return super().visit_If(n2)  # type: ignore

45
def create_comprehension(for_node: ast.For) -> ast.comprehension:
46
    return ast.comprehension(target=for_node.target, iter=for_node.iter, ifs=[])

51
def test_creating_a_dictionary_with_dictionary_comprehension(self):
52
    dict_of_weapons = {'first': 'fear', 'second': 'surprise',
53
                       'third':'ruthless efficiency', 'forth':'fanatical devotion',
54
                       'fifth': None}
55
56
    dict_comprehension = { k.upper(): weapon for k, weapon in dict_of_weapons.iteritems() if weapon}
57
58
    self.assertEqual(False, 'first' in dict_comprehension)
59
    self.assertEqual(True, 'FIRST' in dict_comprehension)
60
    self.assertEqual(5, len(dict_of_weapons))
61
    self.assertEqual(4, len(dict_comprehension))

446
def visit_comprehension(self, node, parent):
447
    """visit a Comprehension node by returning a fresh instance of it"""
448
    newnode = nodes.Comprehension(parent)
449
    newnode.postinit(
450
        self.visit(node.target, newnode),
451
        self.visit(node.iter, newnode),
452
        [self.visit(child, newnode) for child in node.ifs],
453
        getattr(node, "is_async", None),
454
    )
455
    return newnode

363
def test_dict_comprehension(self):
364
    tree = ast.parse('a = {int:"a" for int in range(3,9)}')
365
    checker = BuiltinsChecker(tree, '/home/script.py')
366
    ret = [c for c in checker.run()]
367
    self.assertEqual(len(ret), 1)

595
def if_else(self, cond, if_body, else_body):
596
    return IfNode(self.pos, cond=cond, body=if_body, else_body=else_body)