@@ -74,12 +74,12 @@ def polars_series_to_narwhals(ser: pl.Series) -> nw.Series:
74
74
75
75
76
76
@pytest .fixture (params = params_frames , scope = "function" )
77
- def df (request : pytest .FixtureRequest ) -> DataFrameLike :
77
+ def df_f (request : pytest .FixtureRequest ) -> DataFrameLike :
78
78
return request .param (DATA )
79
79
80
80
81
81
@pytest .fixture (params = params_frames , scope = "function" )
82
- def small_df (request : pytest .FixtureRequest ) -> DataFrameLike :
82
+ def small_df_f (request : pytest .FixtureRequest ) -> DataFrameLike :
83
83
return request .param ({"x" : [1 , 2 ], "y" : [3 , 4 ]})
84
84
85
85
@@ -154,13 +154,13 @@ def test_serialize_dtype(
154
154
assert serialize_dtype (ser )["type" ] == res_type
155
155
156
156
157
- def test_serialize_frame (df : DataFrameLike ):
157
+ def test_serialize_frame (df_f : DataFrameLike ):
158
158
# TODO: pandas converts datetime entries to int, but Polars
159
159
# preserves the datetime object.
160
- if isinstance (df , pl .DataFrame ):
160
+ if isinstance (df_f , pl .DataFrame ):
161
161
pytest .xfail ()
162
162
163
- res = serialize_frame (df )
163
+ res = serialize_frame (df_f )
164
164
assert res == {
165
165
"columns" : ["num" , "chr" , "cat" , "dt" , "struct" , "arr" , "object" ],
166
166
"index" : [0 , 1 ],
@@ -180,42 +180,42 @@ def test_serialize_frame(df: DataFrameLike):
180
180
}
181
181
182
182
183
- def test_subset_frame (df : DataFrameLike ):
183
+ def test_subset_frame (df_f : DataFrameLike ):
184
184
# TODO: this assumes subset_frame doesn't reset index
185
- res = subset_frame (df , rows = [1 ], cols = ["chr" , "num" ])
185
+ res = subset_frame (df_f , rows = [1 ], cols = ["chr" , "num" ])
186
186
dst = {"chr" : ["b" ], "num" : [2 ]}
187
187
188
188
assert_frame_equal2 (res , dst )
189
189
190
190
191
- def test_get_frame_cell (df : DataFrameLike ):
192
- assert get_frame_cell (df , 1 , 1 ) == "b"
191
+ def test_get_frame_cell (df_f : DataFrameLike ):
192
+ assert get_frame_cell (df_f , 1 , 1 ) == "b"
193
193
194
194
195
- def test_copy_frame (df : DataFrameLike ):
196
- new_df = copy_frame (df )
195
+ def test_copy_frame (df_f : DataFrameLike ):
196
+ new_df = copy_frame (df_f )
197
197
198
- assert new_df is not df
198
+ assert new_df is not df_f
199
199
200
200
201
- def test_subset_frame_rows_single (small_df : DataFrameLike ):
202
- res = subset_frame (small_df , rows = [1 ])
201
+ def test_subset_frame_rows_single (small_df_f : DataFrameLike ):
202
+ res = subset_frame (small_df_f , rows = [1 ])
203
203
204
204
assert_frame_equal2 (
205
205
res ,
206
206
{"x" : [2 ], "y" : [4 ]},
207
207
)
208
208
209
209
210
- def test_subset_frame_cols_single (small_df : DataFrameLike ):
210
+ def test_subset_frame_cols_single (small_df_f : DataFrameLike ):
211
211
# TODO: include test of polars
212
- res = subset_frame (small_df , cols = ["y" ])
212
+ res = subset_frame (small_df_f , cols = ["y" ])
213
213
214
214
assert_frame_equal2 (
215
215
res ,
216
216
{"y" : [3 , 4 ]},
217
217
)
218
218
219
219
220
- def test_shape (small_df : DataFrameLike ):
221
- assert frame_shape (small_df ) == (2 , 2 )
220
+ def test_shape (small_df_f : DataFrameLike ):
221
+ assert frame_shape (small_df_f ) == (2 , 2 )
0 commit comments