Skip to content

Commit b73fae0

Browse files
author
OlivierLuG
committed
pandas-dev#34131 explicit check of _USE_NUNEXPR
1 parent 063b097 commit b73fae0

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pandas/tests/frame/test_arithmetic.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import pandas as pd
1111
from pandas import DataFrame, MultiIndex, Series
1212
import pandas._testing as tm
13+
from pandas.core.computation.expressions import _USE_NUMEXPR, _MIN_ELEMENTS
1314
import pandas.core.common as com
1415
from pandas.tests.frame.common import _check_mixed_float, _check_mixed_int
1516

@@ -379,8 +380,9 @@ def test_floordiv_axis0_numexpr_path(self, opname):
379380
# case that goes through numexpr and has to fall back to masked_arith_op
380381
op = getattr(operator, opname)
381382

382-
arr = np.arange(10 ** 4).reshape(100, -1) * 100
383+
arr = np.arange(_MIN_ELEMENTS + 100).reshape(_MIN_ELEMENTS // 100 + 1, -1) * 100
383384
df = pd.DataFrame(arr)
385+
assert _USE_NUMEXPR
384386
df["C"] = 1.0
385387

386388
ser = df[0]

0 commit comments

Comments
 (0)