diff --git a/tests/test_tablib.py b/tests/test_tablib.py index 77d77d3a..908a005b 100755 --- a/tests/test_tablib.py +++ b/tests/test_tablib.py @@ -195,6 +195,19 @@ def test_header_slicing(self): self.assertEqual(self.founders['gpa'], [self.john[2], self.george[2], self.tom[2]]) + def test_get(self): + """Verify getting rows by index""" + + self.assertEqual(self.founders.get(0), self.john) + self.assertEqual(self.founders.get(1), self.george) + self.assertEqual(self.founders.get(2), self.tom) + + self.assertEqual(self.founders.get(-1), self.tom) + self.assertEqual(self.founders.get(-2), self.george) + self.assertEqual(self.founders.get(-3), self.john) + + self.assertRaises(self.founders.get(3), IndexError) + def test_get_col(self): """Verify getting columns by index"""