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', (), {}) ])