def test_remove_empty_final_cols(self):
ws = Worksheet()
ws.append(Row(['a', None, None]))
ws.append(Row(['b', None, 'd', None, '']))
self.assertEqual(len(ws), 2)
ws.remove_empty_final_cols()
self.assertEqual(len(ws), 2)
self.assertEqual(ws[0], Row(['a', None, None]))
self.assertEqual(ws[1], Row(['b', None, 'd']))
def test_remove_empty_final_rows(self):
ws = Worksheet()
ws.append(Row(['a', 'b']))
ws.append(Row([None, None]))
ws.append(Row(['c', 'd']))
ws.append(Row([None, None]))
ws.append(Row([None, None]))
self.assertEqual(len(ws), 5)
ws.remove_empty_final_rows()
self.assertEqual(len(ws), 3)
self.assertEqual(list(ws[0]), ['a', 'b'])
self.assertEqual(list(ws[1]), [None, None])
self.assertEqual(list(ws[2]), ['c', 'd'])