Skip to content

Commit 3a331eb

Browse files
committed
generate less data in lhs fixture
1 parent 0461452 commit 3a331eb

File tree

1 file changed

+15
-11
lines changed

1 file changed

+15
-11
lines changed

pandas/tests/computation/test_eval.py

Lines changed: 15 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -104,17 +104,21 @@ def _eval_single_bin(lhs, cmp1, rhs, engine):
104104
ids=["DataFrame", "Series", "SeriesNaN", "DataFrameNaN", "float"],
105105
)
106106
def lhs(request):
107-
nan_df1 = DataFrame(np.random.default_rng(2).standard_normal((10, 5)))
108-
nan_df1[nan_df1 > 0.5] = np.nan
109-
110-
opts = (
111-
DataFrame(np.random.default_rng(2).standard_normal((10, 5))),
112-
Series(np.random.default_rng(2).standard_normal(5)),
113-
Series([1, 2, np.nan, np.nan, 5]),
114-
nan_df1,
115-
np.random.default_rng(2).standard_normal(),
116-
)
117-
return opts[request.param]
107+
rng = np.random.default_rng(2)
108+
if request.param == 0:
109+
return DataFrame(rng.standard_normal((10, 5)))
110+
elif request.param == 1:
111+
return Series(rng.standard_normal(5))
112+
elif request.param == 2:
113+
return Series([1, 2, np.nan, np.nan, 5])
114+
elif request.param == 3:
115+
nan_df1 = DataFrame(rng.standard_normal((10, 5)))
116+
nan_df1[nan_df1 > 0.5] = np.nan
117+
return nan_df1
118+
elif request.param == 4:
119+
return rng.standard_normal()
120+
else:
121+
raise ValueError(f"{request.param}")
118122

119123

120124
rhs = lhs

0 commit comments

Comments
 (0)