Add support for __hash__ to Version/Spec.

Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>

2 files changed, 15 insertions, 0 deletions

diff --git a/src/semantic_version/base.py b/src/semantic_version/base.py
index 8638fa1..c927305 100644
--- a/src/semantic_version/base.py
+++ b/src/semantic_version/base.py
@@ -145,6 +145,9 @@ class Version(object):
             self.build,
         )
 
+    def __hash__(self):
+        return hash((self.major, self.minor, self.patch, self.prerelease, self.build))
+
     def _comparison_functions(self, partial=False):
         def prerelease_cmp(a, b):
             if a and b:
@@ -293,6 +296,10 @@ class Spec(object):
             return NotImplemented
         return self.kind == other.kind and self.spec == other.spec
 
+    def __hash__(self):
+        return hash((self.kind, self.spec))
+
+
 
 def compare(v1, v2):
     return cmp(Version(v1), Version(v2))
diff --git a/tests/test_base.py b/tests/test_base.py
index b4f170d..38c7924 100755
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -144,6 +144,10 @@ class VersionTestCase(unittest.TestCase):
                 base.Version(text, partial=True))
             self.assertNotEqual(text, base.Version(text, partial=True))
 
+    def test_hash(self):
+        self.assertEqual(1,
+            len(set([base.Version('0.1.0'), base.Version('0.1.0')])))
+
 
 class SpecTestCase(unittest.TestCase):
     components = {
@@ -255,6 +259,10 @@ class SpecTestCase(unittest.TestCase):
         self.assertEqual('==0.1.0', str(spec))
         self.assertEqual(base.Spec.KIND_EQUAL, spec.kind)
 
+    def test_hash(self):
+        self.assertEqual(1,
+            len(set([base.Spec('==0.1.0'), base.Spec('==0.1.0')])))
+
 
 if __name__ == '__main__':  # pragma: no cover
     unittest.main()