a
    ;Za                     @   s|   d dl Zd dlmZ d dlZd dlmZ d dlm	Z	 dd Z
dd Zd	d
 Zdd Ze	edejdddgdd ZdS )    N)assert_array_almost_equal)_mds)ignore_warningsc                  C   s   t g dg dg dg dg} t ddgddgd	d
gddgg}tj| |dddd\}}t ddgddgddgddgg}t||dd d S )Nr            r   r      r
   r   r
   r      r   r
   r   r   /$ѿsh|?gw/?gT㥛 ?Mb?X9vο皙ɿS㥛?r
   r   )initZn_componentsZmax_itern_initgp=
ףgS㥛gT㥛 ?gʡE?gZd;?gx&gʡEݿgL7A`?r   )Zdecimal)nparraymdssmacofr   )simZX_ZX_true r   >lib/python3.9/site-packages/sklearn/manifold/tests/test_mds.pytest_smacof	   s    ""r    c                  C   s$  t g dg dg dg dg} tt t|  W d    n1 sL0    Y  t g dg dg dg} tt t|  W d    n1 s0    Y  t g dg dg dg dg} t ddgdd	gd
dgg}tt  tj| |dd W d    n1 s0    Y  d S )N)r   r   	   r   r	   r   r   r   r   r   r   r   r   r   r   )r   r   )r   r   pytestZraises
ValueErrorr   r   )r   r   r   r   r   test_smacof_error   s    "(("r$   c                  C   s@   t g dg dg dg dg} tjdddd}||  d S )	Nr   r	   r   r   Fr   precomputed)metricZn_jobsdissimilarity)r   r   r   MDSZfit)r   mds_clfr   r   r   test_MDS*   s    "r*   c                  C   sH   t jdd} d}tjt|d | j W d    n1 s:0    Y  d S )Nr%   )r&   z5Attribute `_pairwise` was deprecated in version 0\.24)match)r   r(   r"   ZwarnsFutureWarning	_pairwise)r)   msgr   r   r   test_MDS_pairwise_deprecated1   s    r/   )categoryz dissimilarity, expected_pairwise)r%   T)Z	euclideanFc                 C   s   t j| d}|j|ksJ d S )N)r'   )r   r(   r-   )r'   Zexpected_pairwiser)   r   r   r   test_MDS_pairwise9   s    
r1   )Znumpyr   Znumpy.testingr   r"   Zsklearn.manifoldr   r   Zsklearn.utils._testingr   r    r$   r*   r/   r,   ZmarkZparametrizer1   r   r   r   r   <module>   s    