a
    LfR                     @   s|   d dl Z 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s6J | j jjdksHJ | j jjdksZJ | j jjt	dksrJ t
| j jjt	ddgksJ t
| j jjt	d	d
gksJ t
| j jjt	g dksJ t
| j jjt	dgksJ t
| j jjt	dgksJ d S )N      g      ?       @g-DT!	@y      ?        g      @g      @y      ?       @y            @)g      ?r
   g      @y      *@      5@y      >     E@)moduleZcmplxdatijxypiZmedium_ref_indexnparrayallzmy_arraymy_real_arrayref_index_oneref_index_twoself r   S/var/www/ai-form-bot/venv/lib/python3.9/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sBJ |d d d d d	ks^J |d d d
 d dkszJ |d d d d dksJ |d d d d dksJ |d d d d dksJ |d d d d dksJ 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markZ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                   @   s&   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s6J | j jjdksHJ | j jjdksZJ d S )Nr	   r   r   )r   r+   Zivar1Zivar2Zivar3Zivar4Zevar5r   r   r   r   r   7   s
    z%TestDataMultiplierF77.test_data_stmtsNr$   r%   r&   r   r'   r    r   r   r   r   r   r0   3   s   r0   c                   @   s&   e Zd ZeddddgZdd ZdS )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sBJ | jjjd dksXJ d S )Nr	   r   r      r   )lenr   r+   Zmytabr   r   r   r   r   C   s    z'TestDataWithCommentsF77.test_data_stmtsNr1   r   r   r   r   r2   ?   s   r2   )osr(   numpyr    r   Znumpy.f2py.crackfortranr   ZF2PyTestr   r*   r0   r2   r   r   r   r   <module>   s   