From 5e34a1ce099bfdbb0b6c4292d8f293ecfa3a38dd Mon Sep 17 00:00:00 2001
From: "Kalistratov, Alexander" <alexander.kalistratov@intel.com>
Date: Sun, 5 Jan 2020 14:54:46 +0300
Subject: [PATCH] Add parallel test with condition
---
sdc/tests/test_hpat_jit.py | 19 +++++++++++++++++++
1 file changed, 19 insertions(+)
diff --git a/sdc/tests/test_hpat_jit.py b/sdc/tests/test_hpat_jit.py
index d585bd233..798006a45 100644
--- a/sdc/tests/test_hpat_jit.py
+++ b/sdc/tests/test_hpat_jit.py
@@ -436,6 +436,25 @@ def test_impl():
hpat_func = self.jit(test_impl)
pd.testing.assert_frame_equal(hpat_func(), test_impl())
+ @unittest.expectedFailure
+ def test_parallel_with_condition_1(self):
+ import numpy
+ import numba
+
+ def test_impl(arr, half=False):
+ size = len(arr)
+ parr = arr[0:size]
+
+ if half:
+ parr = arr[0:size//2]
+
+ return parr.sum()
+
+ jtest = numba.njit(test_impl)
+ ptest = numba.njit(test_impl, parallel=True)
+
+ arr = numpy.ones(128)
+ self.assertEqual(jtest(arr, True), ptest(arr, True))
if __name__ == "__main__":
unittest.main()