change test

apache · Nov 19, 2019 · 57e27c4 · 57e27c4
1 parent a4030e1
commit 57e27c4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/tests/python/unittest/test_numpy_interoperability.py b/tests/python/unittest/test_numpy_interoperability.py
@@ -846,9 +846,15 @@ def _add_workload_lcm():
     OpArgMngr.add_workload('lcm', np.array(195225786*2, dtype=np.int32), np.array(195225786*5, dtype=np.int32))
 
 def _add_workload_bitwise_or():
-    OpArgMngr.add_workload('bitwise_or', np.array([12, 120], dtype=np.int8), np.array([20, 200], dtype=np.int8))
-    OpArgMngr.add_workload('bitwise_or', np.array([12, 120], dtype=np.uint8), np.array([20, 200], dtype=np.uint8))
-    OpArgMngr.add_workload('bitwise_or', np.array(195225786*2, dtype=np.int32), np.array(195225786*5, dtype=np.int32))
+    OpArgMngr.add_workload('bitwise_or', np.array([False, False, True, True], dtype=np.bool),
+                           np.array([False, True, False, True], dtype=np.bool))
+    for dtype in [np.int8, np.int32, np.int64]:
+        zeros = np.array([0], dtype=dtype)
+        ones = np.array([-1], dtype=dtype)
+        OpArgMngr.add_workload('bitwise_or', zeros, zeros)
+        OpArgMngr.add_workload('bitwise_or', ones, zeros)
+        OpArgMngr.add_workload('bitwise_or', zeros, ones)
+        OpArgMngr.add_workload('bitwise_or', ones, ones)
 
 def _add_workload_bitwise_xor():
     OpArgMngr.add_workload('bitwise_xor', np.array([False, False, True, True], dtype=np.bool),