Add support for BoolOp.

artiq/py2llvm/typing.py

@@ -301,18 +301,24 @@ class Inferencer(algorithm.Transformer):
 
     def visit_BoolOp(self, node):
         node = self.generic_visit(node)
-        for value, op_loc in zip(node.values, node.op_locs):
+        node = asttyped.BoolOpT(type=types.TVar(),
+                                op=node.op, values=node.values,
+                                op_locs=node.op_locs, loc=node.loc)
         def makenotes(printer, typea, typeb, loca, locb):
             return [
                 diagnostic.Diagnostic("note",
-                        "py2llvm requires boolean operations to have boolean operands", {},
+                    "an operand of type {typea}",
-                        op_loc)
+                    {"typea": printer.name(node.values[0].type)},
+                    node.values[0].loc),
+                diagnostic.Diagnostic("note",
+                    "an operand of type {typeb}",
+                    {"typeb": printer.name(typeb)},
+                    locb)
             ]
-            self._unify(value.type, types.TBool(),
+        for value in node.values:
-                        value.loc, None, makenotes)
+            self._unify(node.type, value.type,
-        return asttyped.BoolOpT(type=types.TBool(),
+                        node.loc, value.loc, makenotes)
-                                op=node.op, values=node.values,
+        return node
-                                op_locs=node.op_locs, loc=node.loc)
 
     # Visitors that just unify types
     #

lit-test/py2llvm/typing/error_unify.py

@@ -11,3 +11,8 @@ a = b
 [1, []]
 # CHECK-L: note: a list of type list(elt=int(width='a))
 # CHECK-L: note: a list element of type list(elt='b)
+
+# CHECK-L: ${LINE:+1}: error: cannot unify int(width='a) with bool
+1 and False
+# CHECK-L: note: an operand of type int(width='a)
+# CHECK-L: note: an operand of type bool

lit-test/py2llvm/typing/unify.py

@@ -38,3 +38,6 @@ j += [1.0]
 
 True and False
 # CHECK-L: True:bool and False:bool:bool
+
+1 and 0
+# CHECK-L: 1:int(width='g) and 0:int(width='g):int(width='g)