U
    ¯ËbŒ  ã                   @   s:   d dl Z d dlZd dlZddlmZ G dd„ dejƒZdS )é    Né   )Úutilc                   @   sN   e Zd Ze dddd¡gZejjdd„ ƒZ	ejjdd„ ƒZ
ejjd	d
„ ƒZdS )ÚTestSizeSumExampleÚtestsÚsrcÚsizezfoo.f90c                 C   s    | j  g g¡}|dgkst‚| j  ddgg¡}|dgks<t‚| j  ddgddgg¡}t |ddg¡sht‚| j  ddgddgddgg¡}t |ddd	g¡sœt‚d S )
Nr   r   é   é   é   é   é   é   é   )ÚmoduleZfooÚAssertionErrorÚnpÚallclose©ÚselfÚr© r   úZ/home/fireinfo/NEWAFireInfo/venv/lib/python3.8/site-packages/numpy/f2py/tests/test_size.pyÚtest_all   s    zTestSizeSumExample.test_allc                 C   s”   | j  g g¡}t |jt g g¡¡s(t‚| j  ddgg¡}t |dgdgg¡sRt‚| j  dddgdddgg¡}t |ddgddgddgg¡st‚d S )	Nr   r   g      ð?g       @r	   r
   r   r   )r   Ztransr   r   ÚTÚarrayr   r   r   r   r   Útest_transpose   s    z!TestSizeSumExample.test_transposec              	   C   s€   | j  g g¡}t |g ¡st‚| j  ddgg¡}t |ddg¡sDt‚| j  dddgdddgg¡}t |ddddddg¡s|t‚d S )Nr   r   r	   r
   r   r   )r   Úflattenr   r   r   r   r   r   r   Útest_flatten$   s    zTestSizeSumExample.test_flattenN)Ú__name__Ú
__module__Ú__qualname__r   ZgetpathÚsourcesÚpytestÚmarkZslowr   r   r   r   r   r   r   r      s   


r   )Úosr"   Znumpyr   Ú r   ZF2PyTestr   r   r   r   r   Ú<module>   s   