a
    )a                     @   s0   d dl Zd dlmZ dd Zdd Zdd ZdS )	    N)assert_allclosec                  C   sb   ddl m}  dgdgdgdgg}g d}| dd||}d|jv sHJ t|d	ggdg d S )
Nr   )KNeighborsClassifier         r   r   r   r   Zn_neighborsdaal4pyg?)sklearnex.neighborsr   fit
__module__r   predict)r   Xyneigh r   Glib/python3.9/site-packages/sklearnex/neighbors/tests/test_neighbors.py$test_sklearnex_import_knn_classifier   s    r   c                  C   sb   ddl m}  dgdgdgdgg}g d}| dd||}d|jv sHJ t|d	ggd
g d S )Nr   )KNeighborsRegressorr   r   r   r   r   r	   g      ?g      ?)r
   r   r   r   r   r   )r   r   r   r   r   r   r   $test_sklearnex_import_knn_regression   s    r   c                  C   sj   ddl m}  g dg dg dg}| dd|}d|jv s@J |jg d	gdd
d}t|ddgg d S )Nr   )NearestNeighbors)r   r   r   )r   r   r   )r   r   r   r   r   r	   )r   r   g?F)Zreturn_distance)r
   r   r   r   Z
kneighborsr   )r   r   r   resultr   r   r   test_sklearnex_import_nn(   s    r   )ZnumpyZnpZnumpy.testingr   r   r   r   r   r   r   r   <module>   s   		