@@ -549,16 +549,12 @@ def test_to_csv_compression(
549549 # see gh-15008
550550 compression = compression_only
551551
552- # We'll complete file extension subsequently.
553- filename = "test."
554- filename += compression_to_extension [compression ]
555-
556552 df = DataFrame ({"A" : [1 ]})
557553
558554 to_compression = "infer" if to_infer else compression
559555 read_compression = "infer" if read_infer else compression
560556
561- path = str (temp_file )
557+ path = str (temp_file ) + "." + compression_to_extension [ compression ]
562558 df .to_csv (path , compression = to_compression )
563559 result = pd .read_csv (path , index_col = 0 , compression = read_compression )
564560 tm .assert_frame_equal (result , df )
@@ -567,14 +563,12 @@ def test_to_csv_compression_dict(self, compression_only, temp_file):
567563 # GH 26023
568564 method = compression_only
569565 df = DataFrame ({"ABC" : [1 ]})
570- filename = "to_csv_compress_as_dict."
571566 extension = {
572567 "gzip" : "gz" ,
573568 "zstd" : "zst" ,
574569 }.get (method , method )
575- filename += extension
576570
577- path = str (temp_file )
571+ path = str (temp_file ) + "." + extension
578572 df .to_csv (path , compression = {"method" : method })
579573 read_df = pd .read_csv (path , index_col = 0 )
580574 tm .assert_frame_equal (read_df , df )
@@ -595,7 +589,7 @@ def test_to_csv_zip_arguments(self, compression, archive_name, temp_file):
595589 # GH 26023
596590 df = DataFrame ({"ABC" : [1 ]})
597591
598- path = str (temp_file )
592+ path = str (temp_file ) + ".zip"
599593 df .to_csv (
600594 path , compression = {"method" : compression , "archive_name" : archive_name }
601595 )
0 commit comments