From 58465e49fad776dba334bc261a0cea5d4f638756 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Sun, 7 Sep 2014 14:07:07 +0800
Subject: [PATCH] test/py2llvm: clean up test_types function

---
 test/py2llvm.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/test/py2llvm.py b/test/py2llvm.py
index 7b0074421..e33daac86 100644
--- a/test/py2llvm.py
+++ b/test/py2llvm.py
@@ -6,6 +6,7 @@ from llvm import core as lc
 from llvm import passes as lp
 from llvm import ee as le
 
+from artiq.language.core import int64
 from artiq.py2llvm.infer_types import infer_function_types
 from artiq.py2llvm import values
 from artiq.py2llvm import compile_function
@@ -22,10 +23,10 @@ def test_types(choice):
     foo = True
     bar = None
 
-    if choice:
-        return 3
+    if choice and foo and not bar:
+        return d
     else:
-        return x
+        return x + c
 
 class FunctionTypesCase(unittest.TestCase):
     def setUp(self):