def test_load(self):
obj = XLSObject(data_path("data.xls"), encoding="latin1")
self.assertEqual(["id", "name", "amount"], obj.fields.names())
rows = list(obj.rows())
self.assertEqual(4, len(rows))
self.assertEqual(4, len(obj))
self.assertSequenceEqual([1, "Adam", 10], rows[0])
def test_skip(self):
obj = XLSObject(data_path("data.xls"), "numbers", skip_rows=2)
self.assertEqual(["number", "name"], obj.fields.names())
rows = list(obj.rows())
self.assertEqual(10, len(rows))
self.assertEqual(10, len(obj))
self.assertSequenceEqual([1, "one"], rows[0])
def test_skip(self):
obj = XLSObject(data_path("data.xls"), FieldList("number", "name"),
skip_rows=2)
self.assertEqual(["number", "name"], obj.fields.names())
rows = list(obj.rows())
self.assertEqual(2, len(obj))
self.assertEqual(2, len(rows))
self.assertSequenceEqual([3.0, "Cecil"], rows[0])
def test_skip_too_much(self):
with self.assertRaises(ArgumentError):
obj = XLSObject(data_path("data.xls"), FieldList("numbers"), skip_rows=20)
