Merge pull request #6 from mhluongo/master

Closed issue #5

Author: mhluongo

lucenequerybuilder/query.py

@@ -108,6 +108,12 @@ def __add__(self, other):
     def __sub__(self, other):
         return self | Q._make_must_not(self)
 
+    def __eq__(self, other):
+        return hash(self) == hash(other)
+
+    def __hash__(self):
+        return hash((tuple(self.should), tuple(self.must), tuple(self.must_not)))
+
     def __str__(self):
         rv = ''
         if hasattr(self, 'field'):

lucenequerybuilder/tests.py

@@ -4,7 +4,6 @@
 """
 
 from lucenequerybuilder import Q
-import re
 
 def test_general():
     a = 'a'
@@ -38,6 +37,23 @@ def test_simple_phrase():
     query_string = str(Q('abc 123'))
     assert query_string == '"abc 123"', query_string
 
+def test_hashing():
+    q1 = Q('a') & Q('b') | Q('c')
+    q2 = Q('a') & Q('b') | Q('c')
+    q3 = q1 | Q('d')
+
+    assert q1 == q2, "Queries aren't being properly evaluated for equality."
+    assert q2 != q3, "Queries aren't being properly evaluated for inequality."
+
+    d = {}
+    try:
+        d[q1] = 1
+        d[q2] = 2
+    except:
+        raise AssertionError('There was an error using queries as dict keys.')
+    assert d[q2] == 2, 'Got the wrong value back from the query dict!'
+
+
 #this test doesn't work, but might be worth rewriting
 #
 #def test_escaping():