31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
import numpy as np
|
|
import pytest
|
|
from sklearn.utils.parallel import delayed, Parallel
|
|
|
|
from sklearn.neighbors._kd_tree import KDTree
|
|
|
|
DIMENSION = 3
|
|
|
|
METRICS = {"euclidean": {}, "manhattan": {}, "chebyshev": {}, "minkowski": dict(p=3)}
|
|
|
|
|
|
def test_array_object_type():
|
|
"""Check that we do not accept object dtype array."""
|
|
X = np.array([(1, 2, 3), (2, 5), (5, 5, 1, 2)], dtype=object)
|
|
with pytest.raises(ValueError, match="setting an array element with a sequence"):
|
|
KDTree(X)
|
|
|
|
|
|
def test_kdtree_picklable_with_joblib():
|
|
"""Make sure that KDTree queries work when joblib memmaps.
|
|
|
|
Non-regression test for #21685 and #21228."""
|
|
rng = np.random.RandomState(0)
|
|
X = rng.random_sample((10, 3))
|
|
tree = KDTree(X, leaf_size=2)
|
|
|
|
# Call Parallel with max_nbytes=1 to trigger readonly memory mapping that
|
|
# use to raise "ValueError: buffer source array is read-only" in a previous
|
|
# version of the Cython code.
|
|
Parallel(n_jobs=2, max_nbytes=1)(delayed(tree.query)(data) for data in 2 * [X])
|