diff --git a/test_tablib.py b/test_tablib.py index 09d05f6..a88a297 100644 --- a/test_tablib.py +++ b/test_tablib.py @@ -26,10 +26,12 @@ class TablibTestCase(unittest.TestCase): self.founders.append(self.george) self.founders.append(self.tom) + def tearDown(self): """teardown""" pass + def test_empty_append(self): """Verify append() correctly adds tuple with no headers""" new_row = (1, 2, 3) @@ -39,6 +41,7 @@ class TablibTestCase(unittest.TestCase): self.assertTrue(data.width == len(new_row)) self.assertTrue(data[0] == new_row) + def test_empty_append_with_headers(self): """Verify append() correctly detects mismatch of number of headers and data @@ -48,6 +51,7 @@ class TablibTestCase(unittest.TestCase): self.assertRaises(tablib.InvalidDimensions, data.append, new_row) + def test_add_column(self): """Verify adding column works with/without headers""" @@ -68,6 +72,7 @@ class TablibTestCase(unittest.TestCase): self.assertEquals(data[new_col[0]], new_col[1:]) + def test_add_column_no_data_no_headers(self): """Verify adding new column with no headers""" @@ -79,6 +84,7 @@ class TablibTestCase(unittest.TestCase): self.assertEquals(data.width, 1) self.assertEquals(data.height, len(new_col)) + def test_add_column_no_data_with_headers(self): """Verify adding new column with headers""" @@ -94,9 +100,11 @@ class TablibTestCase(unittest.TestCase): self.assertRaises(tablib.InvalidDimensions, data.append, col=new_col) + def tuple_check(self): data.append(col=(1, 2, 3)) + def test_header_slicing(self): """Verify slicing by headers""" @@ -107,6 +115,7 @@ class TablibTestCase(unittest.TestCase): self.assertEqual(self.founders['gpa'], [self.john[2], self.george[2], self.tom[2]]) + def test_data_slicing(self): """Verify slicing by data""" @@ -123,6 +132,7 @@ class TablibTestCase(unittest.TestCase): self.assertEqual(self.founders[1:3], [self.george, self.tom]) self.assertEqual(self.founders[2:], [self.tom]) + def test_delete(self): """Verify deleting from dataset works""" @@ -145,6 +155,7 @@ class TablibTestCase(unittest.TestCase): # Delete from invalid index self.assertRaises(IndexError, self.founders.__delitem__, 3) + def test_csv_export(self): """Verify exporting dataset object as CSV"""