Merge pull request #294 from havelock/master

Added Choice.__hash__()
kvesteri · Nov 5, 2017 · 96057c0 · 96057c0
2 parents a9389f4 + 8a68c4b
commit 96057c0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/sqlalchemy_utils/types/choice.py b/sqlalchemy_utils/types/choice.py
@@ -20,6 +20,9 @@ def __eq__(self, other):
             return self.code == other.code
         return other == self.code
 
+    def __hash__(self):
+        return hash(self.code)
+
     def __ne__(self, other):
         return not (self == other)
 

diff --git a/tests/types/test_choice.py b/tests/types/test_choice.py
@@ -16,6 +16,9 @@ def test_non_equality_operator(self):
         assert Choice(1, 1) != 2
         assert not (Choice(1, 1) != 1)
 
+    def test_hash(self):
+        assert hash(Choice(1, 1)) == hash(1)
+
 
 class TestChoiceType(object):