@@ -429,15 +429,20 @@ def test_read_nrows(self):
429
429
430
430
def test_read_nrows_large (self ):
431
431
# GH-7626 - Read only nrows of data in for large inputs (>262144b)
432
- header_narrow = '\t ' .join (['COL_HEADER_' + str (i ) for i in range (10 )]) + '\n '
433
- data_narrow = '\t ' .join (['somedatasomedatasomedata1' for i in range (10 )])+ '\n '
434
- header_wide = '\t ' .join (['COL_HEADER_' + str (i ) for i in range (15 )]) + '\n '
435
- data_wide = '\t ' .join (['somedatasomedatasomedata2' for i in range (15 )])+ '\n '
436
- test_input = header_narrow + data_narrow * 1050 + header_wide + data_wide * 2
432
+ header_narrow = '\t ' .join (['COL_HEADER_' + str (i )
433
+ for i in range (10 )]) + '\n '
434
+ data_narrow = '\t ' .join (['somedatasomedatasomedata1'
435
+ for i in range (10 )]) + '\n '
436
+ header_wide = '\t ' .join (['COL_HEADER_' + str (i )
437
+ for i in range (15 )]) + '\n '
438
+ data_wide = '\t ' .join (['somedatasomedatasomedata2'
439
+ for i in range (15 )]) + '\n '
440
+ test_input = (header_narrow + data_narrow * 1050 +
441
+ header_wide + data_wide * 2 )
437
442
438
443
df = self .read_table (StringIO (test_input ), nrows = 1010 )
439
444
440
- self .assertTrue (df .size == 1010 * 10 )
445
+ self .assertTrue (df .size == 1010 * 10 )
441
446
442
447
def test_read_chunksize (self ):
443
448
reader = self .read_csv (StringIO (self .data1 ), index_col = 0 , chunksize = 2 )
0 commit comments