make coordinates hashable

Author: pattonw

funlib/geometry/coordinate.py

@@ -3,6 +3,7 @@
 
 import numpy as np
 
+
 class Coordinate(tuple):
     """A ``tuple`` of numbers or None.
 
@@ -37,6 +38,9 @@ def __new__(cls, *array_like):
     def __eq__(self, other) -> bool:
         return np.array_equal(self, other)
 
+    def __hash__(self) -> int:
+        return hash(tuple(x for x in self))
+
     @property
     def dims(self) -> int:
         return len(self)

tests/test_coordinate.py

@@ -96,6 +96,7 @@ def test_none():
     assert abs(a) == (None, 1, 2)
     assert abs(-a) == (None, 1, 2)
 
+
 def test_json():
     a = Coordinate((1, 2, 3))
     b = Coordinate((None, 2, 3))
@@ -104,4 +105,10 @@ def test_json():
     b_json = json.dumps(b)
 
     assert a == json.loads(a_json)
-    assert b == json.loads(b_json)
+    assert b == json.loads(b_json)
+
+
+def test_hash():
+    assert {Coordinate(1, 2, 3): "A", Coordinate(4.5, 5.5, 6.5): "B"}[
+        Coordinate(1, 2, 3)
+    ] == "A"