o
    gH                     @   st   d dl Z d dlZddlmZ d dlmZ G dd dejZG dd dejZ	G d	d
 d
ejZ
G dd dejZdS )    N   )util)crackfortranc                   @   s6   e Zd ZeddddgZejjdd Z	dd Z
d	S )
TestDatatestssrcr   zdata_stmts.f90c                 C   s  | j jjdks	J | j jjdksJ | j jjdksJ | j jjdks$J | j jjdks-J | j jjt	dks9J t
| j jjt	ddgksJJ t
| j jjt	d	d
gks[J t
| j jjt	g dkslJ t
| j jjt	dgks|J t
| j jjt	dgksJ d S )N      g      ?       @g-DT!	@y      ?        g      @g      @y      ?       @y            @)g      ?r
   g      @y      *@      5@y      >     E@)modulecmplxdatijxypimedium_ref_indexnparrayallzmy_arraymy_real_arrayref_index_oneref_index_twoself r   Q/var/www/visachat/venv/lib/python3.10/site-packages/numpy/f2py/tests/test_data.pytest_data_stmts   s   """ $zTestData.test_data_stmtsc                 C   s   t | j}|d d d d dksJ |d d d d dks!J |d d d d d	ks/J |d d d
 d dks=J |d d d d dksKJ |d d d d dksYJ |d d d d dksgJ |d d d d dksuJ d S )Nr   varsr   =z1.5r   z2.0r   zM3.1415926535897932384626433832795028841971693993751058209749445923078164062d0r   z(/1.0d0, 2.0d0, 3.0d0/)r   z(13.0d0, 21.0d0)r   z(-30.0d0, 43.0d0)r   z#(/(1.0d0, 2.0d0), (-3.0d0, 4.0d0)/)r   z(/3.5,  7.0/))r   sourcesr   modr   r   r   test_crackedlines   s   
 zTestData.test_crackedlinesN)__name__
__module____qualname__r   getpathr"   pytestmarkslowr   r%   r   r   r   r   r      s
    
r   c                   @   s.   e Zd ZeddddgZdd Zdd Zd	S )
TestDataF77r   r   r   zdata_common.fc                 C   s   | j jjdks	J d S )Nr   )r   mycommydatar   r   r   r   r   )   s   zTestDataF77.test_data_stmtsc                 C   sB   t t| jd }t|d d  |d d d d dksJ d S )Nr   r    r/   r!   0)r   strr"   printr#   r   r   r   r%   ,   s    zTestDataF77.test_crackedlinesN)r&   r'   r(   r   r)   r"   r   r%   r   r   r   r   r-   %   s    r-   c                   @   &   e Zd ZeddddgZdd ZdS )TestDataMultiplierF77r   r   r   zdata_multiplier.fc                 C   s^   | j jjdks	J | j jjdksJ | j jjdksJ | j jjdks$J | j jjdks-J d S )Nr	   r   r   )r   r.   ivar1ivar2ivar3ivar4evar5r   r   r   r   r   6   s
   z%TestDataMultiplierF77.test_data_stmtsNr&   r'   r(   r   r)   r"   r   r   r   r   r   r4   2       r4   c                   @   r3   )TestDataWithCommentsF77r   r   r   zdata_with_comments.fc                 C   s\   t | jjjdksJ | jjjd dksJ | jjjd dks!J | jjjd dks,J d S )Nr	   r   r      r   )lenr   r.   mytabr   r   r   r   r   B   s   z'TestDataWithCommentsF77.test_data_stmtsNr:   r   r   r   r   r<   >   r;   r<   )r*   numpyr    r   numpy.f2py.crackfortranr   F2PyTestr   r-   r4   r<   r   r   r   r   <module>   s    