@@ -104,17 +104,21 @@ def _eval_single_bin(lhs, cmp1, rhs, engine):
104104 ids = ["DataFrame" , "Series" , "SeriesNaN" , "DataFrameNaN" , "float" ],
105105)
106106def 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
120124rhs = lhs
0 commit comments