a
    “Àh•  ã                   @   sJ   d Z ddlmZ ddlmZmZmZ ddlZddlm	Z	 G dd„ dƒZ
dS )z)test sparse matrix construction functionsé    )Úassert_equal)Ú
csr_matrixÚ	csr_arrayÚsparrayN)Ú_extractc                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚTestExtractc                 C   sÒ   t ddggƒt ddggƒt ddggƒt dgdggƒt dgdggƒt dgdggƒt ddgddggƒt ddgddggƒt ddgddggƒt ddgddggƒt g d¢g d¢g d¢gƒt g d¢g d¢g d¢gƒjg| _d S )	Né   é   r   é   é   )r   r	   r   r   r
   )r   é   r   é   é   )r   r   é   é	   r   )r   ÚTÚcases)Úself© r   ú[/var/www/html/assistant/venv/lib/python3.9/site-packages/scipy/sparse/tests/test_extract.pyÚsetup_method   s    ôzTestExtract.setup_methodc                 C   sH   | j D ]<}t |¡\}}}t|||ff|jd}t| ¡ | ¡ ƒ qd S )N)Úshape)r   r   Úfindr   r   r   Útoarray)r   ÚAÚIÚJÚVÚBr   r   r   Ú	test_find   s    
zTestExtract.test_findc                 C   sD   | j D ]8}| ¡ }dD ]&}ttj||d ¡ tj||dƒ qqd S ©N)éýÿÿÿéþÿÿÿéÿÿÿÿr   r   r	   r
   )Úk)r   r   r   r   ÚtrilÚnp©r   r   r   r$   r   r   r   Ú	test_tril!   s    
zTestExtract.test_trilc                 C   sD   | j D ]8}| ¡ }dD ]&}ttj||d ¡ tj||dƒ qqd S r    )r   r   r   r   Útriur&   r'   r   r   r   Ú	test_triu'   s    
zTestExtract.test_triuc                 C   sh   | j D ]\}tt |¡tƒsJ ‚tt |¡tƒs2J ‚t|ƒ}tt |¡tƒrNJ ‚tt |¡tƒrJ ‚qd S )N)r   Ú
isinstancer   r%   r   r)   r   )r   r   ÚMr   r   r   Útest_array_vs_matrix-   s    
z TestExtract.test_array_vs_matrixN)Ú__name__Ú
__module__Ú__qualname__r   r   r(   r*   r-   r   r   r   r   r   
   s
   r   )Ú__doc__Znumpy.testingr   Zscipy.sparser   r   r   Únumpyr&   r   r   r   r   r   r   Ú<module>   s
   