fix basestring for py3

Fantomas42 · Jan 30, 2015 · 742b9b7 · 742b9b7
1 parent 500cee1
commit 742b9b7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mots_vides/stop_words.py b/mots_vides/stop_words.py
@@ -4,6 +4,8 @@
 """
 import re
 
+TEXT_TYPE_LIST = ('str', 'unicode', 'byte')
+
 
 class StopWord(object):
     """
@@ -21,7 +23,7 @@ def __add__(self, entry):
         """
         Adds an entry or collection of entries to an instance.
         """
-        if isinstance(entry, basestring):
+        if type(entry).__name__ in TEXT_TYPE_LIST:
             self.collection.add(entry)
         else:
             self.collection = self.collection.union(entry)
@@ -32,7 +34,7 @@ def __sub__(self, entry):
         """
         Substracts an entry or collection of entries to an instance.
         """
-        if isinstance(entry, basestring):
+        if type(entry).__name__ in TEXT_TYPE_LIST:
             self.collection.remove(entry)
         else:
             self.collection = self.collection.difference(entry)