From 5468dd7e679cd33ec6223ef451197d341ff2f32d Mon Sep 17 00:00:00 2001 From: Luke Lee Date: Mon, 13 Sep 2010 21:23:20 -0500 Subject: [PATCH] Added test for slicing data elements --- test_tablib.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/test_tablib.py b/test_tablib.py index 8dcbb94..670a41b 100644 --- a/test_tablib.py +++ b/test_tablib.py @@ -50,11 +50,25 @@ class TablibTestCase(unittest.TestCase): def test_header_slicing(self): """Verify slicing by headers""" - # Slice by headers self.assertEqual(self.founders['first_name'], [self.john[0], self.george[0]]) self.assertEqual(self.founders['last_name'], [self.john[1], self.george[1]]) self.assertEqual(self.founders['gpa'], [self.john[2], self.george[2]]) + def test_data_slicing(self): + """Verify slicing by data""" + + # Slice individual rows + self.assertEqual(self.founders[0], self.john) + self.assertEqual(self.founders[:1], [self.john]) + self.assertEqual(self.founders[1:2], [self.george]) + self.assertEqual(self.founders[2:], []) + self.assertEqual(self.founders[-1], self.george) + + # Slice multiple rows + self.assertEqual(self.founders[:], [self.john, self.george]) + self.assertEqual(self.founders[0:2], [self.john, self.george]) + self.assertEqual(self.founders[2:], []) + # def test_adding_header with (self):