o
    e]                  	   @   s  d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZm	Z	 d dl
mZ ddlmZ daed Zed5deedejd	ejejfieZd
d Zdd Zdd ZG dd dZe Zg dZddgiZed dg ed< ed dg ed< dged< dged< ed ddg ed< ed ddg ed< ed ddg ed< ed ddg ed< ed dg ed< ed dg ed< ed dg ed< ed dg ed< ed ddg ed< ed g d ed < ed d!g ed!< d"ged"< d#ged#< dged< e jjd$kse jjd	krBejd%krBe  e! fd&krBe"g d' ed g d( ed)< ed) g d* ed+< ed  d!d,g ed,< G d-d. d.Z#G d/d0 d0Z$G d1d2 d2Z%G d3d4 d4Z&dS )6    N)assert_assert_equal)typeinfo   )utilSTRING	CHARACTERc   c                  C   sf   t  s	td tdu r1d} tjt}t 	ddddt 	ddt 	dd	g}t 
|| d
adS dS )z6
    Build the required testing extension module

    zNo C compiler availableNz
        config.add_extension('test_array_from_pyobj_ext',
                             sources=['wrapmodule.c', 'fortranobject.c'],
                             define_macros=[])
        testssrcarray_from_pyobjzwrapmodule.czfortranobject.czfortranobject.htest_array_from_pyobj_ext)r   has_c_compilerpytestskipwrapospathdirname__file__getpathbuild_module_distutils)config_codedr    r   Y/var/www/html/venv/lib/python3.10/site-packages/numpy/f2py/tests/test_array_from_pyobj.pysetup_module   s   


r   c                 C   s   t | d }t|S )N   )r   array_attrsflags2names)arrflagsr   r   r   
flags_info1   s   r#   c                 C   s0   g }dD ]}t | tt|d@ r|| q|S )N)
CONTIGUOUSFORTRANOWNDATA
ENSURECOPYENSUREARRAYALIGNED
NOTSWAPPED	WRITEABLEWRITEBACKIFCOPYUPDATEIFCOPYBEHAVED
BEHAVED_ROCARRAYFARRAYr   )absgetattrr   append)r"   infoflagnamer   r   r   r    6   s   
r    c                   @   s@   e Zd Zg fddZdd Zdd Zdd Zd	d
 Zdd ZdS )Intentc                 C   sP   |d d  | _ d}|D ]}|dkr|tjO }q|ttd|  O }q|| _d S )Nr   optionalF2PY_INTENT_)intent_listr   F2PY_OPTIONALr3   upperr"   )selfr:   r"   ir   r   r   __init__N   s   
zIntent.__init__c                 C   s&   |  }|dkr
d}| | j|g S )Nin_in)lower	__class__r:   )r=   namer   r   r   __getattr__X   s   zIntent.__getattr__c                 C   s   dd | j S )Nz
intent(%s),)joinr:   r=   r   r   r   __str__^   s   zIntent.__str__c                 C   s
   d| j  S )Nz
Intent(%r)r:   rH   r   r   r   __repr__a      
zIntent.__repr__c                 G   s   |D ]
}|| j vr dS qdS )NFTrJ   )r=   namesrD   r   r   r   	is_intentd   s
   
zIntent.is_intentc                 G   s   t | jt |ko| j| S N)lenr:   rN   )r=   rM   r   r   r   is_intent_exactj   s   zIntent.is_intent_exactN)	__name__
__module____qualname__r?   rE   rI   rK   rN   rQ   r   r   r   r   r7   M   s    
r7   )BOOLBYTEUBYTESHORTUSHORTINTUINTLONGULONGLONGLONG	ULONGLONGFLOATDOUBLECFLOATSTRING1STRING5r   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   )r[   r`   ra   ra   rb   rc   rd      win32)Darwinarm)
LONGDOUBLECDOUBLECLONGDOUBLE)r]   r`   ra   ri   ri   )rb   rj   rk   rk   rj   c                   @   sP   e Zd Zi Zdd Zdd Zdd Zdd Zd	d
 Zdd Z	dd Z
