From c6d5538eade22fec6587a0177631dcb5fabfa63e Mon Sep 17 00:00:00 2001
From: Siu Kwan Lam <michael.lam.sk@gmail.com>
Date: Wed, 8 Aug 2012 17:33:34 -0700
Subject: [PATCH] Add usecase tests

---
 test_usecase_1_1.py | 56 +++++++++++++++++++++++++++++++++++++++++++
 test_usecase_2_1.py | 58 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 114 insertions(+)
 create mode 100644 test_usecase_1_1.py
 create mode 100644 test_usecase_2_1.py

diff --git a/test_usecase_1_1.py b/test_usecase_1_1.py
new file mode 100644
index 0000000..5cc4d7a
--- /dev/null
+++ b/test_usecase_1_1.py
@@ -0,0 +1,56 @@
+from parallel_vectorize import *
+from llvm_cbuilder import shortnames as C
+from llvm.core import *
+import numpy as np
+import unittest
+from random import random
+
+class OneOne(CDefinition):
+
+    def body(self, inval):
+        self.ret( (inval * inval).cast(self.OUT_TYPE) )
+
+    @classmethod
+    def specialize(cls, itype, otype):
+        cls._name_ = '.'.join(map(str, ['oneone', itype, otype]))
+        cls._retty_ = otype
+        cls._argtys_ = [
+            ('inval', itype),
+        ]
+        cls.OUT_TYPE = otype
+
+
+class TestParallelVectorize(unittest.TestCase):
+    def test_parallelvectorize_d_d(self):
+        self.template(C.double, C.double)
+
+    def test_parallelvectorize_d_f(self):
+        self.template(C.double, C.float)
+
+    def template(self, itype, otype):
+        module = Module.new(__name__)
+        exe = CExecutor(module)
+
+        def_oneone = OneOne(itype, otype)
+        oneone = def_oneone(module)
+        ufunc = parallel_vectorize_from_func(oneone, exe.engine)
+        # print(module)
+        module.verify()
+
+        x = np.linspace(.0, 10., 1000)
+        x.dtype = np.double
+
+        ans = ufunc(x)
+        gold = x * x
+
+        for x, y in zip(ans, gold):
+            if y != 0:
+                err = abs(x - y)/y
+                self.assertLess(err, 1e-6)
+            else:
+                self.assertEqual(x, y)
+
+
+if __name__ == '__main__':
+    unittest.main()
+
diff --git a/test_usecase_2_1.py b/test_usecase_2_1.py
new file mode 100644
index 0000000..087ee9f
--- /dev/null
+++ b/test_usecase_2_1.py
@@ -0,0 +1,58 @@
+from parallel_vectorize import *
+from llvm_cbuilder import shortnames as C
+from llvm.core import *
+import numpy as np
+import unittest
+from random import random
+
+class TwoOne(CDefinition):
+
+    def body(self, a, b):
+        self.ret( (a * b).cast(self.OUT_TYPE) )
+
+    @classmethod
+    def specialize(cls, itype1, itype2, otype):
+        cls._name_ = '.'.join(map(str, ['oneone', itype1, itype2, otype]))
+        cls._retty_ = otype
+        cls._argtys_ = [
+            ('a', itype1),
+            ('b', itype2),
+        ]
+        cls.OUT_TYPE = otype
+
+
+class TestParallelVectorize(unittest.TestCase):
+    def test_parallelvectorize_dd_d(self):
+        self.template(C.double, C.double, C.double)
+
+    def test_parallelvectorize_dd_f(self):
+        self.template(C.double, C.double, C.float)
+
+    def template(self, itype1, itype2, otype):
+        module = Module.new(__name__)
+        exe = CExecutor(module)
+
+        def_twoone = TwoOne(itype1, itype2, otype)
+        twoone = def_twoone(module)
+        ufunc = parallel_vectorize_from_func(twoone, exe.engine)
+        # print(module)
+        module.verify()
+
+        A = np.linspace(.0, 10., 1000)
+        A.dtype = np.double
+        B = np.linspace(-10., 0., 1000)
+        B.dtype = np.double
+
+        ans = ufunc(A, B)
+        gold = A * B
+
+        for x, y in zip(ans, gold):
+            if y != 0:
+                err = abs(x - y)/y
+                self.assertLess(err, 1e-6)
+            else:
+                self.assertEqual(x, y)
+
+if __name__ == '__main__':
+    unittest.main()
+