@@ -60,11 +60,13 @@ def setUp(self):
60
60
self .dta14_113 = os .path .join (self .dirpath , 'stata5_113.dta' )
61
61
self .dta14_114 = os .path .join (self .dirpath , 'stata5_114.dta' )
62
62
self .dta14_115 = os .path .join (self .dirpath , 'stata5_115.dta' )
63
+ self .dta14_117 = os .path .join (self .dirpath , 'stata5_117.dta' )
63
64
64
65
self .csv15 = os .path .join (self .dirpath , 'stata6.csv' )
65
66
self .dta15_113 = os .path .join (self .dirpath , 'stata6_113.dta' )
66
67
self .dta15_114 = os .path .join (self .dirpath , 'stata6_114.dta' )
67
68
self .dta15_115 = os .path .join (self .dirpath , 'stata6_115.dta' )
69
+ self .dta15_117 = os .path .join (self .dirpath , 'stata6_117.dta' )
68
70
69
71
def read_dta (self , file ):
70
72
return read_stata (file , convert_dates = True )
@@ -281,6 +283,11 @@ def test_encoding(self):
281
283
self .assertEqual (result , expected )
282
284
self .assertIsInstance (result , unicode )
283
285
286
+ with tm .ensure_clean () as path :
287
+ encoded .to_stata (path ,encoding = 'latin-1' , write_index = False )
288
+ reread_encoded = read_stata (path , encoding = 'latin-1' )
289
+ tm .assert_frame_equal (encoded , reread_encoded )
290
+
284
291
def test_read_write_dta11 (self ):
285
292
original = DataFrame ([(1 , 2 , 3 , 4 )],
286
293
columns = ['good' , compat .u ('b\u00E4 d' ), '8number' , 'astringwithmorethan32characters______' ])
@@ -354,9 +361,12 @@ def test_read_write_reread_dta14(self):
354
361
parsed_114 .index .name = 'index'
355
362
parsed_115 = self .read_dta (self .dta14_115 )
356
363
parsed_115 .index .name = 'index'
364
+ parsed_117 = self .read_dta (self .dta14_117 )
365
+ parsed_117 .index .name = 'index'
357
366
358
367
tm .assert_frame_equal (parsed_114 , parsed_113 )
359
368
tm .assert_frame_equal (parsed_114 , parsed_115 )
369
+ tm .assert_frame_equal (parsed_114 , parsed_117 )
360
370
361
371
with tm .ensure_clean () as path :
362
372
parsed_114 .to_stata (path , {'date_td' : 'td' })
@@ -375,10 +385,12 @@ def test_read_write_reread_dta15(self):
375
385
parsed_113 = self .read_dta (self .dta15_113 )
376
386
parsed_114 = self .read_dta (self .dta15_114 )
377
387
parsed_115 = self .read_dta (self .dta15_115 )
388
+ parsed_117 = self .read_dta (self .dta15_117 )
378
389
379
390
tm .assert_frame_equal (expected , parsed_114 )
380
391
tm .assert_frame_equal (parsed_113 , parsed_114 )
381
392
tm .assert_frame_equal (parsed_114 , parsed_115 )
393
+ tm .assert_frame_equal (parsed_114 , parsed_117 )
382
394
383
395
def test_timestamp_and_label (self ):
384
396
original = DataFrame ([(1 ,)], columns = ['var' ])
0 commit comments