dd ZdS )Typec                 C   s   t |tjr"|}d }t D ]\}}t |ts!|j|ju r!|} nq| j| d }|d ur1|S t	
| }|| || j| < |S rO   )
isinstancenpdtyper   itemstype_type_cachegetr<   object__new___init)clsrD   dtype0nr>   objr   r   r   ru      s   

zType.__new__c                 C   s   |  | _| jdkrt| j }ttd| _d| _td| _nG| j	drKt| jd d  }ttd| _t
| jdd  p=d| _td| j | _nt| j }ttd	| j | _|jd
 | _t|j| _| j|jksnJ |j| _|j| _d S )Nr   
NPY_STRINGr   r	   r   r   r   SNPY_r
   )r<   NAMEr   r3   r   type_numelsizern   ro   
startswithintbitsrq   numchar	dtypechar)r=   rD   r5   r   r   r   rv      s$   



z
Type._initc                 C   s4   d| j  d| j d| j d| j d| j d| j S )NzType(z)|type_num=z, dtype=z, type=z	, elsize=z, dtypechar=)r~   r   ro   rq   r   r   rH   r   r   r   rK      s   zType.__repr__c                    s    fddt  j D S )Nc                       g | ]}  |qS r   rC   .0_mrH   r   r   
<listcomp>       z#Type.cast_types.<locals>.<listcomp>)
_cast_dictr~   rH   r   rH   r   
cast_types   s   zType.cast_typesc                    s    fddt D S )Nc                    r   r   r   r   rH   r   r   r      r   z"Type.all_types.<locals>.<listcomp>)_type_namesrH   r   rH   r   	all_types   s   zType.all_typesc                 C   s:   t | j j}g }tD ]}t | j|k r|t| q
|S rO   r   r~   	alignmentr   r4   rl   r=   r   typesrD   r   r   r   smaller_types      zType.smaller_typesc                 C   sF   t | j j}g }tD ]}|| jkrq
t | j|kr |t| q
|S rO   r   r   r   r   r   equal_types   s   
zType.equal_typesc                 C   s:   t | j j}g }tD ]}t | j|kr|t| q
|S rO   r   r   r   r   r   larger_types   r   zType.larger_typesN)rR   rS   rT   rr   ru   rv   rK   r   r   r   r   r   r   r   r   r   rl      s    
rl   c                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )Arrayc              
   C   s,   d| j  d| j d| j d| j d| j 
S )NzArray(z, z)|arr=)rq   dimsintentrz   r!   rH   r   r   r   rK     s
   zArray.__repr__c                 C   sF  || _ || _|| _t|| _|| _t|j	|j
||j|| _t| jtjs(J t| j| _t|dkr| jdr^|jtj@ sCJ | jjd rKJ | jjd sSJ | jd tj@ r]J n"|jtj@ rfJ | jjd snJ | jjd rvJ | jd tj@ sJ |d u rd | _d | _d S |drt|tjsJ tt |t|j|  | _n!tjtj||jdj| | jdrdpd	d
| _| jj|ksJ | jj| jjd d | jjd sJ ||ft| j| _t|dkr3| jdr| jjd rJ | jjd s
J | jd tj@ rJ n| jjd sJ | jjd r(J | jd tj@ s3J | jd | jd ks@J | jd | jd ksMJ | jd dkrv| jd | jd ksvJ t| jd | jd | j | j f| jd dd  | jd dd  ksJ t| jd | jd f| jd | jd ksJ t| jd | jd td| jd  | jd  t| jd |f|dr| jd d | j j
ksJ n| jd d | j j
ksJ |  | j| jsJ t| jtjr|j
t!|jj
kr|ds| jd dkr| " s!J d S d S d S d S d S )Nr   r	   r%   r$   r   cachero   CF)orderr+   writer&            r   copy)#rq   r   r   r   deepcopyobj_copyrz   r   callr   r   r"   r!   rm   rn   ndarrayr   arr_attrrP   rN   F2PY_INTENT_Cr%   pyarr
pyarr_attrreprarrayreshaper   ro   setflagstobytesr    	arr_equalrl   has_shared_memory)r=   typr   r   rz   r   r   r   r?     s   
*zArray.__init__c                 C   s   |j |j krdS ||k S )NF)shapeall)r=   arr1arr2r   r   r   r   c  s   zArray.arr_equalc                 C   s
   t | jS rO   )strr!   rH   r   r   r   rI   h  rL   zArray.__str__c                 C   s@   | j | ju rdS t| j tjsdS t| j }|d | jd kS )z6Check that created array shares data with input array.TFr   )rz   r!   rm   rn   r   r   r   r   )r=   obj_attrr   r   r   r   k  s   zArray.has_shared_memoryN)rR   rS   rT   rK   r?   r   rI   r   r   r   r   r   r     s    Wr   c                   @   s   e Zd Zdd ZdS )
