1212import pandas ._testing as tm
1313from 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
2331def 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
3338def 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