o
    zqf!]                     @   s  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 d dlZd dl	m
Z
mZ d dlmZ ddlmZ daed7dedieZd	d
 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r?ejd'kr?e  e! fd(kr?e"g d) ed g d* ed+< ed+ g d, ed-< ed! d"d.g ed.< G d/d0 d0Z#G d1d2 d2Z$G d3d4 d4Z%G d5d6 d6Z&dS )8    N)Path)assert_assert_equal)c_names_dict   )util	CHARACTERcc                  C   s   t t jd } | d S )Nsrcarray_from_pyobj)r   __file__resolveparent)testroot r   e/root/parts/websockify/install/lib/python3.10/site-packages/numpy/f2py/tests/test_array_from_pyobj.pyget_testdir   s   r   c                  C   s*   t du rt d g} tj| dda dS dS )z6
    Build the required testing extension module

    Nzwrapmodule.ctest_array_from_pyobj_ext)module_name)wrapr   r   build_meson)r
   r   r   r   setup_module   s
   r   c                 C   s   t | d }t|S )N   )r   array_attrsflags2names)arrflagsr   r   r   
flags_info)   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   .   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_OPTIONALr-   upperr   )selfr4   r   ir   r   r   __init__F   s   
zIntent.__init__c                 C   s&   |  }|dkr
d}| | j|g S )Nin_in)lower	__class__r4   )r7   namer   r   r   __getattr__P   s   zIntent.__getattr__c                 C   s   dd | j S )Nz
intent(%s),)joinr4   r7   r   r   r   __str__V   s   zIntent.__str__c                 C   s
   d| j  S )Nz
Intent(%r)r4   rB   r   r   r   __repr__Y      
zIntent.__repr__c                 G   s   |D ]
}|| j vr dS qdS )NFTrD   )r7   namesr>   r   r   r   	is_intent\   s
   
zIntent.is_intentc                 G   s   t | jt |ko| j| S N)lenr4   rH   )r7   rG   r   r   r   is_intent_exactb   s   zIntent.is_intent_exactN)	__name__
__module____qualname__r9   r?   rC   rE   rH   rK   r   r   r   r   r1   E   s    
r1   )BOOLBYTEUBYTESHORTUSHORTINTUINTLONGULONGLONGLONG	ULONGLONGFLOATDOUBLECFLOATSTRING1STRING5r   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   )rU   rZ   r[   r[   r\   r]   r^         win32)Darwinarm)
LONGDOUBLECDOUBLECLONGDOUBLE)rW   rZ   r[   rd   rd   )r\   re   rf   rf   re   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 rI   )
isinstancenpdtyper   itemstype_type_cachegetr6   object__new___init)clsr>   dtype0nr8   objr   r   r   rp      s   

