Prettied up parse output.

1 files changed, 5 insertions, 0 deletions

diff --git a/parse/tuple_table.py b/parse/tuple_table.py
index cb5a72a..434eb22 100644
--- a/parse/tuple_table.py
+++ b/parse/tuple_table.py
@@ -48,6 +48,7 @@ class TupleTable(object):
         self.table = defaultdict(lambda: [])
         self.reduced = False
 
+    # TODO: rename, make exp agnostic, extend for exps
     def add_exp(self, kv, point):
         key = self.col_map.get_key(kv)
         self.table[key] += [point]
@@ -56,6 +57,10 @@ class TupleTable(object):
         key = self.col_map.get_key(kv)
         return self.table[key]
 
+    def __contains__(self, kv):
+        key = self.col_map.get_key(kv)
+        return key in self.table
+
     def reduce(self):
         if self.reduced:
             raise Exception("cannot reduce twice!")