Merge pull request #22 from mortacious/bugfix/flipped_compact_balanced

Bugfix/flipped compact balanced

Author: mortacious

numba_kdtree/_version.py

@@ -473,7 +473,7 @@ def KDTree(data: DataArray, leafsize: int = 10, compact: bool = False, balanced:
 
     idx = np.arange(n_data, dtype=INT_TYPE)
 
-    kdtree = _make_kdtree(data, root_bbox, idx, leafsize, compact, balanced)
+    kdtree = _make_kdtree(data, root_bbox, idx, leafsize, balanced, compact)
     return kdtree
 
 
@@ -511,7 +511,7 @@ def KDTree_impl(data, leafsize=10, compact=False, balanced=False, root_bbox=None
 
         idx = np.arange(n_data, dtype=INT_TYPE)
 
-        kdtree = _make_kdtree(data, root_bbox, idx, leafsize, compact, balanced)
+        kdtree = _make_kdtree(data, root_bbox, idx, leafsize, balanced, compact)
 
         return kdtree
 

numba_kdtree/kd_tree.py

@@ -242,12 +242,12 @@ def query_numba_parallel(data, kdtree, k):
 
 def test_construct_in_numba_function(data):
     @nb.njit(nogil=True, fastmath=True)
-    def construct_kdtree_in_numba(data):
-        kdtree = KDTree(data, leafsize=10)
+    def construct_kdtree_in_numba(data, compact=False, balanced=False):
+        kdtree = KDTree(data, leafsize=10, compact=compact, balanced=balanced)
         return kdtree
 
     num_executions = 10
-    runtime_kdtree_numba = timeit(lambda: construct_kdtree_in_numba(data),
+    runtime_kdtree_numba = timeit(lambda: construct_kdtree_in_numba(data, compact=True, balanced=True),
                               number=num_executions) / num_executions
 
     runtime_kdtree_python = timeit(lambda: KDTree(data), number=num_executions) / num_executions