Fixed Wordlist bugs: hi must be int, and list may be empty.

darius · darius · commit 72bfc9d9f951 · 2011-09-15T01:22:17.000Z
diff --git a/search.py b/search.py
@@ -593,14 +593,15 @@ def lookup(self, prefix, lo=0, hi=None):
         words[i].startswith(prefix), or is None; the second is
         True iff prefix itself is in the Wordlist."""
         words = self.words
+        if hi is None: hi = len(words)
         i = bisect.bisect_left(words, prefix, lo, hi)
         if i < len(words) and words[i].startswith(prefix): 
             return i, (words[i] == prefix)
         else: 
             return None, False
 
     def __contains__(self, word): 
-        return self.words[bisect.bisect_left(self.words, word)] == word
+        return self.lookup(word)[1]
 
     def __len__(self): 
         return len(self.words)
