Skip to content

Commit a97bddd

Browse files
authored
CI/TST #34131 fixed test_floordiv_axis0_numexpr_path (#34537)
1 parent 07cef29 commit a97bddd

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pandas/tests/frame/test_arithmetic.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from pandas import DataFrame, MultiIndex, Series
1212
import pandas._testing as tm
1313
import pandas.core.common as com
14+
from pandas.core.computation.expressions import _MIN_ELEMENTS, _NUMEXPR_INSTALLED
1415
from pandas.tests.frame.common import _check_mixed_float, _check_mixed_int
1516

1617
# -------------------------------------------------------------------
@@ -374,13 +375,13 @@ def test_floordiv_axis0(self):
374375
result2 = df.floordiv(ser.values, axis=0)
375376
tm.assert_frame_equal(result2, expected)
376377

377-
@pytest.mark.slow
378+
@pytest.mark.skipif(not _NUMEXPR_INSTALLED, reason="numexpr not installed")
378379
@pytest.mark.parametrize("opname", ["floordiv", "pow"])
379380
def test_floordiv_axis0_numexpr_path(self, opname):
380381
# case that goes through numexpr and has to fall back to masked_arith_op
381382
op = getattr(operator, opname)
382383

383-
arr = np.arange(10 ** 6).reshape(100, -1)
384+
arr = np.arange(_MIN_ELEMENTS + 100).reshape(_MIN_ELEMENTS // 100 + 1, -1) * 100
384385
df = pd.DataFrame(arr)
385386
df["C"] = 1.0
386387

0 commit comments

Comments
 (0)