Added support for detecting unicode column headers

Also added tests!

Fix for kennethreitz#26

1 files changed, 3 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