Added test for slicing data elements

1 files changed, 15 insertions, 1 deletions

diff --git a/test_tablib.py b/test_tablib.py
index 8dcbb94..670a41b 100644
--- a/test_tablib.py
+++ b/test_tablib.py
@@ -50,11 +50,25 @@ class TablibTestCase(unittest.TestCase):
 	def test_header_slicing(self):
 		"""Verify slicing by headers"""
 
-		# Slice 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]])
 
+	def test_data_slicing(self):
+		"""Verify slicing by data"""
+
+		# Slice individual rows
+		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)
+
+		# Slice multiple rows
+		self.assertEqual(self.founders[:], [self.john, self.george])
+		self.assertEqual(self.founders[0:2], [self.john, self.george])
+		self.assertEqual(self.founders[2:], [])
+
 	# def test_adding_header with (self):