@@ -549,16 +549,12 @@ def test_to_csv_compression(
549
549
# see gh-15008
550
550
compression = compression_only
551
551
552
- # We'll complete file extension subsequently.
553
- filename = "test."
554
- filename += compression_to_extension [compression ]
555
-
556
552
df = DataFrame ({"A" : [1 ]})
557
553
558
554
to_compression = "infer" if to_infer else compression
559
555
read_compression = "infer" if read_infer else compression
560
556
561
- path = str (temp_file )
557
+ path = str (temp_file ) + "." + compression_to_extension [ compression ]
562
558
df .to_csv (path , compression = to_compression )
563
559
result = pd .read_csv (path , index_col = 0 , compression = read_compression )
564
560
tm .assert_frame_equal (result , df )
@@ -567,14 +563,12 @@ def test_to_csv_compression_dict(self, compression_only, temp_file):
567
563
# GH 26023
568
564
method = compression_only
569
565
df = DataFrame ({"ABC" : [1 ]})
570
- filename = "to_csv_compress_as_dict."
571
566
extension = {
572
567
"gzip" : "gz" ,
573
568
"zstd" : "zst" ,
574
569
}.get (method , method )
575
- filename += extension
576
570
577
- path = str (temp_file )
571
+ path = str (temp_file ) + "." + extension
578
572
df .to_csv (path , compression = {"method" : method })
579
573
read_df = pd .read_csv (path , index_col = 0 )
580
574
tm .assert_frame_equal (read_df , df )
@@ -595,7 +589,7 @@ def test_to_csv_zip_arguments(self, compression, archive_name, temp_file):
595
589
# GH 26023
596
590
df = DataFrame ({"ABC" : [1 ]})
597
591
598
- path = str (temp_file )
592
+ path = str (temp_file ) + ".zip"
599
593
df .to_csv (
600
594
path , compression = {"method" : compression , "archive_name" : archive_name }
601
595
)
0 commit comments