diff options
| -rw-r--r-- | src/semantic_version/base.py | 7 | ||||
| -rwxr-xr-x | tests/test_base.py | 8 |
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() |