Fix python test and buffer protocol type hinting

python/python/geoindex_rs/kdtree.pyi

@@ -1,9 +1,16 @@
 from __future__ import annotations
+
+import sys
 from typing import Literal, Union
 
 import numpy as np
-from arro3.core.types import ArrowArrayExportable
 from arro3.core import Array
+from arro3.core.types import ArrowArrayExportable
+
+if sys.version_info > (3, 12):
+    from collections.abc import Buffer
+else:
+    from typing_extensions import Buffer
 
 ArrayLike = Union[np.ndarray, ArrowArrayExportable, memoryview, bytes]
 IndexLike = Union[np.ndarray, ArrowArrayExportable, memoryview, bytes, KDTree]
@@ -32,4 +39,4 @@ class KDTreeBuilder:
     def add(self, x: ArrayLike, y: ArrayLike | None = None) -> Array: ...
     def finish(self) -> KDTree: ...
 
-class KDTree: ...
+class KDTree(Buffer): ...

python/python/geoindex_rs/rtree.pyi

@@ -1,9 +1,16 @@
 from __future__ import annotations
+
+import sys
 from typing import Literal, Union
 
 import numpy as np
-from arro3.core.types import ArrowArrayExportable
 from arro3.core import Array
+from arro3.core.types import ArrowArrayExportable
+
+if sys.version_info > (3, 12):
+    from collections.abc import Buffer
+else:
+    from typing_extensions import Buffer
 
 ArrayLike = Union[np.ndarray, ArrowArrayExportable, memoryview, bytes]
 IndexLike = Union[np.ndarray, ArrowArrayExportable, memoryview, bytes, RTree]
@@ -36,7 +43,7 @@ class RTreeBuilder:
     ) -> Array: ...
     def finish(self, method: Literal["hilbert", "str", None] = None) -> RTree: ...
 
-class RTree:
+class RTree(Buffer):
     @property
     def num_items(self) -> int: ...
     @property

python/tests/test_buffers.py

@@ -1,6 +1,5 @@
 import numpy as np
-
-# from .. import RTree
+from geoindex_rs import rtree
 
 
 def generate_random_boxes():
@@ -11,7 +10,9 @@ def generate_random_boxes():
 
 def test_buffer_protocol():
     boxes = generate_random_boxes()
-    initial = RTree.from_interleaved(boxes)
+    builder = rtree.RTreeBuilder(len(boxes))
+    builder.add(boxes)
+    initial = builder.finish()
     # construct a memoryview transparently
     view = memoryview(initial)
     assert initial.num_bytes == view.nbytes