20
20
import pandas .util .testing as tm
21
21
import pandas
22
22
import pandas as pd
23
- from pandas .core .config import set_option ,get_option
23
+ from pandas .core .config import (set_option , get_option ,
24
+ reset_option )
24
25
25
26
_frame = DataFrame (tm .getSeriesData ())
26
27
@@ -404,6 +405,7 @@ def test_wide_repr(self):
404
405
set_option ('test.interactive' , True )
405
406
col = lambda l , k : [tm .rands (k ) for _ in xrange (l )]
406
407
df = DataFrame ([col (20 , 25 ) for _ in range (10 )])
408
+ set_option ('print.expand_frame_repr' , False )
407
409
rep_str = repr (df )
408
410
set_option ('print.expand_frame_repr' , True )
409
411
wide_repr = repr (df )
@@ -413,7 +415,7 @@ def test_wide_repr(self):
413
415
wider_repr = repr (df )
414
416
self .assert_ (len (wider_repr ) < len (wide_repr ))
415
417
416
- set_option ('print.expand_frame_repr' , False )
418
+ reset_option ('print.expand_frame_repr' )
417
419
set_option ('test.interactive' , False )
418
420
set_option ('print.line_width' , 80 )
419
421
@@ -422,6 +424,8 @@ def test_wide_repr_named(self):
422
424
col = lambda l , k : [tm .rands (k ) for _ in xrange (l )]
423
425
df = DataFrame ([col (20 , 25 ) for _ in range (10 )])
424
426
df .index .name = 'DataFrame Index'
427
+ set_option ('print.expand_frame_repr' , False )
428
+
425
429
rep_str = repr (df )
426
430
set_option ('print.expand_frame_repr' , True )
427
431
wide_repr = repr (df )
@@ -434,7 +438,7 @@ def test_wide_repr_named(self):
434
438
for line in wide_repr .splitlines ()[1 ::13 ]:
435
439
self .assert_ ('DataFrame Index' in line )
436
440
437
- set_option ('print.expand_frame_repr' , False )
441
+ reset_option ('print.expand_frame_repr' )
438
442
set_option ('test.interactive' , False )
439
443
set_option ('print.line_width' , 80 )
440
444
@@ -446,6 +450,7 @@ def test_wide_repr_multiindex(self):
446
450
df = DataFrame ([col (20 , 25 ) for _ in range (10 )],
447
451
index = midx )
448
452
df .index .names = ['Level 0' , 'Level 1' ]
453
+ set_option ('print.expand_frame_repr' , False )
449
454
rep_str = repr (df )
450
455
set_option ('print.expand_frame_repr' , True )
451
456
wide_repr = repr (df )
@@ -458,7 +463,7 @@ def test_wide_repr_multiindex(self):
458
463
for line in wide_repr .splitlines ()[1 ::13 ]:
459
464
self .assert_ ('Level 0 Level 1' in line )
460
465
461
- set_option ('print.expand_frame_repr' , False )
466
+ reset_option ('print.expand_frame_repr' )
462
467
set_option ('test.interactive' , False )
463
468
set_option ('print.line_width' , 80 )
464
469
@@ -472,6 +477,7 @@ def test_wide_repr_multiindex_cols(self):
472
477
df = DataFrame ([col (20 , 25 ) for _ in range (10 )],
473
478
index = midx , columns = mcols )
474
479
df .index .names = ['Level 0' , 'Level 1' ]
480
+ set_option ('print.expand_frame_repr' , False )
475
481
rep_str = repr (df )
476
482
set_option ('print.expand_frame_repr' , True )
477
483
wide_repr = repr (df )
@@ -483,14 +489,15 @@ def test_wide_repr_multiindex_cols(self):
483
489
484
490
self .assert_ (len (wide_repr .splitlines ()) == 14 * 10 - 1 )
485
491
486
- set_option ('print.expand_frame_repr' , False )
492
+ reset_option ('print.expand_frame_repr' )
487
493
set_option ('test.interactive' , False )
488
494
set_option ('print.line_width' , 80 )
489
495
490
496
def test_wide_repr_unicode (self ):
491
497
set_option ('test.interactive' , True )
492
498
col = lambda l , k : [tm .randu (k ) for _ in xrange (l )]
493
499
df = DataFrame ([col (20 , 25 ) for _ in range (10 )])
500
+ set_option ('print.expand_frame_repr' , False )
494
501
rep_str = repr (df )
495
502
set_option ('print.expand_frame_repr' , True )
496
503
wide_repr = repr (df )
@@ -500,7 +507,7 @@ def test_wide_repr_unicode(self):
500
507
wider_repr = repr (df )
501
508
self .assert_ (len (wider_repr ) < len (wide_repr ))
502
509
503
- set_option ('print.expand_frame_repr' , False )
510
+ reset_option ('print.expand_frame_repr' )
504
511
set_option ('test.interactive' , False )
505
512
set_option ('print.line_width' , 80 )
506
513
0 commit comments