TestIntentc                 C   st   t tjjdks
J tjjdsJ tjjdrJ tjjdds&J tjjdds0J tjdr8J d S )Nzintent(in,out)r	   rA   )r   r   r@   outr	   rN   rQ   rH   r   r   r   test_in_outv  s   zTestIntent.test_in_outN)rR   rS   rT   r   r   r   r   r   r   u  s    r   c                   @   s"  e Zd Zejddeddd Zedd Zedd	 Z	d
d Z
dd Zejdddgejdddgejdddgdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Z d=d> Z!d?S )@TestSharedMemoryTclass)autousescopeparamsc                    s"   t  j j_ fdd j_d S )Nc                    s   t t j|||S rO   )r   rl   param)r=   r   r   rz   requestr   r   <lambda>  s    z-TestSharedMemory.setup_type.<locals>.<lambda>)rl   r   rw   rq   r   )r=   r   r   r   r   
setup_type  s   zTestSharedMemory.setup_typec                 C   s.   | j jdr| j j}d| d| gS ddgS )Nr   12r   r   rq   r~   r   r   r=   r   r   r   r   num2seq  s   zTestSharedMemory.num2seqc                 C   sR   | j jdr!| j j}d| d| d| gd| d| d| ggS g dg d	gS )
Nr   r   r   3456)r   r   r   )re   r   r   r   r   r   r   r   num23seq  s   zTestSharedMemory.num23seqc                 C   s$   |  dgtj| j}| rJ d S )Nr   )r   r   r@   r   r   r=   ar   r   r   test_in_from_2seq  s   z"TestSharedMemory.test_in_from_2seqc                 C   sx   | j  D ]4}tj| j|jd}| t| jgtj|}|j	| j j	kr3|
 s2J t| j j|jfq|
 r9J qd S Nr   )rq   r   rn   r   r   ro   rP   r   r@   r   r   r   r=   trz   r   r   r   r   test_in_from_2casttype  s    z'TestSharedMemory.test_in_from_2casttyper   wror   r   r   inp2seq23seqc                 C   sd   t | d| }tj|| jj|d}|j|dkd | |j|dkr%tjj	p'tj|}|
 s0J dS )z5Test if intent(in) array can be passed without copiesr   ro   r   r   r   r   N)r3   rn   r   rq   ro   r   r   r   r@   r	   r   )r=   r   r   r   seqrz   r   r   r   r   test_in_nocopy  s   zTestSharedMemory.test_in_nocopyc              
   C   s   t j| j| jjd}| t| jgtj|}| sJ z| dgtj	j| j}W td t
yG } zt|ds< W Y d }~d S d }~ww )Nr   r   z6failed to initialize intent(inout|inplace|cache) arrayz,intent(inout) should have failed on sequence)rn   r   r   rq   ro   rP   r   inoutr   r@   	TypeErrorr   r   SystemError)r=   rz   r   msgr   r   r   test_inout_2seq  s   z TestSharedMemory.test_inout_2seqc              
   C   s   t j| j| jjdd}t| jt| jd f}| |tjj|}|	 s&J t j| j| jjdd}t| jt| jd f}z| |tjj|}W td t
