diff options
Diffstat (limited to 'test_tablib.py')
| -rw-r--r-- | test_tablib.py | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/test_tablib.py b/test_tablib.py index 457d74b..cea6d24 100644 --- a/test_tablib.py +++ b/test_tablib.py @@ -13,12 +13,27 @@ class TablibTestCase(unittest.TestCase): pass def test_empty_append(self): + data = tablib.Dataset() + new_row = (1,2,3) data.append(new_row) - assert data.width == len(new_row) + + self.assertTrue(data.width == len(new_row)) + + + def test_empty_append_with_headers(self): + + data = tablib.Dataset() + + data.headers = ['first', 'second'] + new_row = (1,2,3,4) - + self.assertRaises(tablib.InvalidDimensions, data.append, new_row) + + # def test_adding_header with (self): + + if __name__ == '__main__': |