Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions sdc/tests/test_basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,6 +86,7 @@ def _follow_cpython(self, ptr, seed=2):

class TestBasic(BaseTest):

@skip_numba_jit("hang with numba.jit. ok with sdc.jit")
def test_getitem(self):
def test_impl(N):
A = np.ones(N)
Expand All @@ -99,6 +100,7 @@ def test_impl(N):
self.assertEqual(count_array_REPs(), 0)
self.assertEqual(count_parfor_REPs(), 0)

@skip_numba_jit("Failed in nopython mode pipeline (step: Preprocessing for parfors)")
def test_setitem1(self):
def test_impl(N):
A = np.arange(10) + 1.0
Expand All @@ -111,6 +113,7 @@ def test_impl(N):
self.assertEqual(count_array_REPs(), 0)
self.assertEqual(count_parfor_REPs(), 0)

@skip_numba_jit("Failed in nopython mode pipeline (step: Preprocessing for parfors)")
def test_setitem2(self):
def test_impl(N):
A = np.arange(10) + 1.0
Expand All @@ -123,6 +126,7 @@ def test_impl(N):
self.assertEqual(count_array_REPs(), 0)
self.assertEqual(count_parfor_REPs(), 0)

@skip_numba_jit("hang with numba.jit. ok with sdc.jit")
def test_astype(self):
def test_impl(N):
return np.ones(N).astype(np.int32).sum()
Expand Down Expand Up @@ -152,6 +156,7 @@ def test_impl(N):
# self.assertEqual(count_array_REPs(), 0)
# self.assertEqual(count_parfor_REPs(), 0)

@skip_numba_jit("hang with numba.jit. ok with sdc.jit")
def test_inplace_binop(self):
def test_impl(N):
A = np.ones(N)
Expand All @@ -165,6 +170,7 @@ def test_impl(N):
self.assertEqual(count_array_REPs(), 0)
self.assertEqual(count_parfor_REPs(), 0)

@skip_numba_jit("hang with numba.jit. ok with sdc.jit")
def test_getitem_multidim(self):
def test_impl(N):
A = np.ones((N, 3))
Expand All @@ -178,6 +184,7 @@ def test_impl(N):
self.assertEqual(count_array_REPs(), 0)
self.assertEqual(count_parfor_REPs(), 0)

@skip_numba_jit("Failed in nopython mode pipeline (step: Preprocessing for parfors)")
def test_whole_slice(self):
def test_impl(N):
X = np.ones((N, 4))
Expand All @@ -190,6 +197,7 @@ def test_impl(N):
self.assertEqual(count_array_REPs(), 0)
self.assertEqual(count_parfor_REPs(), 0)

@skip_numba_jit("hang with numba.jit. ok with sdc.jit")
def test_strided_getitem(self):
def test_impl(N):
A = np.ones(N)
Expand Down Expand Up @@ -229,6 +237,7 @@ def f():

pd.testing.assert_series_equal(self.jit(f)(), f())

@skip_numba_jit("Failed in nopython mode pipeline (step: Preprocessing for parfors)")
def test_reduce(self):
import sys
dtypes = ['float32', 'float64', 'int32', 'int64']
Expand Down Expand Up @@ -423,6 +432,7 @@ def test_impl(N):
finally:
sdc.distributed_analysis.auto_rebalance = False

@skip_numba_jit("Failed in nopython mode pipeline (step: Preprocessing for parfors)")
def test_transpose(self):
def test_impl(n):
A = np.ones((30, 40, 50))
Expand Down
3 changes: 2 additions & 1 deletion sdc/tests/test_hiframes.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@
from sdc.tests.test_utils import (count_array_REPs, count_parfor_REPs,
count_parfor_OneDs, count_array_OneDs, dist_IR_contains,
get_start_end,
skip_numba_jit)
skip_sdc_jit, skip_numba_jit)


class TestHiFrames(TestCase):
Expand Down Expand Up @@ -424,6 +424,7 @@ def test_impl(df):
pd.testing.assert_series_equal(
hpat_func(df), test_impl(df), check_names=False)

@skip_sdc_jit("Could not get length of Series(StringArraySplitView)")
@skip_numba_jit
def test_str_split_filter(self):
def test_impl(df):
Expand Down