@@ -203,16 +203,6 @@ def test_cross_engine_fp_pa(df_cross_compat, pa, fp):
203
203
result = read_parquet (path , engine = pa , columns = ['a' , 'd' ])
204
204
tm .assert_frame_equal (result , df [['a' , 'd' ]])
205
205
206
- def test_s3_roundtrip (df_compat , s3_resource , engine ):
207
- # GH #19134
208
- if engine == 'pyarrow' :
209
- df_compat .to_parquet ('s3://pandas-test/test.parquet' ,
210
- engine , compression = None )
211
-
212
- expected = df_compat
213
- actual = pd .read_parquet ('s3://pandas-test/test.parquet' , engine )
214
-
215
- tm .assert_frame_equal (expected , actual )
216
206
217
207
class Base (object ):
218
208
@@ -436,6 +426,16 @@ def test_categorical_unsupported(self, pa_lt_070):
436
426
df = pd .DataFrame ({'a' : pd .Categorical (list ('abc' ))})
437
427
self .check_error_on_write (df , pa , NotImplementedError )
438
428
429
+ def test_s3_roundtrip (self , df_compat , s3_resource , pa ):
430
+ # GH #19134
431
+ df_compat .to_parquet ('s3://pandas-test/test.parquet' ,
432
+ engine = pa , compression = None )
433
+
434
+ expected = df_compat
435
+ actual = read_parquet ('s3://pandas-test/test.parquet' , engine = pa )
436
+
437
+ tm .assert_frame_equal (expected , actual )
438
+
439
439
440
440
class TestParquetFastParquet (Base ):
441
441
0 commit comments