@@ -229,21 +229,26 @@ def check_round_trip(self, df, engine, expected=None, path=None,
229
229
with tm .ensure_clean () as path :
230
230
df .to_parquet (path , engine , ** write_kwargs )
231
231
actual = read_parquet (path , engine , ** read_kwargs )
232
- tm .assert_frame_equal (expected , actual , check_names = check_names )
232
+ tm .assert_frame_equal (expected , actual ,
233
+ check_names = check_names )
233
234
234
235
# repeat
235
236
df .to_parquet (path , engine , ** write_kwargs )
236
237
actual = read_parquet (path , engine , ** read_kwargs )
237
- tm .assert_frame_equal (expected , actual , check_names = check_names )
238
+ tm .assert_frame_equal (expected , actual ,
239
+ check_names = check_names )
238
240
else :
239
241
df .to_parquet (path , engine , ** write_kwargs )
240
242
actual = read_parquet (path , engine , ** read_kwargs )
241
- tm .assert_frame_equal (expected , actual , check_names = check_names )
243
+ tm .assert_frame_equal (expected , actual ,
244
+ check_names = check_names )
242
245
243
246
# repeat
244
247
df .to_parquet (path , engine , ** write_kwargs )
245
248
actual = read_parquet (path , engine , ** read_kwargs )
246
- tm .assert_frame_equal (expected , actual , check_names = check_names )
249
+ tm .assert_frame_equal (expected , actual ,
250
+ check_names = check_names )
251
+
247
252
248
253
class TestBasic (Base ):
249
254
@@ -433,7 +438,8 @@ def test_categorical_unsupported(self, pa_lt_070):
433
438
434
439
def test_s3_roundtrip (self , df_compat , s3_resource , pa ):
435
440
# GH #19134
436
- self .check_round_trip (df_compat , pa , path = 's3://pandas-test/pyarrow.parquet' )
441
+ self .check_round_trip (df_compat , pa ,
442
+ path = 's3://pandas-test/pyarrow.parquet' )
437
443
438
444
439
445
class TestParquetFastParquet (Base ):
@@ -497,4 +503,5 @@ def test_filter_row_groups(self, fp):
497
503
def test_s3_roundtrip (self , df_compat , s3_resource , fp ):
498
504
# GH #19134
499
505
with pytest .raises (NotImplementedError ):
500
- self .check_round_trip (df_compat , fp , path = 's3://pandas-test/fastparquet.parquet' )
506
+ self .check_round_trip (df_compat , fp ,
507
+ path = 's3://pandas-test/fastparquet.parquet' )
0 commit comments