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