From 6313437a27d68e42247802f7f68ec3270cb9a2da Mon Sep 17 00:00:00 2001
From: Mark Walling <mark@markwalling.org>
Date: Fri, 1 Jul 2011 17:51:43 -0400
Subject: [PATCH] Added support for detecting unicode column headers

Also added tests!

Fix for kennethreitz#26
---
 tablib/core.py |  6 +++---
 test_tablib.py | 42 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 45 insertions(+), 3 deletions(-)

diff --git a/tablib/core.py b/tablib/core.py
index 53f3767..00e55af 100644
--- a/tablib/core.py
+++ b/tablib/core.py
@@ -162,7 +162,7 @@ class Dataset(object):
 
 
     def __getitem__(self, key):
-        if isinstance(key, str):
+        if isinstance(key, str) or isinstance(key, unicode):
             if key in self.headers:
                 pos = self.headers.index(key) # get 'key' index from each data
                 return [row[pos] for row in self._data]
@@ -182,7 +182,7 @@ class Dataset(object):
 
 
     def __delitem__(self, key):
-        if isinstance(key, str):
+        if isinstance(key, str) or isinstance(key, unicode):
 
             if key in self.headers:
 
@@ -730,7 +730,7 @@ class Dataset(object):
         sorted.
         """
 
-        if isinstance(col, str):
+        if isinstance(col, str) or isinstance(col, unicode):
 
             if not self.headers:
                 raise HeadersNeeded
diff --git a/test_tablib.py b/test_tablib.py
index 252dde4..6695f5e 100755
--- a/test_tablib.py
+++ b/test_tablib.py
@@ -511,6 +511,48 @@ class TablibTestCase(unittest.TestCase):
 
 
         data.csv
+    
+    def test_csv_column_select(self):
+        """Build up a CSV and test selecting a column"""
+
+        data = tablib.Dataset()
+        data.csv = self.founders.csv
+
+        headers = data.headers
+        self.assertTrue(isinstance(headers[0], unicode))
+
+        orig_first_name = self.founders[self.headers[0]]
+        csv_first_name = data[headers[0]]
+        self.assertEquals(orig_first_name, csv_first_name)
+
+
+    def test_csv_column_delete(self):
+        """Build up a CSV and test deleting a column"""
+
+        data = tablib.Dataset()
+        data.csv = self.founders.csv
+
+        target_header = data.headers[0]
+        self.assertTrue(isinstance(target_header, unicode))
+
+        del data[target_header]
+
+        self.assertTrue(target_header not in data.headers)
+
+    def test_csv_column_sort(self):
+        """Build up a CSV and test sorting a column by name"""
+
+        data = tablib.Dataset()
+        data.csv = self.founders.csv
+
+        orig_target_header = self.founders.headers[1]
+        target_header = data.headers[1]
+
+        self.founders.sort(orig_target_header)
+        data.sort(target_header)
+
+        self.assertEquals(self.founders[orig_target_header], data[target_header])
+
 
 if __name__ == '__main__':
     unittest.main()