Add llvm_cbuilder tests

llvm_cbuilder/tests/test_print.py

@@ -0,0 +1,68 @@
+from llvm.core import *
+from llvm.passes import *
+from llvm.ee import *
+from llvm_cbuilder import *
+import llvm_cbuilder.shortnames as C
+import sys, unittest, logging
+from subprocess import Popen, PIPE
+
+def gen_debugprint(mod):
+    functype = Type.function(C.void, [])
+    func = mod.add_function(functype, 'debugprint')
+
+    cb = CBuilder(func)
+    fmt = cb.constant_string("Show %d %.3f %.3e\n")
+
+    an_int = cb.constant(C.int, 123)
+    a_float = cb.constant(C.double, 1.234)
+    a_double = cb.constant(C.double, 1e-31)
+    cb.printf(fmt, an_int, a_float, a_double)
+
+    cb.debug('an_int =', an_int, 'a_float =', a_float, 'a_double =', a_double)
+
+    cb.ret()
+    cb.close()
+    return func
+
+def main_debugprint():
+    # generate code
+    mod = Module.new(__name__)
+    lfunc = gen_debugprint(mod)
+    logging.debug(mod)
+    mod.verify()
+    # run
+    exe = CExecutor(mod)
+    func = exe.get_ctype_function(lfunc, 'void')
+    func()
+
+class TestPrint(unittest.TestCase):
+    def test_debugprint(self):
+        p = Popen([sys.executable, __file__, "-child"], stdout=PIPE)
+        p.wait()
+
+        # The encode(utf-8) is for Python 3 compatibility
+        lines = p.stdout.read().encode('utf-8').splitlines(False)
+
+        # Try to account for variations in the system printf
+        if lines[0].find('e-031') >= 0:
+            expect = [
+                'Show 123 1.234 1.000e-031',
+                'an_int = 123 a_float = 1.234000e+000 a_double = 1.000000e-031',
+                ]
+        else:
+            expect = [
+                'Show 123 1.234 1.000e-31',
+                'an_int = 123 a_float = 1.234000e+00 a_double = 1.000000e-31',
+                ]
+        self.assertEqual(expect, lines)
+
+        p.stdout.close()
+
+if __name__ == '__main__':
+    try:
+        if sys.argv[1] == '-child':
+            main_debugprint()
+    except IndexError:
+        unittest.main()
+
+