Skip to content

Commit acb7daf

Browse files
committed
Append a skipif
1 parent 7abe783 commit acb7daf

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

pandas/tests/apply/test_numba.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,15 @@
1212
import pandas._testing as tm
1313
from pandas.util.version import Version
1414

15-
pytestmark = [td.skip_if_no("numba"), pytest.mark.single_cpu]
15+
pytestmark = [td.skip_if_no("numba"), pytest.mark.single_cpu, pytest.mark.skipif()]
16+
17+
numba = pytest.importorskip("numba")
18+
pytestmark.append(
19+
pytest.mark.skipif(
20+
Version(numba.__version__) == Version("0.61") and is_platform_arm(),
21+
reason=f"Segfaults on ARM platforms with numba {numba.__version__}",
22+
)
23+
)
1624

1725

1826
@pytest.fixture(params=[0, 1])
@@ -21,9 +29,6 @@ def apply_axis(request):
2129

2230

2331
def test_numba_vs_python_noop(float_frame, apply_axis):
24-
numba = pytest.importorskip("numba")
25-
if Version(numba.__version__) == Version("0.61") and is_platform_arm():
26-
pytest.skip(f"Segfaults on ARM platforms with numba {numba.__version__}")
2732
func = lambda x: x
2833
result = float_frame.apply(func, engine="numba", axis=apply_axis)
2934
expected = float_frame.apply(func, engine="python", axis=apply_axis)
@@ -32,9 +37,6 @@ def test_numba_vs_python_noop(float_frame, apply_axis):
3237

3338
def test_numba_vs_python_string_index():
3439
# GH#56189
35-
numba = pytest.importorskip("numba")
36-
if Version(numba.__version__) == Version("0.61") and is_platform_arm():
37-
pytest.skip(f"Segfaults on ARM platforms with numba {numba.__version__}")
3840
df = DataFrame(
3941
1,
4042
index=Index(["a", "b"], dtype=pd.StringDtype(na_value=np.nan)),

0 commit comments

Comments
 (0)