1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import unittest
import xlwt
class TestByName(unittest.TestCase):
def setUp(self):
self.wb = xlwt.Workbook()
self.wb.add_sheet('Plan1')
self.wb.add_sheet('Plan2')
self.wb.add_sheet('Plan3')
self.wb.add_sheet('Plan4')
def test_sheet_index(self):
'Return sheet index by sheet name'
idx = self.wb.sheet_index('Plan3')
self.assertEqual(2, idx)
def test_get_by_name(self):
'Get sheet by name'
ws = self.wb.get_sheet('Plan2')
self.assertEqual('Plan2', ws.name)
def test_get_by_index(self):
'Get sheet by index'
ws = self.wb.get_sheet(1)
self.assertEqual('Plan2', ws.name)
def test_invalid_sheet_parameter(self):
'Raises exception when sheet is not string or integer'
self.assertRaises(Exception, self.wb.get_sheet, 1.1)
|
