Skip to content

Commit 3cae0a2

Browse files
alysivjijreback
authored andcommitted
Refactor to use context managers (#21139)
1 parent 1c3418b commit 3cae0a2

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

pandas/tests/io/test_packers.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -128,9 +128,8 @@ def test_string_io(self):
128128
with ensure_clean(self.path) as p:
129129

130130
s = df.to_msgpack()
131-
fh = open(p, 'wb')
132-
fh.write(s)
133-
fh.close()
131+
with open(p, 'wb') as fh:
132+
fh.write(s)
134133
result = read_msgpack(p)
135134
tm.assert_frame_equal(result, df)
136135

pandas/tests/series/test_io.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -75,9 +75,8 @@ def test_from_csv(self):
7575
series_h = self.read_csv(path, header=0)
7676
assert series_h.name == "series"
7777

78-
outfile = open(path, "w")
79-
outfile.write("1998-01-01|1.0\n1999-01-01|2.0")
80-
outfile.close()
78+
with open(path, "w") as outfile:
79+
outfile.write("1998-01-01|1.0\n1999-01-01|2.0")
8180

8281
series = self.read_csv(path, sep="|")
8382
check_series = Series({datetime(1998, 1, 1): 1.0,

0 commit comments

Comments
 (0)