From 57e27c4cec7d961da0cabb8a56e22156763eed87 Mon Sep 17 00:00:00 2001 From: Yiyan66 Date: Tue, 19 Nov 2019 11:51:17 +0000 Subject: [PATCH] change test --- tests/python/unittest/test_numpy_interoperability.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/python/unittest/test_numpy_interoperability.py b/tests/python/unittest/test_numpy_interoperability.py index 2ef8f69f4e86..074e28fb3cb4 100644 --- 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),