hash un-hashable point-types as tuples, fixes #12

kdtree.py

@@ -186,7 +186,11 @@ def __eq__(self, other):
             return self.data == other.data
 
     def __hash__(self):
-        return hash(self.data)
+        try:
+            return hash(self.data)
+        except TypeError:
+            # try to hash un-hashable types as tuples
+            return hash(tuple(self.data))
 
 
 def require_axis(f):

test.py

@@ -7,6 +7,7 @@
 import logging
 import unittest
 import doctest
+import collections
 from itertools import islice
 
 import kdtree
@@ -234,6 +235,21 @@ def test_search_nn_dist(self):
         self.assertTrue( (6,5) in nn)
 
 
+class PointTypeTests(unittest.TestCase):
+    """ test using different types as points """
+
+    def test_point_types(self):
+        emptyTree = kdtree.create(dimensions=3)
+        point1 = (2, 3, 4)
+        point2 = [4, 5, 6]
+        Point = collections.namedtuple('Point', 'x y z')
+        point3 = Point(5, 3, 2)
+        tree = kdtree.create([point1, point2, point3])
+        res = tree.search_nn( (1, 2, 3) )
+
+        self.assertEqual(res, kdtree.KDNode( (2, 3, 4) ))
+
+
 def random_tree(nodes=20, dimensions=3, minval=0, maxval=100):
     points = list(islice(random_points(), 0, nodes))
     tree = kdtree.create(points)