U
|
O ±düY ã @ s d dl Z d dlZd dlmZ d dlZd dlmZmZm Z m
|
Z
|
m Z m Z dd Z ej dddg¡d d
|
Zej d e d ¡e d d dd g¡e d¡ d d ¡f¡dd Zdd Zej dddg¡dd Zej ddddd g¡d d Zd d Zej d!ejd" ejd# ejd$ ¡d%d& Zej d'd(gejej fej ej fe d)¡e d*¡fg¡d+d, Z ej d'd(gejej fej ej fe d)¡e d*¡fg¡d-d. Z ej d/e!e"ej#ej g¡ej d0ej$ej%ej&g¡d1d2 Z'ej d'ej(d3 ¡d4d5 Z)d6d7 Z*d8d9 Z+d:d; Z,d<d= Z-d>d? Z.d@dA Z/ej dBdCdDgdEdFdGge dFdGgdFdGgg¡fdHdIdJge dIdJgdIdJgdIdJgg¡fg¡dKdL Z0dMdN Z1dS )Oé N)Úrational)Úassert_Ú assert_equalÚassert_array_equalÚ assert_raisesÚ assert_warnsÚ HAS_REFCOUNTc C s t t} t dtj¡}t |}ttj|tjd| tr\t |¡}t |¡ t|t |¡ ttjd tjdtjtj
|
tjd tr®t | ¡}tjd tjd t|t | ¡ ttjdtjdtjdtjd trt tj¡}tjtjdtjdtjd t|t tj¡ t t df¡}t t df¡}t t df¡}ttjdtjdtjdtjd tt d¡j | ttjdt dj | ttjd|dt d ¡ ttjd
|
|dtjd|d t t df¡}t t df¡}t t df¡} ttjd tjdtjdtjd tt d ¡j | ttjd t dj | ttjd |dt t d ¡ ttjd | dtjd| d ttdt}
|
tt|
|
d tjtdtjd} t| dddg t| j j tjk t td¡} t| dddg t| j j tjk tjdtjd} t dtft| jd} ttj| tjd| tjddgdtfd tfd t fgd} t dtft| jd } tt t | ¡jt | j t dtftd d d! } ttj| tjdt dtj¡ d"}ttjD ] }|g}qÆt |¡ t t tj|gtjd ttjd gd# tjdtj d$tj
|
tjd ttjd ggd# tjdtj d%tj
|
tjd ttjd gd# gtjdtj d&tj
|
tjd ttjd gd# gd# tjdtj d'tj
|
tjd ttjdgd# tjdtjd$tjd ttjdggd# tjdtjd%tjd ttjdgd# gtjdtjd&tjd ttjdgd# gd# tjdtjd'tjd ttjd(tjdtj d$tj
|
tjd ttjd)gd# tjdtj d%tj
|
tjd ttjd(gtjdtj d&tj
|
tjd ttjd(gd# tjdtj d'tj
|
tjd ttjd*tjdtjd$tjd ttjd+gd# tjdtjd%tjd ttjd*gtjdtjd&tjd ttjd*gd# tjdtjd'tjd d S ),N)é r ©Údtypeç ð?© é é é s 1.0s 1.ó 1z1.0z1.Ú1Ú__dict__Úgetg H@g G@g H@é1 é. é0 ç Y@Úo)Ú__array_interface__)r ç @ZHello)r ç @ZWorldZf0Úf1Úf2)Ú__array_struct__c W s t jdt jdS )Nr r
|
)ÚnpÚarrayÚfloat64)Úxr r úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_api.pyÚ<lambda>Z ó z"test_array_array.<locals>.<lambda>)Z __array__ç ø?é
|
)r( )r( r )r r( )r( r( )
|
NNNNNNNNNN©N)
|
r r r r r r r r r r )r ) ÚtypeÚobjectr Úonesr" r r! r ÚsysÚ getrefcountÚnanr ÚbytesÚstrÚgetattrÚ __builtins__r ÚhasattrÚ
|
memoryviewÚuint8Údictr ÚintÚfloatr ÚdataÚrangeZMAXDIMSr Ú
|
ValueErrorÚfull)ZtobjZones11ZtndarrayZ old_refcountZS2ZS3ZS5ZU2ZU3ZU5ÚbuiltinsZdatÚar ÚnestedÚir r r$ Útest_array_array sî
|
|
ÿ
|
ÿ ÿ ÿ
|
ÿ
|
ÿ
|
ÿ ÿ
|
|
ÿÿÿ ÿÿÿÿ ÿÿÿÿÿÿÿÿÿrB r! TFc C s> t dd}| rt |¡}tt tj|dd W 5 Q R X d S )Nr r ZM8r
|
)r r r! r Ú TypeError)r! Úrtr r r$ Útest_array_impossible_casts s
|
|
|
|
rE r? r r é é c C s: t ¡ ( t | ¡}t|| j |jjs,tW 5 Q R X d S r) ) ÚpytestZdeprecated_callr ZfastCopyAndTransposer ÚTÚflagsZowndataÚAssertionError)r? Úbr r r$ Útest_fastCopyAndTranspose s
|
|
rM c C sZ t jddd dd¡} | d¡}t| | t|jt d¡ t| j|j | j d¡}t| j| t|jt d¡ t| jj|j | d¡}t| | t| |k | jddd}t| |k | jdd dd
|
}t| | t| |k t|j j
|
| jdd dd
|
}t| | t| |k t|j j | jd d d}t| | t|jt d ¡ t t | jdd d | jdddd}t| |k G dd dt j}t jdddgdddggdd |¡} | jdddd}t| |k | jddd}t| | tt|| | jdddd}t| | t| |k tt||k t jdgdd} | d¡}t| | t|jt d¡ t jdgdd} | d ¡}t| | t|jt d ¡ t jd gdd} | d¡}t| | t|jt d ¡ t jd gdd} | d ¡}t| | t|jt d!¡ t jd"dd d¡} t| t jd#d$d t jd"dd d ¡} t| t jd%d&d t jd"gdd d¡} t| t jd#d$d t jd"gdd d ¡} t| t jd%d&d t jd"dd} t| t jd#d$d t jd"d d} t| t jd%d&d t jd'd d} t j| d(dd)}t|jdk t jd*gdd} t t | jd+d d t jd*dd} t t | jd,d d d S )-NrG Úf4r
|
r r Úi4F©ÚcopyÚF)ÚorderrQ ÚCZc8Úsafe©Úcastingr )ZsubokrQ c @ s e Zd ZdS )z$test_array_astype.<locals>.MyNDArrayN)Ú__name__Ú
|
__module__Ú __qualname__r r r r$ Ú MyNDArrayÓ s r[ r é r Tsd aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaÚOÚSZS100ZdaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaÚUZU100s
|
aaaaaaaaaaZS10Z
|
aaaaaaaaaaZU10l Ò
|
~ ¹l!]c s 123456789012345678901234567890ZS30Z 123456789012345678901234567890ZU30u aÅZuint32)Úbufferr Úshapeiè ZS1ZU1)r ÚarangeÚreshapeÚastyper r ÚstridesrI r rJ Ú f_contiguousÚ c_contiguousr rC Zndarrayr! Úviewr* r Úsize)r? rL r[ r r r$ Útest_array_astype¨ s
|
|
|
|
|
|
|
$
|
|
|
|
|
|
|
|
|
|
rj Údtr^ r_ c C st t jdgtd}| | ¡jjt | d¡jks4tt j|| ddsHtt j|| ddr\tt jd| ddsptd S )NÚ r
|
r ÚunsaferV Z same_kindr] )r r! r+ rd r ÚitemsizerK Zcan_cast©rk Úarrr r r$ Ú+test_array_astype_to_string_discovery_empty s
|
$rq ÚdÚfZS13ZU32c C s4 t | ¡} t jg | d}| d¡jj| jks0td S )Nr
|
ÚV)r r r! rd rn rK ro r r r$ Útest_array_astype_to_void# s
|
ru c C s& t jg dd d¡} | jdks"td S )Nr] r
|
rt ZV8)r r! rd r rK )rp r r r$ Ú test_object_array_astype_to_void) s rv ÚtZuintr8 r9 c C s$ t jdt jd}tt j|j| d S )Nr( r
|
)r r! Zcomplex_r ZComplexWarningrd )rw r? r r r$ Útest_array_astype_warning/ s rx r Ú out_dtypezS10,S9z?,?c C s@ t jddddg| d}t jddddg|d}t| |¡| dS ) z¹
|
Currently, for `astype` strings are cast to booleans effectively by
|
calling `bool(int(string)`. This is not consistent (see gh-9875) and
|
will eventually be deprecated.
|
Ú10z10 z0 Ú0r
|
TFN)r r! r rd )r ry rp Úexpectedr r r$ Útest_string_to_boolean_cast7 s
|
r} c
|
C s< dD ]2}t j|g| d}tt | |¡ W 5 Q R X qdS )zn
|
These currently error out, since cast to integers fails, but should not
|
error out in the future.
|
)ÚFalseÚTruerl ú z non-emptyr
|
N)r r! r r< rd )r ry Úinvalidrp r r r$ Ú"test_string_to_boolean_cast_errorsE s
|
r Ústr_typeÚ scalar_typec C s |d}||dkst tj|gtd |¡d dks:t t |¡ |¡d dksVt tjd|d}||d< |d dks|t d S )Ns 1+3jy ð? @r
|
r r r )rK r r! r+ rd Úzeros)r r Úvaluerp r r r$ Útest_string_to_complex_castS s " r ZAllFloatc C s t jd| d}d |d< t |¡d s(tt t jd | d¡d sDtt t jd g| d¡d sbtt t d ¡ | ¡¡d std S )Nr r
|
r r )r r
ÚisnanrK r! rd )r rp r r r$ Útest_none_to_nan_cast_ s r c C sÜ t jddd dd¡} t | d¡ t| d t | jd¡ t| d t jdd dgddd ggd
|
d}t j| d |d t| dd dgddd gg t jdd gd d gd dggd
|
d}t j| jd |d t| dd d gd d d gg d S )NrG rN r
|
r r r' g @r r ú?ç @©Úwhereg @)r rb rc Úcopytor rI r! )r? Úmaskr r r$ Útest_copyto_fromscalarj s
|
|
r c C sÀ t jddd dd¡} t | dddgdddgg¡ t| dddgdddgg t | d d
d d
f | d d d
dd d
f ¡ t| dddgdddgg ttt j| d t j| d d
|
d t| d t j| dd d d gd t| dddgdddgg ttt j| dd d d gd t j| dd
|
dddgdddggd t| dddgdddgg t j| d d
d d
f | d d d
dd d
f ddgddggd t| dddgdddgg ttt jdddgdddg d S )NrG rO r
|
r r r r éÿÿÿÿr' rm rV TFr r r r )rW r r\ )r rb rc r r r rC )r? r r r$ Ú test_copyto{ s 0
|
"@r c s d} dg| ddddg }t t|| ¡}t t|| ¡}t |¡| d
}t j|||d d d
d d}t |¡}td| D ] t |¡} fddt|D }t |¡}t j|||d t|d k| t| ¡ t | t |¡}t j|||d d d
d t|d k|d d d
t| ¡ t | t |¡}t j|d d d
|d d d
|d d d
d t|d d d
d k|d d d
t|d d d
¡ t |d d d
t |¡}t j|d d d
|d d d
|d d d
|
d t|d d d
d k|d d d
|
t|d d d
¡ t |d d d
|
d D ]h}t |¡}t |¡}t |¡
|
t j ¡}|||d k< t j|||d t|d k| t| ¡ t | q6qt |¡}t j||dd t| ¡ |j t |¡}t |¡}t j||d d t| ¡ |j d S )Niô Tr r é r c s g | ]} |@ d kqS )r r )Ú.0r# ©rA r r$ Ú
|
<listcomp>® s z&test_copyto_permut.<locals>.<listcomp>r éþÿÿÿ)éÿ é r é r F) r r
Úlenr, r! r r; r r Úsumrh r6 ri )ÚpadÚlÚrrr r ÚpowerÚcZimaskr r r$ Útest_copyto_permut sT
|
|
|
|
|
."&
|
."&
|
|
|
|
|
r¢ c sÐ t d¡ dddd¡} | jdd}t d¡ dddd¡ dd¡}d fd d
|
}t| jj t| jj t|jj t|jj t|jj t|jj | jd d | | d dd d |jd d | |d ddd |jd d | |d ddd t j| d d | | d dd d t j|d d | |d ddd t j|d d | |d ddd | jdd | | dd dd |jdd | |dd d d |jdd | |dd dd t j| dd | | dd dd t j|dd | |dd d d t j|dd | |dd dd | jdd | | d dd d |jdd | |dd d d |jdd | |ddd d t j| dd | | d dd d t j|dd | |dd d d t j|dd | |ddd d d S )Né r r r r\ rR ©rS Fc s6 t | |k t| | t jj| t jj| d S r) )r r rJ rg rf )r# ÚyÚccontigÚfcontigre ©Úresr r$ Úcheck_copy_resultÙ s
|
z*test_copy_order.<locals>.check_copy_resultrT T)r¦ r§ re ÚK)F) r rb rc rQ Úswapaxesr rJ rg rf )r? rL r¡ rª r r¨ r$ Útest_copy_orderÔ s\ r c C s t d¡d d d
d d
d d
f } | jd d
d | _t d¡ dd¡}dd }|| d d ||d d |t d
|
¡d d |t jd gdgggd dd d |t d¡d d |t jdd dd d |t j| d dd d |t j| d ddd d |t j| dd d dd d || d d d || d d d d
dd f d d ||d d d ||d d
d d
dd
d d
d d
f d d || ¡ d d |t d¡ ¡ d d d S )N)r\ r\ r r )i
ÿÿÿ)r r r r r r r\ c S s$ t | jj|k t | jj|k d S r) )r rJ rg rf )r? r¦ r§ r r r$ Ú check_contig s z+test_contiguous_flags.<locals>.check_contigF)r r r r r Tr rR r¤ ©r r rP rT )rQ rS )ZndminrQ rS r .)r r .)r r r )r r, re r¬ Úemptyr! ZravelZsqueeze)r? rL r® r r r$ Útest_contiguous_flags s& " .r± c C sp t jdgdd} t jdddgdd}t | |¡}t|d t jdddgdd t|d t jdddgdd d S )N)r r r zu4,u4,u4r
|
)r\ r rG )rF é r r r )r r! Zbroadcast_arraysr )r? rL Úresultr r r$ Útest_broadcast_arrays1 s
|
r´ ra Ú
|
fill_valueÚexpected_outputr¯ g @r )r r r g @c C s t | |¡}t|| d S r) )r r= r )ra rµ r¶ Úoutputr r r$ Útest_full_from_list9 s r¸ c C s t jdt jd} | jt jdd}t | |¡r0t| jt jt jjd}t | |¡rTt| jt jdd}t | |¡stt| jt jt jjd}t | |¡st| jt jt jj d}t | |¡s¼t| jt j
|
dd}t || | jt j
|
t jjd}t || t t | jt j
|
t jj d d S )Nr( r
|
TrP F)r rb Zintprd Zmay_share_memoryrK Z _CopyModeZALWAYSZ IF_NEEDEDZNEVERr" r r r< )rp Zres_trueZ
|
res_alwaysZ res_falseZ res_if_neededZ res_neverr r r$ Útest_astype_copyflag@ s(
|
ÿ
|
ÿr¹ )2r- Únumpyr Znumpy.core._rational_testsr rH Z numpy.testingr r r r r r rB ÚmarkZ parametrizerE r! rb rc rM rj rq ru rv Zsctypesrx Úbytes_Zbool_Zunicode_r r} r r1 r0 Zstr_Z complex64Z
|
complex128Z clongdoubler Z typecodesr r r r¢ r r± r´ r¸ r¹ r r r r$ Ú<module> sv
|
ýÿ
|
m
|
|
ÿ
|
|
|
þÿ
|
|
|
|
þÿ
|
|
ÿ
|
|
$5= $ÿÿ
|
|
