tests

Author: Alexander-Makaryev

mkl_umath/tests/test_basic.py

@@ -1,3 +1,67 @@
+import numpy as np
+import mkl_umath._ufuncs as mu
+import numpy.core.umath as nu
 
-def test1():
-    pass
+np.random.seed(42)
+
+def get_args(args_str):
+    args = []
+    for s in args_str:
+        if s == 'f':
+            args.append(np.single(np.random.random_sample()))
+        elif s == 'd':
+            args.append(np.double(np.random.random_sample()))
+        elif s == 'F':
+            args.append(np.single(np.random.random_sample()) + np.single(np.random.random_sample()) * 1j)
+        elif s == 'D':
+            args.append(np.double(np.random.random_sample()) + np.double(np.random.random_sample()) * 1j)
+        elif s == 'i':
+            args.append(np.int(np.random.randint(low=1, high=10)))
+        elif s == 'l':
+            args.append(np.long(np.random.randint(low=1, high=10)))
+        else:
+            raise ValueError("Unexpected type specified!")
+    return tuple(args)
+
+umaths = [i for i in dir(mu) if isinstance(getattr(mu, i), np.ufunc)]
+
+umaths.remove('arccosh') # expects input greater than 1
+
+# dictionary with test cases
+# (umath, types) : args
+generated_cases = {}
+for umath in umaths:
+    mkl_umath = getattr(mu, umath)
+    types = mkl_umath.types
+    for type in types:
+        args_str = type[:type.find('->')]
+        args = get_args(args_str)
+        generated_cases[(umath, type)] = args
+
+additional_cases = {
+('arccosh', 'f->f') : (np.single(np.random.random_sample() + 1),),
+('arccosh', 'd->d') : (np.double(np.random.random_sample() + 1),),
+}
+
+test_cases = {}
+for d in (generated_cases, additional_cases):
+    test_cases.update(d)
+
+for case in test_cases:
+    umath = case[0]
+    type = case[1]
+    args = test_cases[case]
+    mkl_umath = getattr(mu, umath)
+    np_umath = getattr(nu, umath)
+    print('*'*80)
+    print(umath, type)
+    print("args", args)
+    mkl_res = mkl_umath(*args)
+    np_res = np_umath(*args)
+    print("mkl res", mkl_res)
+    print("npy res", np_res)
+
+    assert np.array_equal(mkl_res, np_res)
+
+print("Test cases count:", len(test_cases))
+print("All looks good!")