Skip to content

Commit b4d7346

Browse files
committed
rewrite test: no type hinting + context manager call
1 parent f637661 commit b4d7346

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

pandas/tests/io/sas/test_sas7bdat.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -137,19 +137,17 @@ def test_encoding_options(datapath):
137137

138138

139139
def test_encoding_infer(datapath):
140-
from pandas.io.sas.sas7bdat import SAS7BDATReader
141-
142140
fname = datapath("io", "sas", "data", "test1.sas7bdat")
143141

144-
# check if inferred correctly
145-
df1_reader = pd.read_sas(fname, encoding="infer", iterator=True)
146-
assert df1_reader.inferred_encoding == "cp1252"
142+
with pd.read_sas(fname, encoding="infer", iterator=True) as df1_reader:
143+
# check: is encoding inferred correctly from file
144+
assert df1_reader.inferred_encoding == "cp1252"
145+
df1 = df1_reader.read()
147146

148-
# check if the reader reads correctly with encoding
149-
df1 = df1_reader.read()
150-
df2_reader: SAS7BDATReader = pd.read_sas(fname, encoding="cp1252", iterator=True)
151-
df2 = df2_reader.read()
147+
with pd.read_sas(fname, encoding="cp1252", iterator=True) as df2_reader:
148+
df2 = df2_reader.read()
152149

150+
# check: reader reads correct information
153151
tm.assert_frame_equal(df1, df2)
154152

155153

0 commit comments

Comments
 (0)