U }h @sXddlZddlZddlmZddlmZmZdZiiie dddZ ddZ d d Z dS) N)KDTree)Paralleldelayed)p)Z euclideanZ manhattanZ chebyshevZ minkowskic Cs:tjdddgtd}tjtddt|W5QRXdS)z/Check that we do not accept object dtype array.)r)r)r r rr)Zdtypez(setting an array element with a sequence)matchN)nparrayobjectpytestZraises ValueErrorr)XrH/tmp/pip-unpacked-wheel-ig1s1lm8/sklearn/neighbors/tests/test_kd_tree.pytest_array_object_type srcsJtjd}|d}t|ddtdddfddd|gDd S) zgMake sure that KDTree queries work when joblib memmaps. Non-regression test for #21685 and #21228.r) rr)Z leaf_sizer)Zn_jobsZ max_nbytesc3s|]}tj|VqdS)N)rquery).0datatreerr sz4test_kdtree_picklable_with_joblib..N)r randomZ RandomStateZ random_samplerr)rngrrrr!test_kdtree_picklable_with_joblibs   r) Znumpyr rZsklearn.neighbors._kd_treerZsklearn.utils.parallelrrZ DIMENSIONdictZMETRICSrrrrrrs