U
|
P ±dDR ã @ s d dl Z d dlZd dlmZmZmZ d dlZd dlm Z G dd de j
|
Z G dd de j
|
Z G dd d e j
|
Z G d
|
d d e j
|
ZdS ) é N)Úassert_array_equalÚ assert_equalÚ assert_raises)Úutilc 7 @ s^ e Zd ZdZdZdddgZdZeD ]ÀZeZe dd
|
ee¡Z ee d e d
|
e d e d e d
|
e d e d e d e de d e d e de de de de d e de de de de d e de de de de d3¡7 Zq"ej de¡dd Zej ded d
¡d d Zej de¡d d! Zej de¡d"d# Zej de¡d$d% Zd S )&ÚTestCharacterStringú.f90Ztest_character_stringÚ1Ú3ÚstarÚ ú(*))r
|
z
|
|
subroutine Ú_input_z (c, o, n)
|
character*zß, intent(in) :: c
|
integer n
|
!f2py integer, depend(c), intent(hide) :: n = slen(c)
|
integer*1, dimension(n) :: o
|
!f2py intent(out) o
|
o = transfer(c, o)
|
end subroutine Ú_output_zÍ, intent(out) :: c
|
integer n
|
integer*1, dimension(n), intent(in) :: o
|
!f2py integer, depend(o), intent(hide) :: n = len(o)
|
c = transfer(o, c)
|
end subroutine Ú _array_input_z;(c, o, m, n)
|
integer m, i, n
|
character*aH , intent(in), dimension(m) :: c
|
!f2py integer, depend(c), intent(hide) :: m = len(c)
|
!f2py integer, depend(c), intent(hide) :: n = f2py_itemsize(c)
|
integer*1, dimension(m, n), intent(out) :: o
|
do i=1,m
|
o(i, :) = transfer(c(i), o(i, :))
|
end do
|
end subroutine Ú_array_output_z!(c, o, m, n)
|
character*a , intent(out), dimension(m) :: c
|
integer n
|
integer*1, dimension(m, n), intent(in) :: o
|
!f2py character(f2py_len=n) :: c
|
!f2py integer, depend(o), intent(hide) :: m = len(o)
|
!f2py integer, depend(o), intent(hide) :: n = shape(o, 1)
|
do i=1,m
|
c(i) = transfer(o(i, :), c(i))
|
end do
|
end subroutine Ú_2d_array_input_zH(c, o, m1, m2, n)
|
integer m1, m2, i, j, n
|
character*aÍ , intent(in), dimension(m1, m2) :: c
|
!f2py integer, depend(c), intent(hide) :: m1 = len(c)
|
!f2py integer, depend(c), intent(hide) :: m2 = shape(c, 1)
|
!f2py integer, depend(c), intent(hide) :: n = f2py_itemsize(c)
|
integer*1, dimension(m1, m2, n), intent(out) :: o
|
do i=1,m1
|
do j=1,m2
|
o(i, j, :) = transfer(c(i, j), o(i, j, :))
|
end do
|
end do
|
end subroutine z
|
Úlengthc C s\ ddi ||¡}t| j| jd | }dddd| }t||tjttt |dd d S )
|
Nr r
|
r ÚaÚabcÚabcdeabcdeabcde©r r r
|
Úu1©Údtype)
|
ÚgetÚgetattrÚmoduleÚfprefixr ÚnpÚarrayÚlistÚmapÚord©Úselfr ÚfsuffixÚfr © r' úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/f2py/tests/test_character.pyÚ
|
test_inputL s z TestCharacterString.test_inputNéÿÿÿÿc C sR |}t | j| jd | }ddd| }t|tjttt|dd| ¡ d S )Nr r r )r r r r )
|
r r r r r r r r! r" Úencoder# r' r' r( Ú test_outputU s ÿz TestCharacterString.test_outputc C sn |}t | j| jd | }tjdddd| dddd| gd d
|
}tjd d |D d d
|
}t||| d S )Nr r r r r ÚAÚABCÚABCDEABCDEABCDEÚSr c S s g | ]}d d |D qS )c S s g | ]}|qS r' r' ©Ú.0Úcr' r' r( Ú
|
<listcomp>h s zCTestCharacterString.test_array_input.<locals>.<listcomp>.<listcomp>r' ©r2 Úsr' r' r( r4 h s z8TestCharacterString.test_array_input.<locals>.<listcomp>r ©r r r r r r ©r$ r r% r& r Úexpectedr' r' r( Útest_array_input_ s ÿþz$TestCharacterString.test_array_inputc C sn |}t | j| jd | }tjdddd| dddd| gd d
|
}tjd d |D d d
|
}t||| d S )Nr r r r r r- r. r/ r0 r c S s g | ]}d d |D qS )c S s g | ]}|qS r' r' r1 r' r' r( r4 t s zDTestCharacterString.test_array_output.<locals>.<listcomp>.<listcomp>r' r5 r' r' r( r4 t s z9TestCharacterString.test_array_output.<locals>.<listcomp>r r7 )r$ r r% r& r9 r r' r' r( Útest_array_outputk s ÿþz%TestCharacterString.test_array_outputc C s |}t | j| jd | }tjdddd| dddd| gd d
|
d d| d d dd| ggdd}tjdd |D dd d}t||| d S )Nr r r r r r- r. r/ r& ZfghZfghijfghijfghijÚFZFGHZFGHIJFGHIJFGHIJr0 r c S s g | ]}d d |D qS )c S s g | ]}d d |D qS )c S s g | ]}|qS r' r' r1 r' r' r( r4 s zQTestCharacterString.test_2d_array_input.<locals>.<listcomp>.<listcomp>.<listcomp>r' )r2 Úitemr' r' r( r4 s zFTestCharacterString.test_2d_array_input.<locals>.<listcomp>.<listcomp>r' )r2 Úrowr' r' r( r4 s z;TestCharacterString.test_2d_array_input.<locals>.<listcomp>r ©r Úorderr7 r8 r' r' r( Útest_2d_array_inputw s ÿÿþü ÿz'TestCharacterString.test_2d_array_input)Ú__name__Ú
|
__module__Ú __qualname__Úsuffixr Z length_listÚcoder r% Údictr ÚclengthÚtextwrapÚdedentÚpytestÚmarkÚ parametrizer) r, r: r; rA r' r' r' r( r s
|
þþýþþþþýþþþþýþþþþýþþþþýþþ9
|
|
|
|
r c 3 @ s¤ e Zd ZdZdZe de de de de de de d e d
|
e d e d e d e de de de de de de de de de de de de de d1¡Zej
|
d d d g¡d d Z d!d" Z ej
|
d d d d#g¡d$d% Z d&d' Zej
|
d d d d#g¡d(d) Zd*d+ Zd,d- Zd.d/ Zej
|
d d d g¡d0d1 Zd2d3 Zej
|
d d d g¡d4d5 Zej
|
d d d g¡d6d7 Zd8d9 Zej d:¡d;d< Zd=d> Zd?S )@Ú TestCharacterr Ztest_characterú
|
subroutine z_input(c, o)
|
character, intent(in) :: c
|
integer*1 o
|
!f2py intent(out) o
|
o = transfer(c, o)
|
end subroutine z_input
|
|
subroutine z_output(c, o)
|
character :: c
|
integer*1, intent(in) :: o
|
!f2py intent(out) c
|
c = transfer(o, c)
|
end subroutine z_output
|
|
subroutine z_input_output(c, o)
|
character, intent(in) :: c
|
character o
|
!f2py intent(out) o
|
o = c
|
end subroutine z!_input_output
|
|
subroutine z_inout(c, n)
|
character :: c, n
|
!f2py intent(in) n
|
!f2py intent(inout) c
|
c = n
|
end subroutine z_inout
|
|
function z
_return(o) result (c)
|
character :: c
|
character, intent(in) :: o
|
c = transfer(o, c)
|
end function z_return
|
|
subroutine zè_array_input(c, o)
|
character, intent(in) :: c(3)
|
integer*1 o(3)
|
!f2py intent(out) o
|
integer i
|
do i=1,3
|
o(i) = transfer(c(i), o(i))
|
end do
|
end subroutine z _array_input
|
|
subroutine a' _2d_array_input(c, o)
|
character, intent(in) :: c(2, 3)
|
integer*1 o(2, 3)
|
!f2py intent(out) o
|
integer i, j
|
do i=1,2
|
do j=1,3
|
o(i, j) = transfer(c(i, j), o(i, j))
|
end do
|
end do
|
end subroutine z#_2d_array_input
|
|
subroutine zØ_array_output(c, o)
|
character :: c(3)
|
integer*1, intent(in) :: o(3)
|
!f2py intent(out) c
|
do i=1,3
|
c(i) = transfer(o(i), c(i))
|
end do
|
end subroutine z!_array_output
|
|
subroutine zÊ_array_inout(c, n)
|
character :: c(3), n(3)
|
!f2py intent(in) n(3)
|
!f2py intent(inout) c(3)
|
do i=1,3
|
c(i) = n(i)
|
end do
|
end subroutine z _array_inout
|
|
subroutine a _2d_array_inout(c, n)
|
character :: c(2, 3), n(2, 3)
|
!f2py intent(in) n(2, 3)
|
!f2py intent(inout) c(2. 3)
|
integer i, j
|
do i=1,2
|
do j=1,3
|
c(i, j) = n(i, j)
|
end do
|
end do
|
end subroutine z!_2d_array_inout
|
|
function z»_array_return(o) result (c)
|
character, dimension(3) :: c
|
character, intent(in) :: o(3)
|
do i=1,3
|
c(i) = o(i)
|
end do
|
end function z _array_return
|
|
function z_optional(o) result (c)
|
character, intent(in) :: o
|
!f2py character o = "a"
|
character :: c
|
c = o
|
end function z_optional
|
r r3 ÚS1c C s¨ t | j| jd }t|tjd|dtd t|tjd|dtd t|tjdg|dtd t|tjd|dtd t|tjdgg|dtd d S )NÚ_inputr r ó ar )r r r r r r r" ©r$ r r& r' r' r( r) ò s zTestCharacter.test_inputc
|
C s° t | j| jd }t|dtd t|dtd t|dd t|dd t|dd t|dtd t|d td t|dgtd t|t d¡td t|t dg¡td t d¡}t||td t dg¡}t||td z |g W n6 tk
|
rF } zt| d
|
¡s6 W 5 d }~X Y nX t
|
|j d z |d W n6 t k
|
r } zt| d ¡s W 5 d }~X Y nX t
|
|j dd S )NrQ r rR r r ó ó Úabs abz got 0-listz! should have failed on empty listéa z got int instancez should have failed on int value) r r r r r" r r Ú
|
IndexErrorÚstrÚendswithÚ SystemErrorrB Ú TypeError©r$ r& r Úmsgr' r' r( Útest_input_variaü s6
|
z TestCharacter.test_input_variaÚU1c C sz t | j| jd }t|tjdddg|dtjtttddd t|tjdd d
|
g|dtjtttddd d S ) NÚ _array_inputr Úbr3 r r Úi1rR ó bó c) r r r r r r r r! r" rS r' r' r( r: s ÿÿz TestCharacter.test_array_inputc
|
C s¾ t | j| jd }t|dddgtjtttddd t|dd d
|
gtjtttddd z|dddd g W n2 t k
|
r¨ } zt
|
| d ¡s W 5 d }~X Y nX t |j d d S )Nra r rb r3 r rc r rR rd re Údú)th dimension must be fixed to 3 but got 4ú" should have failed on wrong input)r r r r r r r r! r" Ú
|
ValueErrorrY rZ r[ rB )r$ r& r^ r' r' r( Útest_array_input_varia( s" ÿÿÿ
|
ÿz$TestCharacter.test_array_input_variac C s^ t | j| jd }tjdddgdddgg|dd }| |d
|
krDtjntj¡}t||| d S ) NZ_2d_array_inputr rb r3 rf Úer& r< r? r` ) r r r r r ÚviewZuint32Zuint8r )r$ r r& r r9 r' r' r( rA 9 s ÿ ÿz!TestCharacter.test_2d_array_inputc C s6 t | j| jd }t|tdd t|dd d S )NÚ_outputrR r rU )r r r r r" ©r$ r& r' r' r( r, B s zTestCharacter.test_outputc C s< t | j| jd }t|tttdtjtddd d S )NZ _array_outputr rP r ) r r r r r r! r" r r rn r' r' r( r; H s ÿz TestCharacter.test_array_outputc C s@ t | j| jd }t|dd t|dd t|dd d S )NZ _input_outputrR r r rU ©r r r r rn r' r' r( Útest_input_outputN s z TestCharacter.test_input_outputc C sª t | j| jd }tjtd|d}||d t|tjtd|jd ||dd
d t|tjtd|jd tjdg|d}||d t|tjdg|jd d S ) NÚ_inoutr r r- ÚAbcé ÚBZABc)r r r r r r r r )r$ r r& r r' r' r( Ú
|
test_inoutU s
|
|
zTestCharacter.test_inoutc
|
C sÈ t | j| jd }tjddd}||d t|tjd|jd tjdgdd}||d t|tjdg|jd z|dd W n2 tk
|
r² } zt| d¡s¢ W 5 d }~X Y nX t
|
|j dd S ) Nrq r ZS3r r- rr z
|
got 3-strz should have failed on str value) r r r r r r r ri rY rZ r[ rB r] r' r' r( Útest_inout_variac s
|
|
z TestCharacter.test_inout_variac
|
C sF t | j| jd }tjdddg|dd}tjddd g|dd}||| t|| tjddd d
|
g|d }||d d
| t|tjddddg|d tjddd gg|dd}||| t|tjdddgg|d tjddd d
|
g|dd}z||| W n6 tk
|
r0 } zt| d ¡s W 5 d }~X Y nX t |j
|
dd S )NZ _array_inoutr- rt ÚCr< r? r rb r3 rf r rs rg rh ) r r r r r r ri rY rZ r[ rB )r$ r r& Únr r^ r' r' r( Útest_array_inoutu s,
|
|
|
ÿ
|
ÿz TestCharacter.test_array_inoutc C sj t | j| jd }tjdddgdddgg|dd}tjd d
|
d gd d dgg|dd}||| t|| d S )NZ_2d_array_inoutr- rt rw ÚDÚEr< r? r rb r3 rf rk r& r7 )r$ r r& rx r r' r' r( Útest_2d_array_inout s ÿ þ ÿ þ
|
z!TestCharacter.test_2d_array_inoutc C s$ t | j| jd }t|dd d S )NZ_returnr rR ro rn r' r' r( Ú test_return s zTestCharacter.test_returnz*fortran function returning array segfaultsc C s6 t | j| jd }tjtddd}t||| d S )NZ _array_returnr rP r )r r r r r r r )r$ r& r r' r' r( Útest_array_return¢ s z TestCharacter.test_array_returnc C s0 t | j| jd }t| d t|dd d S )NZ _optionalrR ó Bro rn r' r' r( Ú test_optional© s zTestCharacter.test_optionalN)rB rC rD rE r rI rJ rF rK rL rM r) r_ r: rj rA r, r; rp ru rv ry r| r} Úskipr~ r r' r' r' r( rN s ÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿ
|
g
|
#
|
|
|
|
|
|
|
rN c @ s¢ e Zd ZdZdZe de de de de de de d e d
|
e d e d e d e de d¡Zdd Zdd Z dd Z
|
dd Z e j dddg¡dd Zd S ) ÚTestMiscCharacterr Ztest_misc_characterrO a( _gh18684(x, y, m)
|
character(len=5), dimension(m), intent(in) :: x
|
character*5, dimension(m), intent(out) :: y
|
integer i, m
|
!f2py integer, intent(hide), depend(x) :: m = f2py_len(x)
|
do i=1,m
|
y(i) = x(i)
|
end do
|
end subroutine z _gh18684
|
|
subroutine z«_gh6308(x, i)
|
integer i
|
!f2py check(i>=0 && i<12) i
|
character*5 name, x
|
common name(12)
|
name(i + 1) = x
|
end subroutine z_gh6308
|
|
subroutine zÐ_gh4519(x)
|
character(len=*), intent(in) :: x(:)
|
!f2py intent(out) x
|
integer :: i
|
do i=1, size(x)
|
print*, "x(",i,")=", x(i)
|
end do
|
end subroutine z _gh4519
|
|
pure function aq _gh3425(x) result (y)
|
character(len=*), intent(in) :: x
|
character(len=len(x)) :: y
|
integer :: i
|
do i = 1, len(x)
|
j = iachar(x(i:i))
|
if (j>=iachar("a") .and. j<=iachar("z") ) then
|
y(i:i) = achar(j-32)
|
else
|
y(i:i) = x(i:i)
|
endif
|
end do
|
end function z_gh3425
|
|
subroutine aO _character_bc_new(x, y, z)
|
character, intent(in) :: x
|
character, intent(out) :: y
|
!f2py character, depend(x) :: y = x
|
!f2py character, dimension((x=='a'?1:2)), depend(x), intent(out) :: z
|
character, dimension(*) :: z
|
!f2py character, optional, check(x == 'a' || x == 'b') :: x = 'a'
|
!f2py callstatement (*f2py_func)(&x, &y, z)
|
!f2py callprotoargument character*, character*, character*
|
if (y.eq.x) then
|
y = x
|
else
|
y = 'e'
|
endif
|
z(1) = 'c'
|
end subroutine z%_character_bc_new
|
|
subroutine aG _character_bc_old(x, y, z)
|
character, intent(in) :: x
|
character, intent(out) :: y
|
!f2py character, depend(x) :: y = x[0]
|
!f2py character, dimension((*x=='a'?1:2)), depend(x), intent(out) :: z
|
character, dimension(*) :: z
|
!f2py character, optional, check(*x == 'a' || x[0] == 'b') :: x = 'a'
|
!f2py callstatement (*f2py_func)(x, y, z)
|
!f2py callprotoargument char*, char*, char*
|
if (y.eq.x) then
|
y = x
|
else
|
y = 'e'
|
endif
|
z(1) = 'c'
|
end subroutine z_character_bc_old
|
c C s: t | j| jd }tjddgdd}||}t|| d S )NZ_gh18684ÚabcdeZfghijÚS5r r7 )r$ r& ÚxÚyr' r' r( Ú test_gh18684 s z TestMiscCharacter.test_gh18684c C s~ t | j| jd }t| jjjjt d¡ tt| jjjd |dd t| jjjd d |dd t| jjjd d d S )
|
NZ_gh6308r é r r s abcdeZ12345é s 12345) r r r r Z_BLNK_Únamer r Úlenrn r' r' r( Ú test_gh6308
|
s
|
|
z TestMiscCharacter.test_gh6308c
|
C sÌ t | j| jd }dtdt d¡dfdtdt d¡dftjdd d
|
gdd td t d¡dfdd d gtd t d¡dfddgtdt d¡dffD ]2\}}||}| ¡ D ]\}}tt ||| q¬qd S )NZ_gh4519r r' rP )Úshaper ÚtextZS4r Ú2r r )é Z34ZS2r )é ) r r r rG r r r Úitemsr )r$ r& r
r9 ÚrÚkÚvr' r' r( Ú test_gh4519 s ÿÿù z TestMiscCharacter.test_gh4519c C s@ t | j| jd }t|dd t|dd t|dd d S )NZ_gh3425ZabCs ABCr rT ZabC12ds ABC12Dro rn r' r' r( Ú test_gh3425% s z TestMiscCharacter.test_gh3425ÚstateÚnewÚoldc sr t | j| jd | \}}t|d tt|d d\}}t|d tt|d tt fdd d S )NZ_character_bc_rR rs rd r c s dS )Nre r' r' ©r& r' r( Ú<lambda>: rT z5TestMiscCharacter.test_character_bc.<locals>.<lambda>)r r r r r r Ú Exception)r$ r r3 r r' r r( Útest_character_bc. s
|
|
|
z#TestMiscCharacter.test_character_bcN)rB rC rD rE r rI rJ rF r r r r rK rL rM r r' r' r' r( r ° sB ÿÿÿÿÿÿÿÿÿÿÿÿ
|
M r c @ s> e Zd Ze dddd¡gZejjdd Z ejjdd Z
|
d S )
|
ÚTestStringScalarArrÚtestsÚsrcÚstringzscalar_string.f90c C sB | j jj| j jjfD ](}d}|j|ks*td}|j|kstqd S )Nr' z|S8)r Ú string_testr¢ Zstring77r ÚAssertionErrorr ©r$ Úoutr9 r' r' r( Ú test_char@ s ÿz TestStringScalarArr.test_charc C sB | j jj| j jjfD ](}d}|j|ks*td}|j|kstqd S )N)r é z|S12)r r£ ZstrarrZstrarr77r r¤ r r¥ r' r' r( Ú test_char_arrI s ÿz!TestStringScalarArr.test_char_arrN) rB rC rD r ZgetpathÚsourcesrK rL Zslowr§ r© r' r' r' r( r = s
|
|
r )rK rI Z numpy.testingr r r Únumpyr Znumpy.f2py.testsr ZF2PyTestr rN r r r' r' r' r( Ú<module> s ~ ,
|
