Skip to content

Commit c184ea5

Browse files
extension fix
1 parent 889f3d9 commit c184ea5

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

pandas/tests/io/formats/test_to_csv.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)