Created get_col method with tests and tutorial.rst update

Useful when you have multiple columns with the same header

1 files changed, 16 insertions, 0 deletions

diff --git a/test_tablib.py b/test_tablib.py
index 5fdf65a..48990a7 100755
--- a/test_tablib.py
+++ b/test_tablib.py
@@ -113,6 +113,22 @@ class TablibTestCase(unittest.TestCase):
             [self.john[2], self.george[2], self.tom[2]])
 
 
+    def test_get_col(self):
+        """Verify getting columns by index"""
+
+        self.assertEqual(
+            self.founders.get_col(self.headers.index('first_name')),
+            [self.john[0], self.george[0], self.tom[0]])
+
+        self.assertEqual(
+            self.founders.get_col(self.headers.index('last_name')),
+            [self.john[1], self.george[1], self.tom[1]])
+
+        self.assertEqual(
+            self.founders.get_col(self.headers.index('gpa')),
+            [self.john[2], self.george[2], self.tom[2]])
+
+            
     def test_data_slicing(self):
         """Verify slicing by data."""