12
12
import pandas ._testing as tm
13
13
from pandas .util .version import Version
14
14
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
+ )
16
24
17
25
18
26
@pytest .fixture (params = [0 , 1 ])
@@ -21,9 +29,6 @@ def apply_axis(request):
21
29
22
30
23
31
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__ } " )
27
32
func = lambda x : x
28
33
result = float_frame .apply (func , engine = "numba" , axis = apply_axis )
29
34
expected = float_frame .apply (func , engine = "python" , axis = apply_axis )
@@ -32,9 +37,6 @@ def test_numba_vs_python_noop(float_frame, apply_axis):
32
37
33
38
def test_numba_vs_python_string_index ():
34
39
# 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__ } " )
38
40
df = DataFrame (
39
41
1 ,
40
42
index = Index (["a" , "b" ], dtype = pd .StringDtype (na_value = np .nan )),
0 commit comments