U
|
P ±d` ã @ sè d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZm Z d dl
|
m Z ddl mZ dae d Zef deedejd ejejfie Z d
|
d Zd d Zdd ZG dd dZe Zddddddddddd d d d d d!dgZddgiZed dg ed< ed dg ed< dged< dged< ed ddg ed< ed ddg ed< ed ddg ed< ed ddg ed< ed dg ed< ed dg ed< ed dg ed< ed d g ed < ed dd g ed < ed dd d g ed < ed d g ed < d ged < d!ged!< dged< e ¡ j j d"ks8e ¡ j jd kr¬ejd#kr¬e ¡ e !¡ fd$kr¬e "d%d&d'g¡ ed dd d d%g ed%< ed% d d&d'g ed'< ed d d&g ed&< G d(d) d)Z#G d*d+ d+Z$G d,d- d-Z%G d.d/ d/Z&dS )0é N)Úassert_Ú assert_equal)Útypeinfoé )ÚutilÚSTRINGÚ CHARACTERÚcé c C sb t ¡ st d¡ tdkr^d} tj t¡}t dddd¡t dd¡t dd ¡g}t
|
|| d
|
¡adS ) z6
|
Build the required testing extension module
|
|
zNo C compiler availableNz¼
|
config.add_extension('test_array_from_pyobj_ext',
|
sources=['wrapmodule.c', 'fortranobject.c'],
|
define_macros=[])
|
ÚtestsÚsrcZarray_from_pyobjz wrapmodule.czfortranobject.czfortranobject.hZtest_array_from_pyobj_ext) r Zhas_c_compilerÚpytestÚskipÚwrapÚosÚpathÚdirnameÚ__file__ZgetpathZbuild_module_distutils)Z config_codeÚdr © r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\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 }dD ]"}t | tt|d@ r| |¡ q|S )N)Ú
|
CONTIGUOUSÚFORTRANÚOWNDATAZ
|
ENSURECOPYZ ENSUREARRAYZALIGNEDZ
|
NOTSWAPPEDÚ WRITEABLEZWRITEBACKIFCOPYZ UPDATEIFCOPYZBEHAVEDZ
|
BEHAVED_ROZCARRAYZFARRAYr )ÚabsÚgetattrr Úappend)r ÚinfoZflagnamer r r r 6 s
|
r c @ s@ e Zd Zg fddZdd Zdd Zdd Zd d
|
Zd d Zd S )ÚIntentc C sP |d d
| _ d}|D ].}|dkr.|tjO }q|ttd| ¡ O }q|| _d S )Nr ÚoptionalZ F2PY_INTENT_)Ú intent_listr Z F2PY_OPTIONALr# Úupperr )Úselfr( r Úir r r Ú__init__N s zIntent.__init__c C s& | ¡ }|dkrd}| | j|g ¡S )NÚin_Úin)ÚlowerÚ __class__r( )r* Únamer r r Ú __getattr__X s zIntent.__getattr__c C s dd | j¡ S )Nz
|
intent(%s)ú,)Újoinr( ©r* r r r Ú__str__^ s zIntent.__str__c C s
|
d| j S )Nz
|
Intent(%r)©r( r5 r r r Ú__repr__a s zIntent.__repr__c G s |D ]}|| j kr dS qdS )NFTr7 )r* Únamesr1 r r r Ú is_intentd s
|
zIntent.is_intentc G s t | jt |ko| j| S ©N)Úlenr( r: )r* r9 r r r Úis_intent_exactj s zIntent.is_intent_exactN) Ú__name__Ú
|
__module__Ú __qualname__r, r2 r6 r8 r: r= r r r r r& M s
|
r& ÚBOOLÚBYTEZUBYTEÚSHORTÚUSHORTÚINTÚUINTÚLONGÚULONGZLONGLONGZ ULONGLONGÚFLOATÚDOUBLEZCFLOATZSTRING1ZSTRING5é Úwin32)ÚDarwinÚarmZ
|
LONGDOUBLEZCDOUBLEZ CLONGDOUBLEc @ sP e Zd Zi Zdd Zdd Zdd Zdd Zd d
|
Zd d Z d d Z
|
dd Z dS )ÚTypec C s t |tjrD|}d }t ¡ D ]&\}}t |ts |j|jkr |} qDq | j | ¡ d ¡}|d k rb|S t
|
| ¡}| |¡ || j| ¡ < |S r; ) Ú
|
isinstanceÚnpÚdtyper ÚitemsÚtypeÚ _type_cacheÚgetr) ÚobjectÚ__new__Ú_init)Úclsr1 Zdtype0Únr+ Úobjr r r rX ¸ s
|
|
z Type.__new__c C sð | ¡ | _| jdkr>t| j }ttd| _d| _t d¡| _n| j d¡rt| jd d
}ttd| _t
|
| jdd
pzd| _t d| j ¡| _n6t| j }ttd | j | _|j d
|
| _t |j ¡| _| j|j ksÜt|j | _ |j| _d S ) Nr Z
|
NPY_STRINGr r r r r ÚSZNPY_r
|
)r) ÚNAMEr r# r Útype_numÚelsizerQ rR Ú
|
startswithÚintÚbitsrT ÚnumÚAssertionErrorÚcharÚ dtypechar)r* r1 r% r r r rY È s$
|
|
|
|
z
|
Type._initc C s4 d| j d| j d| j d| j d| j d| j S )NzType(z )|type_num=z, dtype=z, type=z , elsize=z , dtypechar=)r^ r_ rR rT r` rg r5 r r r r8 ß s z Type.__repr__c s fddt j D S )Nc s g | ]} |¡qS r ©r0 ©Ú.0Ú_mr5 r r Ú
|
<listcomp>æ s z#Type.cast_types.<locals>.<listcomp>)Ú
|
_cast_dictr^ r5 r r5 r Ú
|
cast_typeså s zType.cast_typesc s fddt D S )Nc s g | ]} |¡qS r rh ri r5 r r rl é s z"Type.all_types.<locals>.<listcomp>)Ú _type_namesr5 r r5 r Ú all_typesè s zType.all_typesc C s: t | j j}g }tD ] }t | j|k r| t|¡ q|S r; ©r r^ Ú alignmentro r$ rO ©r* rc Útypesr1 r r r Ú smaller_typesë s zType.smaller_typesc C sF t | j j}g }tD ],}|| jkr$qt | j|kr| t|¡ q|S r; rq rs r r r Ú equal_typesó s
|
zType.equal_typesc C s: t | j j}g }tD ] }t | j|kr| t|¡ q|S r; rq rs r r r Ú larger_typesý s zType.larger_typesN) r> r? r@ rU rX rY r8 rn rp ru rv rw r r r r rO µ s
|
rO c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚArrayc
|
C s, d| j d| j d| j d| j d| j
|
S )NzArray(z, z)|arr=)rT ÚdimsÚintentr\ r r5 r r r r8 s zArray.__repr__c C sH || _ || _|| _t |¡| _|| _t |j |j
|
||j |¡| _ t | j tjsPtt | j ¡| _t|dkr| j d¡r¾|j tj@ st| j j d rt| j j d s¨t| jd tj@ r¼tnF|j tj@ rÎt| j j d sÞt| j j d rît| jd tj@ st|d kr d | _d | _d S | d¡r`t |tjsHttt |t |¡j| ¡ | _nHtjtj||j dj| | j d¡rdpd d
|
| _| jj |ks¨t| jj | j j d d | jj d sØt||ft | j¡| _t|dkrx| j d¡r>| jj d rt| jj d s&t| jd tj@ rxtn:| jj d sPt| jj d rbt| jd tj@ sxt| jd | jd kst| jd | jd ks¬t| jd dkrþ| jd | jd ksþtt| jd | jd | j ¡ | j ¡ f| jd dd
| jd dd
ks@tt| jd | jd f| jd | jd kstt| jd | jd t d| jd | jd t | jd |f| d¡rÄ| jd d | j j
|
ksötn2| jd d | j j
|
ksàt| !| j| j ¡sött | jtjrD|j
|
t"|j j
|
krD| d¡sD| jd dkrD| #¡ sDtd S )Nr r r r r Úcache©rR ÚCÚF)Úorderr! ©Úwriter é é é éþÿÿÿr Úcopy)$rT ry rz r ÚdeepcopyZobj_copyr\ r Úcallr_ r` r r rP rQ Úndarrayre r Úarr_attrr< r: Z F2PY_INTENT_Cr ZpyarrZ
|
pyarr_attrÚreprÚarrayZreshaperg rR ÚsetflagsÚtobytesr Ú arr_equalrO Úhas_shared_memory)r* Útypry rz r\ r r r r, s þ
|
þü* ÿ û zArray.__init__c C s |j |j krdS ||k ¡ S )NF)ÚshapeÚall)r* Zarr1Zarr2r r r r c s zArray.arr_equalc C s
|
t | jS r; )Ústrr r5 r r r r6 h s z Array.__str__c C s@ | j | jkrdS t| j tjs"dS t | j ¡}|d | jd kS )z6Check that created array shares data with input array.TFr )r\ r rP rQ r r r r )r* Zobj_attrr r r r k s zArray.has_shared_memoryN)r> r? r@ r8 r, r r6 r r r r r rx s
|
Wrx c @ s e Zd Zdd ZdS )Ú
|
TestIntentc C st t tjjdksttjj d¡s&ttjj d¡r8ttjj dd¡sLttjj dd¡s`ttj d¡rptd S )Nzintent(in,out)r r. )r rz r- Úoutre r r: r= r5 r r r Ú test_in_outv s zTestIntent.test_in_outN)r> r? r@ r r r r r r u s r c @ s" e Zd Zejddeddd Zedd Zedd Z d
|
d Z
|
d d Z ej dddg¡ej dddg¡ej dddg¡dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z d9d: Z d;d< Z d=d> Z!d?S )@ÚTestSharedMemoryTÚclass)ZautouseZscopeÚparamsc s" t j j_ fdd j_d S )Nc s t t j|||S r; )rx rO Úparam)r* ry rz r\ ©Úrequestr r Ú<lambda> s
|
ÿz-TestSharedMemory.setup_type.<locals>.<lambda>)rO r rZ rT r )r* r r r r Ú
|
setup_type s zTestSharedMemory.setup_typec C s. | j j d¡r&| j j}d| d| gS ddgS )Nr Ú1Ú2r r ©rT r^ ra r` ©r* r` r r r Únum2seq s zTestSharedMemory.num2seqc C sV | j j d¡rB| j j}d| d| d| gd| d| d| ggS dd d
|
gd d d ggS )Nr r r¡ Ú3Ú4Ú5Ú6r r r rK r r r¢ r£ r r r Únum23seq s ÿzTestSharedMemory.num23seqc C s$ | dgtj| j¡}| ¡ r td S )Nr )r rz r- r¤ r re ©r* Úar r r Útest_in_from_2seq s z"TestSharedMemory.test_in_from_2seqc C sx | j ¡ D ]h}tj| j|jd}| t| jgtj|¡}|j | j j krf|
|
¡ srt t | j j|jfq
|
|
|
¡ r
|
t q
|
d S ©Nr| ) rT rn rQ r r¤ rR r< rz r- r` r re r ©r* Útr\ r« r r r Útest_in_from_2casttype s z'TestSharedMemory.test_in_from_2casttyper ÚwÚror r} r~ ÚinpZ2seqZ23seqc C sd t | d| }tj|| jj|d}|j|dkd | |j|dkrJtjj pNtj|¡}|
|
¡ s`t dS )z5Test if intent(in) array can be passed without copiesrd ©rR r r± r r} N) r# rQ r rT rR r r rz r- r r re )r* r r r³ Úseqr\ r« r r r Útest_in_nocopy£ s ÿz TestSharedMemory.test_in_nocopyc
|
C s t j| j| jjd}| t| jgtj|¡}| ¡ s8t z| dgtj
|
j| j¡}W n2 t k
|
r } zt | d¡st W 5 d }~X Y n
|
X tdd S )Nr| r z6failed to initialize intent(inout|inplace|cache) arrayz,intent(inout) should have failed on sequence)rQ r r¤ rT rR r< rz Úinoutr re r- Ú TypeErrorr ra Ú SystemError)r* r\ r« Úmsgr r r Útest_inout_2seq¯ s ÿz TestSharedMemory.test_inout_2seqc
|
C sÐ t j| j| jjdd}t| jt| jd f}| |tjj|¡}| ¡ sLt
|
t j| j| jjdd}t| jt| jd f}z| |tjj|¡}W n2 t k
|
r } zt | d¡s² W 5 d }~X Y n
|
X tdd S )Nr~ r´ r r} z(failed to initialize intent(inout) arrayz2intent(inout) should have failed on improper array)rQ r r© rT rR r< rz r- r· r re Ú
|
ValueErrorr ra r¹ )r* r\ r r« rº r r r Útest_f_inout_23seq½ s ÿÿz#TestSharedMemory.test_f_inout_23seqc C sP t j| j| jjd}t| jt| jd f}| |tjjj |¡}|
|
¡ sLt d S ©Nr| r ) rQ r r© rT rR r< rz r- r r· r re ©r* r\ r r« r r r Útest_c_inout_23seqÏ s z#TestSharedMemory.test_c_inout_23seqc C sL | j ¡ D ]<}tj| j|jd}| t| jgtjj |¡}|
|
¡ r
|
t q
|
d S r ) rT rn rQ r r¤ rR r< rz r- r r re r® r r r Útest_in_copy_from_2casttypeÕ s z,TestSharedMemory.test_in_copy_from_2casttypec C s6 | t| jt| jd gtj| j¡}| ¡ r2td S ©Nr )r r< r© rz r- r re rª r r r Útest_c_in_from_23seqÛ s þz%TestSharedMemory.test_c_in_from_23seqc C sV | j ¡ D ]F}tj| j|jd}| t| jt| jd gtj|¡}| ¡ r
|
t
|
q
|
d S r¾ ) rT rn rQ r r© rR r< rz r- r re r® r r r Útest_in_from_23casttypeá s ÿz(TestSharedMemory.test_in_from_23casttypec C st | j ¡ D ]d}tj| j|jdd}| t| jt| jd gtj|¡}|j | j j krb|
|
¡ snt q
|
|
|
¡ r
|
t q
|
d S ©Nr~ r´ r ) rT rn rQ r r© rR r< rz r- r` r re r® r r r Útest_f_in_from_23casttypeè s ÿz*TestSharedMemory.test_f_in_from_23casttypec C st | j ¡ D ]d}tj| j|jd}| t| jt| jd gtjj |¡}|j
|
| j j
|
krb| ¡ snt q
|
| ¡ r
|
t q
|
d S r¾ ) rT rn rQ r r© rR r< rz r- r r` r re r® r r r Útest_c_in_from_23casttypeò s ÿz*TestSharedMemory.test_c_in_from_23casttypec C sZ | j ¡ D ]J}tj| j|jdd}| t| jt| jd gtjj |¡}|
|
¡ r
|
t q
|
d S rÅ ) rT rn rQ r r© rR r< rz r- r r re r® r r r Ú test_f_copy_in_from_23casttypeü s þz/TestSharedMemory.test_f_copy_in_from_23casttypec C sZ | j ¡ D ]J}tj| j|jd}| t| jt| jd gtjj j
|
|¡}| ¡ r
|
t q
|
d S r¾ ) rT rn rQ r r© rR r< rz r- r r r re r® r r r Ú test_c_copy_in_from_23casttype s þz/TestSharedMemory.test_c_copy_in_from_23casttypec C s> | j ¡ D ],}|j| j jkr q
|
tj| j|jd}t| jf}| |tj j
|
j |¡}| ¡ s^t | |tj j |¡}| ¡ s|t tj| j|jdd}| |tj j
|
j |¡}| ¡ s°t | |tj j |¡}| ¡ sØt t|jz | |tj j |d d d
¡}W n6 tk
|
r. } zt| d¡s W 5 d }~X Y q
|
X tdq
|
d S )Nr| r~ r´ éÿÿÿÿú(failed to initialize intent(cache) arrayz8intent(cache) should have failed on multisegmented array)rT rp r` rQ r r¤ rR r< rz r- r r{ r re r r¼ r ra r¹ )r* r¯ r\ r r« rº r r r Ú test_in_cache_from_2casttype s0 ÿÿz-TestSharedMemory.test_in_cache_from_2casttypec C s | j ¡ D ]}|jdkrq
|
|j| j jkr*q
|
tj| j|jd}t| jf}z| |t j
|
j |¡ W n2 t k
|
r } zt | d¡s W 5 d }~X Y q
|
X tdq
|
d S )Nr r| rË z1intent(cache) should have failed on smaller array)rT rp r^ r` rQ r r¤ rR r< rz r- r{ r¼ r ra r¹ )r* r¯ r\ r rº r r r Ú$test_in_cache_from_2casttype_failure) s"
|
ÿÿz5TestSharedMemory.test_in_cache_from_2casttype_failurec
|
C s¦ d}| |tjjd ¡}|jj|ks&td}| |tjjd ¡}|jj|ksLtd}z| |tjjd ¡}W n2 tk
|
r } zt| d¡s W 5 d }~X Y n
|
X t
|
dd S )N©r ©r r ©rÊ r ú2failed to create intent(cache|hide)|optional arrayz8intent(cache) should have failed on undefined dimensions) r rz r{ Úhider r re r¼ r ra r¹ ©r* r r« rº r r r Útest_cache_hidden< s" ÿÿz"TestSharedMemory.test_cache_hiddenc
|
C sf d}| |tjd ¡}|jj|ks$t| |jtj|| j j
|
d¡sDtd}| |tjd ¡}|jj|ksht| |jtj|| j j
|
d¡st|jj d r |jj d r¤td}| |tj jd ¡}|jj|ksÊt| |jtj|| j j
|
d¡sêt|jj d s|jj d s
|
td}z| |tjd ¡}W n6 t k
|
rX } zt| d¡sH W 5 d }~X Y n
|
X tdd S ) NrÎ r| rÏ r r rÐ rÑ z7intent(hide) should have failed on undefined dimensions)r rz rÒ r r re r rQ ÚzerosrT rR r r r¼ r ra r¹ rÓ r r r Ú test_hiddenP s2 ÿÿz TestSharedMemory.test_hiddenc C s d}| |tjd ¡}|jj|ks$t| |jtj|| j j
|
d¡sDtd}| |tjd ¡}|jj|ksht| |jtj|| j j
|
d¡st|jj d r |jj d r¤td}| |tj jd ¡}|jj|ksÊt| |jtj|| j j
|
d¡sêt|jj d s|jj d s
|
td S )NrÎ r| rÏ r r ) r rz r' r r re r rQ rÕ rT rR r r )r* r r« r r r Útest_optional_nonem s z#TestSharedMemory.test_optional_nonec C s@ | j }t|f}| |tj|¡}|jj|ks0t| ¡ r<td S r; ) r¤ r< r rz r' r r re r r¿ r r r Útest_optional_from_2seq s
|
|
z(TestSharedMemory.test_optional_from_2seqc C sx | j }t|t|d f}| |tj|¡}|jj|ks:t| ¡ rFt| |tjj |¡}|jj|ksht| ¡ rttd S r )
|
r© r< r rz r' r r re r r r¿ r r r Útest_optional_from_23seq s z)TestSharedMemory.test_optional_from_23seqc C sä t j| j| jjd}|jd s(|jd s,t|j}| |tj |¡}|d d |j
|
d d ksntt ||j
|
fd|j
|
d d< |d d |j
|
d d kr°t jd| jjdks¶n t|j
|
|ksÄt|jd sÒt|jd ràtd S ©Nr| r r r r é6 ) rQ r r© rT rR r re r rz Úinplacer r r¿ r r r Ú test_inplace s ,:z TestSharedMemory.test_inplacec C s@ | j ¡ D ].}|| j kr q
|
tj| j|jd}|jj |j ks@t|jj | j j k sTt|jd sh|jd slt|j}| |t j
|
|¡}|d d |j d d ks®tt ||j fd|j d d< |d d |j d d krðtjd| j jdksön t|j |kst|jd st|jd r&t|jj | j j ks
|
tq
|
d S rÚ ) rT rn rQ r r© rR re r r rz rÜ r r )r* r¯ r\ r r« r r r Útest_inplace_from_casttype s$
|
,$ÿz+TestSharedMemory.test_inplace_from_casttypeN)"r> r? r@ r Zfixturero r Úpropertyr¤ r© r¬ r° ÚmarkZ 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 Úsysr Úplatformr ÚnumpyrQ Z numpy.testingr r Znumpy.core.multiarrayr Z _typeinfoÚ r r Z_tiÚdictrT rd rr r r r r& rz ro rm ZintprR ÚitemsizeZ clongdoubleÚsystemÚ processorÚextendrO rx r r r r r r Ú<module> s ÿþ!ï
|
|
|
|
|
|
$ÿþü
|
ý
|
Qo
|
