diff options
| author | Luke Lee <durdenmisc@gmail.com> | 2010-09-13 21:26:15 -0500 |
|---|---|---|
| committer | Luke Lee <durdenmisc@gmail.com> | 2010-09-13 21:26:15 -0500 |
| commit | 4755020dd7f6e9e7451c3c6518de0db0ba3fc620 (patch) | |
| tree | 05ee3b96020710a344f8cc3094c0303f3721f9d8 | |
| parent | 5468dd7e679cd33ec6223ef451197d341ff2f32d (diff) | |
| download | tablib-4755020dd7f6e9e7451c3c6518de0db0ba3fc620.tar.gz | |
Added extra row to base data set
- Testing with 3 rows is a bit more interesting
| -rw-r--r-- | test_tablib.py | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/test_tablib.py b/test_tablib.py index 670a41b..544ca12 100644 --- a/test_tablib.py +++ b/test_tablib.py @@ -16,10 +16,12 @@ class TablibTestCase(unittest.TestCase): headers = ('first_name', 'last_name', 'gpa') self.john = ('John', 'Adams', 90) self.george = ('George', 'Washington', 67) + self.tom = ('Thomas', 'Jefferson', 50) self.founders = tablib.Dataset(headers=headers) self.founders.append(self.john) self.founders.append(self.george) + self.founders.append(self.tom) def tearDown(self): """teardown""" @@ -50,9 +52,12 @@ class TablibTestCase(unittest.TestCase): def test_header_slicing(self): """Verify slicing 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]]) + self.assertEqual(self.founders['first_name'], + [self.john[0], self.george[0], self.tom[0]]) + self.assertEqual(self.founders['last_name'], + [self.john[1], self.george[1], self.tom[1]]) + self.assertEqual(self.founders['gpa'], + [self.john[2], self.george[2], self.tom[2]]) def test_data_slicing(self): """Verify slicing by data""" @@ -61,13 +66,13 @@ class TablibTestCase(unittest.TestCase): 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) + self.assertEqual(self.founders[-1], self.tom) + self.assertEqual(self.founders[3:], []) # Slice multiple rows - self.assertEqual(self.founders[:], [self.john, self.george]) + self.assertEqual(self.founders[:], [self.john, self.george, self.tom]) self.assertEqual(self.founders[0:2], [self.john, self.george]) - self.assertEqual(self.founders[2:], []) + self.assertEqual(self.founders[2:], [self.tom]) # def test_adding_header with (self): |