zType.__new__c                 C   s   |  | _| jdkrt| j }ttd| _d| _td| _nE| 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| _t|j| _| j|jkslJ |j| _|j| _d S )
Nr   
NPY_STRINGr   r	   STRINGr   r   SNPY_)r6   NAMEr   r-   r   type_numelsizeri   rj   
startswithintitemsizerl   numchar	dtypechar)r7   r>   r/   r   r   r   rq      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=)rz   r{   rj   rl   r|   r   rB   r   r   r   rE      s   zType.__repr__c                    s    fddt  j D S )Nc                       g | ]}  |qS r   r=   .0_mrB   r   r   
<listcomp>       z#Type.cast_types.<locals>.<listcomp>)
_cast_dictrz   rB   r   rB   r   
cast_types   s   zType.cast_typesc                    s    fddt D S )Nc                    r   r   r   r   rB   r   r   r      r   z"Type.all_types.<locals>.<listcomp>)_type_namesrB   r   rB   r   	all_types   s   zType.all_typesc                 C   s:   t | j j}g }tD ]}t | j|k r|t| q
|S rI   r   rz   	alignmentr   r.   rg   r7   bitstypesr>   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 rI   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 rI   r   r   r   r   r   larger_types   r   zType.larger_typesN)rL   rM   rN   rm   rp   rq   rE   r   r   r   r   r   r   r   r   r   rg      s    
rg   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=)rl   dimsintentru   r   rB   r   r   r   rE      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rj   CF)orderr%   writer             r   copy)#rl   r   r   r   deepcopyobj_copyru   r   callr{   r|   r   r   rh   ri   ndarrayr   arr_attrrJ   rH   F2PY_INTENT_Cr   pyarr
pyarr_attrreprarrayreshaper   rj   setflagstobytesr   	arr_equalrg   has_shared_memory)r7   typr   r   ru   r   r   r   r9     s   
*zArray.__init__c                 C   s   |j |j krdS ||k S )NF)shapeall)r7   arr1arr2r   r   r   r   [  s   zArray.arr_equalc                 C   s
   t | jS rI   )strr   rB   r   r   r   rC   `  rF   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   )ru   r   rh   ri   r   r   r   r   )r7   obj_attrr   r   r   r   c  s   zArray.has_shared_memoryN)rL   rM   rN   rE   r9   r   rC   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	   r;   )r   r   r:   outr	   rH   rK   rB   r   r   r   test_in_outn  s   zTestIntent.test_in_outN)rL   rM   rN   r   r   r   r   r   r   m  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 rI   )r   rg   param)r7   r   r   ru   requestr   r   <lambda>|  s    z-TestSharedMemory.setup_type.<locals>.<lambda>)rg   r   rr   rl   r   )r7   r   r   r   r   
setup_typey  s   zTestSharedMemory.setup_typec                 C   s.   | j jdr| j j}d| d| gS ddgS )Nrw   12r   r   rl   rz   r}   r|   r7   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 )
Nrw   r   r   3456)r   r   r   )r_   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   r7   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   )rl   r   ri   r   r   rj   rJ   r   r:   r|   r   r   r7   tru   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   rj   r   r   r   r   N)r-   ri   r   rl   rj   r   r   r   r:   r	   r   )r7   r   r   r   seqru   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)ri   r   r   rl   rj   rJ   r   inoutr   r:   	TypeErrorr   r}   SystemError)r7   ru   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)ri   r   r   rl   rj   rJ   r   r:   r   r   
ValueErrorr   r}   r   )r7   ru   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   )ri   r   r   rl   rj   rJ   r   r:   r	   r   r   r7   ru   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   )rl   r   ri   r   r   rj   rJ   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   rJ   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   )
rl   r   ri   r   r   rj   rJ   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   )rl   r   ri   r   r   rj   rJ   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   )rl   r   ri   r   r   rj   rJ   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   )rl   r   ri   r   r   rj   rJ   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   )rl   r   ri   r   r   rj   rJ   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)rl   r   r|   ri   r   r   rj   rJ   r   r:   r	   r   r   r   r   r   r}   r   )r7   r   ru   r   r   r   r   r   r   test_in_cache_from_2casttype  s8   z-TestSharedMemory.test_in_cache_from_2casttypec                 C   s   | j  D ]`}|jdkrq|j| j jkrqt|jtj}|r.t| j	d t
|jjkr.qtj| j	|jd}t| j	f}z| |tjj| W td tye } zt|ds[ W Y d }~qd }~ww d S )Nrw   r   r   r   z1intent(cache) should have failed on smaller array)rl   r   rz   r|   ri   
issubdtyperj   integerr~   r   iinfomaxr   rJ   r   r:   r   r   r   r}   r   )r7   r   is_intru   r   r   r   r   r   $test_in_cache_from_2casttype_failure!  s0   
 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   r7   r   r   r   r   r   r   test_cache_hidden8  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   ri   zerosrl   rj   r   r	   r   r   r}   r   r  r   r   r   test_hiddenL  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   r2   r   r   r   ri   r  rl   rj   r   r	   )r7   r   r   r   r   r   test_optional_nonei  s       z#TestSharedMemory.test_optional_nonec                 C   s@   | j }t|f}| |tj|}|jj|ksJ | rJ d S rI   )r   rJ   r   r   r2   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   rJ   r   r   r2   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   )ri   r   r   rl   rj   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
  )rl   r   ri   r   r   rj   r   r   r   r  r   r   )r7   r   ru   r   r   r   r   r   test_inplace_from_casttype  s&   
,$z+TestSharedMemory.test_inplace_from_casttypeN)"rL   rM   rN   pytest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   w  s@    


		

r   r   )'ossysr   platformr  pathlibr   numpyri   numpy.testingr   r   numpy._core._type_aliasesr   _c_names_dict r   r   dictrj   r   r   r   r   r1   r   r   r   intpr   clongdoubler   system	processorextendrg   r   r   r   r   r   r   r   <module>   sj    !





$Qo
