Make ComparableVersion hashable

Open-EO · Apr 4, 2023 · eda8e59 · eda8e59
1 parent 1e16536
commit eda8e59
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/openeo/_version.py b/openeo/_version.py
@@ -1 +1 @@
-__version__ = "0.16.0a4"
+__version__ = "0.16.0a5"
diff --git a/openeo/capabilities.py b/openeo/capabilities.py
@@ -118,6 +118,9 @@ def __repr__(self):
     def __str__(self):
         return ".".join(map(str, self._version))
 
+    def __hash__(self):
+        return hash(self._version)
+
     def to_string(self):
         return str(self)
 

diff --git a/tests/test_capabilities.py b/tests/test_capabilities.py
@@ -142,3 +142,19 @@ def test_require_at_least(self):
         v.require_at_least("1.2.0")
         with pytest.raises(ApiVersionException):
             v.require_at_least("1.2.4")
+
+    def test_hashable_dict(self):
+        d = {
+            ComparableVersion("1.2.3"): "red",
+        }
+        assert d[ComparableVersion((1, 2, 3))] == "red"
+
+    def test_hashable_set(self):
+        s = {
+            ComparableVersion("1.2.3"),
+            ComparableVersion("2.4.6"),
+        }
+        assert s == {
+            ComparableVersion((2, 4, 6)),
+            ComparableVersion("1.2.3"),
+        }