diff --git a/test_tablib.py b/test_tablib.py index 0fffdc9..b4b4fbc 100644 --- a/test_tablib.py +++ b/test_tablib.py @@ -130,10 +130,18 @@ class TablibTestCase(unittest.TestCase): del self.founders[0] self.assertEqual(self.founders[:], [self.george, self.tom]) + # Verify dimensions, width should NOT change + self.assertEqual(self.founders.height, 2) + self.assertEqual(self.founders.width, 3) + # Delete from back of object del self.founders[1] self.assertEqual(self.founders[:], [self.george]) + # Verify dimensions, width should NOT change + self.assertEqual(self.founders.height, 1) + self.assertEqual(self.founders.width, 3) + # Delete from invalid index self.assertRaises(IndexError, self.founders.__delitem__, 3)