yf } zt|ds[ W Y d }~d S d }~ww )Nr   r   r   r   z(failed to initialize intent(inout) arrayz2intent(inout) should have failed on improper array)rn   r   r   rq   ro   rP   r   r@   r   r   
ValueErrorr   r   r   )r=   rz   r   r   r   r   r   r   test_f_inout_23seq  s&   z#TestSharedMemory.test_f_inout_23seqc                 C   sP   t j| j| jjd}t| jt| jd f}| |tjjj	|}|
 s&J d S Nr   r   )rn   r   r   rq   ro   rP   r   r@   r	   r   r   r=   rz   r   r   r   r   r   test_c_inout_23seq  s   z#TestSharedMemory.test_c_inout_23seqc                 C   sL   | j  D ]}tj| j|jd}| t| jgtjj	|}|
 r#J qd S r   )rq   r   rn   r   r   ro   rP   r   r@   r   r   r   r   r   r   test_in_copy_from_2casttype  s
   z,TestSharedMemory.test_in_copy_from_2casttypec                 C   s6   |  t| jt| jd gtj| j}| rJ d S Nr   )r   rP   r   r   r@   r   r   r   r   r   test_c_in_from_23seq  s
   z%TestSharedMemory.test_c_in_from_23seqc                 C   sV   | j  D ]#}tj| j|jd}| t| jt| jd gtj|}|	 r(J qd S r   )
rq   r   rn   r   r   ro   rP   r   r@   r   r   r   r   r   test_in_from_23casttype  s   z(TestSharedMemory.test_in_from_23casttypec                 C   st   | j  D ]2}tj| j|jdd}| t| jt| jd gtj|}|j	| j j	kr1|
 s0J q|
 r7J qd S Nr   r   r   )rq   r   rn   r   r   ro   rP   r   r@   r   r   r   r   r   r   test_f_in_from_23casttype  s   z*TestSharedMemory.test_f_in_from_23casttypec                 C   st   | j  D ]2}tj| j|jd}| t| jt| jd gtjj	|}|j
| j j
kr1| s0J q| r7J qd S r   )rq   r   rn   r   r   ro   rP   r   r@   r	   r   r   r   r   r   r   test_c_in_from_23casttype  s   z*TestSharedMemory.test_c_in_from_23casttypec                 C   sZ   | j  D ]%}tj| j|jdd}| t| jt| jd gtjj	|}|
 r*J qd S r   )rq   r   rn   r   r   ro   rP   r   r@   r   r   r   r   r   r   test_f_copy_in_from_23casttype  s   z/TestSharedMemory.test_f_copy_in_from_23casttypec                 C   sZ   | j  D ]%}tj| j|jd}| t| jt| jd gtjj	j
|}| r*J qd S r   )rq   r   rn   r   r   ro   rP   r   r@   r	   r   r   r   r   r   r   test_c_copy_in_from_23casttype  s   z/TestSharedMemory.test_c_copy_in_from_23casttypec                 C   s6  | j  D ]}|j| j jkrqtj| j|jd}t| jf}| |tj	j
j|}| s.J | |tj	j|}| s=J tj| j|jdd}| |tj	j
j|}| sWJ | |tj	j|}| skJ t|jz| |tj	j|d d d }W td ty } zt|ds W Y d }~qd }~ww d S )Nr   r   r   (failed to initialize intent(cache) arrayz8intent(cache) should have failed on multisegmented array)rq   r   r   rn   r   r   ro   rP   r   r@   r	   r   r   r   r   r   r   r   )r=   r   rz   r   r   r   r   r   r   test_in_cache_from_2casttype  s8   z-TestSharedMemory.test_in_cache_from_2casttypec                 C   s   | j  D ]G}|jdkrq|j| j jkrqtj| j|jd}t| jf}z| |t	j
j| W td tyL } zt|dsB W Y d }~qd }~ww d S )Nr   r   r   z1intent(cache) should have failed on smaller array)rq   r   r~   r   rn   r   r   ro   rP   r   r@   r   r   r   r   r   )r=   r   rz   r   r   r   r   r   $test_in_cache_from_2casttype_failure)  s*   
z5TestSharedMemory.test_in_cache_from_2casttype_failurec              
   C   s   d}|  |tjjd }|jj|ksJ d}|  |tjjd }|jj|ks&J d}z|  |tjjd }W t	d tyQ } zt|dsF W Y d }~d S d }~ww )Nr   r   r   r   r   2failed to create intent(cache|hide)|optional arrayz8intent(cache) should have failed on undefined dimensions)
