def test_start(self): f = ColumnTrimmer() f.next = m = Mock() f.rdsheet = 'junk' f.pending_rdsheet = 'junk' f.ranges = 'junk' f.max_nonjunk = 'junk' f.max = 'junk' f.start() compare(f.rdsheet,None) compare(f.pending_rdsheet,None) compare(f.ranges,[]) compare(f.max_nonjunk,0) compare(f.max,0) compare(m.method_calls,[ ('start', (), {}) ])
def test_use_write_sheet_name_in_logging(self,h): r = TestReader( ('Sheet1',[['X',' ']]), ) book = tuple(r.get_workbooks())[0][0] # fire methods on filter f = ColumnTrimmer() f.next = c = Mock() f.start() f.workbook(book,'new.xls') f.sheet(book.sheet_by_index(0),'new') f.row(0,0) f.cell(0,0,0,0) f.cell(0,1,0,1) f.finish() compare(c.method_calls,[ ('start', (), {}), ('workbook', (C('xlutils.tests.fixtures.DummyBook'), 'new.xls'),{}), ('sheet', (C('xlrd.sheet.Sheet',name='Sheet1',strict=False), u'new'),{}), ('row', (0, 0),{}), ('cell', (0, 0, 0, 0),{}), ('finish', (),{}) ]) h.check(( 'xlutils.filter', 'DEBUG', "Number of columns trimmed from 2 to 1 for sheet 'new'" ))
def test_set_rdsheet_trim(self): r = TestReader( ('Sheet1',[['X',' ']]), ('Sheet2',[['X','X']]), ) book = tuple(r.get_workbooks())[0][0] # fire methods on filter f = ColumnTrimmer() f.next = c = Mock() f.start() f.workbook(book,'new.xls') f.sheet(book.sheet_by_index(0),'new') f.row(0,0) f.cell(0,0,0,0) f.cell(0,1,0,1) f.set_rdsheet(book.sheet_by_index(1)) f.cell(0,0,1,0) f.cell(0,1,1,1) f.finish() compare(c.method_calls,[ ('start', (), {}), ('workbook', (C('xlutils.tests.fixtures.DummyBook'), 'new.xls'),{}), ('sheet', (C('xlrd.sheet.Sheet',name='Sheet1',strict=False), u'new'),{}), ('row', (0, 0),{}), ('cell', (0, 0, 0, 0),{}), ('cell', (0, 1, 0, 1),{}), ('set_rdsheet', (C('xlrd.sheet.Sheet',name='Sheet2',strict=False),),{}), ('cell', (0, 0, 1, 0),{}), ('cell', (0, 1, 1, 1),{}), ('finish', (), {}) ])