File tree 1 file changed +7
-9
lines changed 1 file changed +7
-9
lines changed Original file line number Diff line number Diff line change @@ -137,19 +137,17 @@ def test_encoding_options(datapath):
137
137
138
138
139
139
def test_encoding_infer (datapath ):
140
- from pandas .io .sas .sas7bdat import SAS7BDATReader
141
-
142
140
fname = datapath ("io" , "sas" , "data" , "test1.sas7bdat" )
143
141
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 ()
147
146
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 ()
152
149
150
+ # check: reader reads correct information
153
151
tm .assert_frame_equal (df1 , df2 )
154
152
155
153
You can’t perform that action at this time.
0 commit comments