r   r   r   hider!   r   r   r   r   r   r=   r   r   r   r   r   r   test_cache_hidden<  s(   z"TestSharedMemory.test_cache_hiddenc              
   C   s\  d}|  |tjd }|jj|ksJ ||jtj|| jj	ds"J d}|  |tjd }|jj|ks4J ||jtj|| jj	dsDJ |jj
d rP|jj
d rRJ d}|  |tjjd }|jj|kseJ ||jtj|| jj	dsuJ |jj
d s|jj
d sJ d}z|  |tjd }W td ty } zt|ds W Y d }~d S d }~ww )	Nr   r   r   r%   r$   r   r   z7intent(hide) should have failed on undefined dimensions)r   r   r   r!   r   r   rn   zerosrq   ro   r"   r	   r   r   r   r   r  r   r   r   test_hiddenP  s8      zTestSharedMemory.test_hiddenc                 C   s
  d}|  |tjd }|jj|ksJ ||jtj|| jj	ds"J d}|  |tjd }|jj|ks4J ||jtj|| jj	dsDJ |jj
d rP|jj
d rRJ d}|  |tjjd }|jj|kseJ ||jtj|| jj	dsuJ |jj
d s|jj
d sJ d S )Nr   r   r   r%   r$   )r   r   r8   r!   r   r   rn   r  rq   ro   r"   r	   )r=   r   r   r   r   r   test_optional_nonem  s       z#TestSharedMemory.test_optional_nonec                 C   s@   | j }t|f}| |tj|}|jj|ksJ | rJ d S rO   )r   rP   r   r   r8   r!   r   r   r   r   r   r   test_optional_from_2seq  s
   
z(TestSharedMemory.test_optional_from_2seqc                 C   sx   | j }t|t|d f}| |tj|}|jj|ksJ | r#J | |tjj|}|jj|ks4J | r:J d S r   )	r   rP   r   r   r8   r!   r   r   r	   r   r   r   r   test_optional_from_23seq  s   z)TestSharedMemory.test_optional_from_23seqc                 C   s   t j| j| jjd}|jd s|jd sJ |j}| |tj|}|d d |j	d d ks7J t
||j	fd|j	d d< |d d |j	d d   krYt jd| jjdks\J  J |j	|u scJ |jd sjJ |jd rqJ d S Nr   r%   r$   r   r   6   )rn   r   r   rq   ro   r"   r   r   inplacer!   r   r   r   r   r   test_inplace  s   ,<zTestSharedMemory.test_inplacec                 C   s:  | j  D ]}|| j u rqtj| j|jd}|jj |j ksJ |jj | j j us)J |jd s3|jd s5J |j}| |tj	|}|d d |j
d d ksVJ t||j
fd|j
d d< |d d |j
d d   krxtjd| j jdks{J  J |j
|u sJ |jd sJ |jd rJ |jj | j j u sJ qd S r  )rq   r   rn   r   r   ro   r"   r   r   r
  r!   r   )r=   r   rz   r   r   r   r   r   test_inplace_from_casttype  s&   
,$z+TestSharedMemory.test_inplace_from_casttypeN)"rR   rS   rT   r   fixturer   r   propertyr   r   r   r   markparametrizer   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r   r   r   r   r     s@    


		

r   r   )'r   sysr   platformr   numpyrn   numpy.testingr   r   numpy.core.multiarrayr   	_typeinfo r   r   _tidictrq   r   r   r   r#   r    r7   r   r   r   intpro   itemsizeclongdoublesystem	processorextendrl   r   r   r   r   r   r   r   <module>   sh    !





$Qo
