U
|
O ±d- ã @ sæ d Z ddlZddlZddlZddlZddlZddlZddlm Z ddl
|
m Z ddl m Z dZe d¡je d¡jkre dd ¡ d
|
d ¡Zd d eD Zd d Zdd ZG dd dejZdd Ze ZG dd dZG dd dZdS )zí
|
The tests exercise the casting machinery in a more low-level manner.
|
The reason is mostly to test a new implementation of the casting machinery.
|
|
Unlike most tests in NumPy, these are closer to unit-tests rather
|
than integration tests.
|
é N)Ú
|
as_strided)Úassert_array_equal)Ú_get_castingimplz?bhilqBHILQefdFDÚlÚqÚ ÚLc C s g | ]}t t |¡qS © ©ÚtypeÚnpÚdtype©Ú.0Úcr r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_casting_unittests.pyÚ
|
<listcomp> s r c c sN t D ]D} | }tj|t|dV |jdkr| ¡ }tj|t|dV qd S )N)Úidú|)Ú simple_dtypesÚpytestÚparamÚstrÚ byteorderÚ newbyteorder)Z dtype_classÚdtr r r Úsimple_dtype_instances s
|
r c C sÆ | t jkrdS | jdkr|| jdkr(d}n>| jdkr8d}n.| jdkrHd}n | jdkrXd }ntd
|
| | jd krx|d7 }|S | jd krd S | jdkrdS | jdkr¦dS | jdkr´dS td
|
| dS )zHReturns the string length when casting the basic dtypes to strings.
|
é Úiué é é é é
|
é é z!did not find expected length for ÚiÚgé0 ÚGé` Úfé r é@ N)r Zbool_ÚkindÚitemsizeÚAssertionErrorÚchar)r Úlengthr r r Úget_expected_stringlength' s0
|
|
|
|
|
|
|
|
|
|
|
r3 c @ s e Zd ZdZdZdZdZdZdS )ÚCastingr r r! r r" N)Ú__name__Ú
|
__module__Ú __qualname__ÚnoÚequivÚsafeÚ same_kindÚunsafer r r r r4 J s
|
r4 c C s¢ t d¡ ¡ d¡} dd | d dd d
D }tjtjtjtjdd}i }t || d d
D ]>\}}i ||< t ||dd d
D ]\}}|| || |< qq^|S )
|
Na³
|
X ? b h i l q B H I L Q e f d g F D G S U V O M m
|
? # = = = = = = = = = = = = = = = = = = = = = . =
|
b . # = = = = . . . . . = = = = = = = = = = = . =
|
h . ~ # = = = . . . . . ~ = = = = = = = = = = . =
|
i . ~ ~ # = = . . . . . ~ ~ = = ~ = = = = = = . =
|
l . ~ ~ ~ # # . . . . . ~ ~ = = ~ = = = = = = . =
|
q . ~ ~ ~ # # . . . . . ~ ~ = = ~ = = = = = = . =
|
B . ~ = = = = # = = = = = = = = = = = = = = = . =
|
H . ~ ~ = = = ~ # = = = ~ = = = = = = = = = = . =
|
I . ~ ~ ~ = = ~ ~ # = = ~ ~ = = ~ = = = = = = . =
|
L . ~ ~ ~ ~ ~ ~ ~ ~ # # ~ ~ = = ~ = = = = = = . ~
|
Q . ~ ~ ~ ~ ~ ~ ~ ~ # # ~ ~ = = ~ = = = = = = . ~
|
e . . . . . . . . . . . # = = = = = = = = = = . .
|
f . . . . . . . . . . . ~ # = = = = = = = = = . .
|
d . . . . . . . . . . . ~ ~ # = ~ = = = = = = . .
|
g . . . . . . . . . . . ~ ~ ~ # ~ ~ = = = = = . .
|
F . . . . . . . . . . . . . . . # = = = = = = . .
|
D . . . . . . . . . . . . . . . ~ # = = = = = . .
|
G . . . . . . . . . . . . . . . ~ ~ # = = = = . .
|
S . . . . . . . . . . . . . . . . . . # = = = . .
|
U . . . . . . . . . . . . . . . . . . . # = = . .
|
V . . . . . . . . . . . . . . . . . . . . # = . .
|
O . . . . . . . . . . . . . . . . . . . . = # . .
|
M . . . . . . . . . . . . . . . . . . . . = = # .
|
m . . . . . . . . . . . . . . . . . . . . = = . #
|
Ú
|
c S s g | ]}t t |¡qS r r
|
r r r r r n s z&_get_cancast_table.<locals>.<listcomp>r r! éÿÿÿÿ)Ú.ú~ú=ú#ú r )
|
ÚtextwrapÚdedentÚstripÚsplitr4 r< r; r: r9 Úzip)ÚtableZdtypesZ convert_castZcancastÚfrom_dtÚrowÚto_dtr r r r Ú_get_cancast_tableR s æ þ rM c @ sH e Zd ZdZej dddg¡ej ddddd g¡d
|
d Zd d ZdS )Ú TestChangesz:
|
These test cases exercise some behaviour changes
|
ÚstringÚSÚUÚfloatingÚer+ Údr' c C s* t ||¡stt || d¡s&td S )NZ100©r Úcan_castr0 )ÚselfrR rO r r r Útest_float_to_string s z TestChanges.test_float_to_stringc C s t dd¡stt dd¡s tt dd¡r0tt dd¡r@tt dd¡rPtt jddddsdtt jd dd
|
dsxtt jd dd
|
dstd S ) NrT ÚVZS20ZV1ZU1zd,ir; ©ÚcastingÚV3r8 ZV0rU ©rW r r r Ú test_to_void s zTestChanges.test_to_voidN) r5 r6 r7 Ú__doc__r ÚmarkÚ parametrizerX r^ r r r r rN s
|
rN c @ s2 e Zd ZdZdd ZdvddZej de ¡dd Z
|
ej d
|
¡ej d e ¡d d Z ej de ¡dd Zej d dddddgddejdddfee d¡ ¡ dejdddfddejdddfddejdddfddejdddfddejddd fddejdd dfdd ejddd fd d ejddd!dd"d#d$d%gfd&dejdddfee d&¡ ¡ dejdddfd'd(ejdddfd(d'ejdddfd)d)ejdddfd&d(ejddd fd(d&ejdd dfd(d*ejddd fd+d,ejddd!ddd#d$d%gfg¡d-d. Zd/d0 Zej d1e ¡ej d2d3d4g¡d5d6 Zej d2d3d4g¡ej d7e ¡d8d9 Z ej d7d:d;d<g¡ej d2d3d4g¡d=d> Z ej d?d@dAg¡ej dBd@dAg¡dCdD Z dEdF Z dGdH Z ej dIe ¡dJdK Z!ej dIe ¡dLdM Z"ej dNdOdPg¡dQdR Z#ej ddSge dTdUgdVdWgddXgdY¡dZfe dUdTgdVdWgddXgdY¡dZfe dUdTgdVdWgdd[gdY¡dfg¡d\d] Z$ej d^d_d`dadbe%dTgdbgdZgdYdfe%dTgdbgdZgdYdbdZfdbe%dTdUgdbdbgdZdZgdYdfdcdddedfdgdhdidjdkdldcg¡dmdn Z&ej dIej'do ¡dpdq Z(ej dIej'dr ej'ds ¡dtdu Z)dS )wÚ TestCastingiÜ c C s` |d ks|j |j kr"| j|j }n | j|j }tj||d}|jjsHt|jjsTtdd t|D }t |D ]6\}}|dk r|j
|
dkr|t |¡j d }|||< qn|d krÎ|j dkrÆdd |D }||fS |j dkræd d |D }tj||d}|jjst|jjstt |D ]<\}}|dk rJ|j
|
dkrJ|t |¡j d }|||< q|||fS )
|
N©r c S s g | ]}t d d¡qS )iÿÿÿé )ÚrandomÚ randrange)r Ú_r r r r ¨ s z(TestCasting.get_data.<locals>.<listcomp>r Úur ú?c S s g | ] }t |qS r ©Úbool©r Úvr r r r ³ s c S s g | ] }t |qS r rj rl r r r r · s )r/ Úsizer ÚemptyÚflagsÚ c_contiguousr0 ÚalignedÚrangeÚ enumerater. ZiinfoÚmaxr1 ) rW Údtype1Údtype2r2 Úarr1Úvaluesr& ÚvalueÚarr2r r r Úget_data s2
|
|
|
zTestCasting.get_dataTc C s> |r|j j}|j j}n6|r4d|j j }d|j j }n|j jd }|j jd }t|d |j j d }t|d |j j d }tj|tjd} tj|tjd}
|
|t| | jks²t|t| |
|
jksÈt|rt| dd
|j ¡|j
|
|f} t|
|
dd
|j ¡|j
|
|f} n@t| dd
|j ¡|j
|
|f} t|
|
dd
|j ¡|j
|
|f} || d< |s|rl|j jnd} | | d|
¡ }| d ¡t|kst|r¾| j js®t| j jsÚtn | j jrÌt| j jrÚt|rþ| j jsît| j js6tn8| j jr| j jdkst| j jr6| j jdks6t| | fS )
|
z
|
Returns a copy of arr1 that may be non-contiguous or unaligned, and a
|
matching array for arr2 (although not a copy).
|
r! r r rc Nr> .r ó )r r/ Úlenr ÚzerosÚuint8Únbytesr0 r ÚviewÚshapeÚtobytesÚcountrp rq rr Ú alignment)rW rx r{ rr ÚcontigZstride1Zstride2Z max_size1Z max_size2Ú
|
from_bytesÚto_bytesZnew1Znew2ÚoffsetÚbufr r r Úget_data_variationÆ s^
|
ÿ ÿ ÿ ÿ z TestCasting.get_data_variationÚfrom_Dtc C s4 t D ](}t||}| | ¡ fD ]}| |d f¡d d }|| ksNt~| | ¡ fD ]Ê}| ||f¡\}\}} }
|
t||kstt| |kst|
|
d k rÚ|tjks²ttjt | | ksÈt|j
|
| j
|
ksØtn.||krò|j
|
| j
|
ksòt|t | | kst||kr`||ks t|| ks`tq`q$qd S )Nr ) r Úget_castingimplr Ú_resolve_descriptorsr0 r r4 r8 r9 Ú
|
CAST_TABLEÚisnative) rW r Zto_DtÚcastrJ ÚdefaultrL r[ Úfrom_resÚto_resÚview_offr r r Útest_simple_cancast s*
|
|
ÿ z TestCasting.test_simple_cancastz ignore::numpy.ComplexWarningrJ c C s t D ]x}|jd }tt|t|}| ||f¡\}\}}}||k sP||k rV dS |tjk}~~~| ||¡\} }
|
} | | |
|
f¡ |
|
¡ | kst
|
| | |
|
dd¡\} } | | | f¡ t | |
|
| ¡ |
|
¡ ksÖt
|
|jdkrê|jdksð|jsö dS | | |
|
dd¡\} } | | | f¡ t | |
|
| ¡ |
|
¡ ks8t
|
| | |
|
dd¡\} } | | | f¡ t | |
|
| ¡ |
|
¡ kszt
|
~ ~ ~qdS )a
|
This test checks numeric direct casts for dtypes supported also by the
|
struct module (plus complex). It tries to be test a wide range of
|
inputs, but skips over possibly undefined behaviour (e.g. int rollover).
|
Longdouble and CLongdouble are tested, but only using double precision.
|
|
If this test creates issues, it should possibly just be simplified
|
or even removed (checking whether unaligned/non-contiguous casts give
|
the same results is useful, though).
|
r NTFr )r ry r r r r4 r: r| Ú_simple_strided_callÚtolistr0 r r r r Ú_supports_unaligned)rW rJ rL r r[ r r r r: rx r{ ry Úarr1_oÚarr2_or r r Útest_simple_direct_casts# s<
|
ÿ
|
|
ÿ
|
|
z$TestCasting.test_simple_direct_castsc
|
C s¢ | }t d¡t d¡t d¡t d¡t d¡t d¡g}|D ]\}tt|t|}| ||f¡\}\}}} ||kszt||kst~~|t| t| @ s¢t| d ks®tt t j¡}
|
| ||
|
¡\} } } | |¡} t
|
d¡| d< |t d¡kr2d | d
|
< | | | f¡ t t¡ t| d
|
W 5 Q R X d S | | | f¡ d d | ¡ D | ks\t| | | d d¡\}}| ||f¡ t|| | ¡ | ¡ ks>tq>d S )NÚM8úM8[ms]úM8[4D]Úm8úm8[ms]úm8[4D]ÚNaT.r r> c S s g | ] }t |qS r )Úintrl r r r r s z5TestCasting.test_numeric_to_times.<locals>.<listcomp>TF)r r r r r r0 r Úint64r| r Z
|
datetime64r r ÚraisesÚ
|
ValueErrorr r r r r )rW r rJ Z time_dtypesZtime_dtr r[ r r r Zint64_dtrx r{ ry r r r r r Útest_numeric_to_timesa s@ ÿ
|
ÿ
|
|
z!TestCasting.test_numeric_to_timesrL Úexpected_castingÚexpected_view_offÚnomÚdenomzM8[ns]Nr r r r zM8[5ms]i@B zM8[7ms]é r zM8[1M]ìûÿÿÿ r> i" iÝúÿÿiòµ¤!zm8[ns]r¡ r¢ zm8[5ms]zm8[7ms]r£ zm8[1M]c C sÎ t |¡}|d k r t |¡}t ddddddg¡}| t d¡ |j¡¡}|jj|jks\tt | |¡d ¡sttt |}t
|
||} | ||f¡\}
|
\} } } | |ksªt| |ks¾|d ks¾t|
|
|ksÊt| |ksÖt|d k rú|| | | ¡}d |d< n t |¡}||d
|
< | |¡}| |¡}t |¡}|
|
t jkrr|d ksJ|d krrt t¡ | ||f¡ W 5 Q R X d S d D ]R}d D ]F}| ||||¡\}}d|d
|
< | ||f¡ t| d¡| d¡ q~qvd S )Nr¯ r l ÿÿÿÿ i' iðØÿÿl r¦ r r¤ .r¡ r )TT)r r ÚarrayÚastyper r r0 Zisnatr r r r Ú
|
empty_liker4 r< r r§ r¨ r r r )rW rJ rL rª r« r¬ r ry ZDTyper r[ r r r Z expected_outÚorig_arrZorig_outrr r ZarrÚoutr r r Útest_time_to_time sN
|
|
|
ÿ
|
|
|
|
|
ÿz TestCasting.test_time_to_timec C s: |j dkrdnd}|j| | }t |j |j | ¡S )NrP r r" )r1 r/ r r r )rW r Ú change_lengthÚfactr2 r r r Ústring_with_modified_lengthÝ s z'TestCasting.string_with_modified_lengthÚother_DTÚ string_charrP rQ c C s |dkr dnd}t t |¡}t||}| }t|}t | | ¡}| |d f¡\} \}
|
} } | j|| ksrt| tj kst| d kstt
|
| |stdD ]b} | dkr²tj }ntj }| || ¡}| ||f¡\} \}} } | |ksèt| |ksôt| d kstqt||}| ||f¡\} }} | tj ks0t| d ks>tt||}| |d f¡\} \}} } | tj kspt| d ks~t|| kstd S )NrP r r" ©r> r r r )r r r r r3 r r/ r0 r4 r: Ú
|
isinstancer; r¸ r< )rW r¹ rº r· Ú string_DTr Úother_dtÚexpected_lengthÚ string_dtÚsafetyÚ res_other_dtÚres_dtr r¶ Úexpected_safetyrL rg r r r Útest_string_cancastâ sF
|
ÿ ÿ
|
|
ÿz TestCasting.test_string_cancastr¾ c C s° t t |¡}tt ||}t|t |}| |df¡\}\}}}||k r^|j|jksZtdS | |d¡\} }
|
tjt | |d} |
|
|d¡} tjt | | d} |
|
|d¡}tjt | |d}|j rÆt|j rÐtdD ]Ö}| | | d|¡\}} | | | ¡ d|¡\}} | | |d|¡\}}| || f¡ | || f¡ t| | ¡| | ||f¡ t| | |jdkrjqÔd|d < | | |f¡ t| | d|d < | ||f¡ t| | qÔdS )
|
ah
|
Tests casts from and to string by checking the roundtripping property.
|
|
The test also covers some string to string casts (but not all).
|
|
If this test creates issues, it should possibly just be simplified
|
or even removed (checking whether unaligned/non-contiguous casts give
|
the same results is useful, though).
|
Nrc r> r )TFTÚbr .)r r r r r r r0 r| r r~ r¸ r r Úcopyr r r± r. )rW r¾ rº r½ r Z cast_backrg r rÀ r³ ry Zstr_arrZstring_dt_shortZ str_arr_shortZstring_dt_longZ str_arr_longr Z other_arrr r r Ú"test_simple_string_casts_roundtrip sb ÿ
|
|
ÿ ÿ ÿ
|
|
z.TestCasting.test_simple_string_casts_roundtripZS8z<U8z>U8c C s t |¡}|dkrdnd}|jdkr(dnd}tt |¡}tt||}|j| }t | | ¡}| |d f¡\} \}
|
} } | j|| kstt| |std } |j|krÄ|j r¼t
|
j }d} qÚt
|
j }n|dkrÔt
|
j }nt
|
j}| | ksæt|| ksòtdD ]Æ}| ||¡}| ||f¡\} \}} } | |ks,t|dkrF| | ksTtn| d ksTt|t
|
jkrp| |ks¼tqö|dk r| t
|
jks¼tqö|dkr¦| |ks¼tqö|dkrö| t
|
j ksötqöd S )NrP r r" r rQ r» )r r r1 r r r/ r r0 r¼ r r4 r8 r9 r: r< r¸ r; )rW r¾ rº r· Údivr½ r r¿ rÀ rÁ r rà r r« rÄ r¶ rL rg r r r Ú test_string_to_string_cancastP sP
|
|
ÿ
|
ÿ
|
|
|
z)TestCasting.test_string_to_string_cancastÚorder1ú>ú<Úorder2c C sÌ t | d¡}t | d¡}t jdt jddd
|¡}t jdt jddd
|¡}|jdkr~|jjrrt|jjr~td}||d< || ¡ fD ]0}||d< |d |ks²t| ¡ d |kstqd S )NZU30éy rc r u this is a ünicode stringâ½r .)
|
r r ro r r r rp rr r0 rÇ ) rW rË rÎ rv rw Zdata1Zdata2ÚelementÚdatar r r Ú test_unicode_byteswapped_cast
s
|
z)TestCasting.test_unicode_byteswapped_castc C sD t jg dd d¡jjdks tt jg dd d¡jjdks@td S )NZV5rc rP r rQ r% )r r° r± r r/ r0 r] r r r Ú test_void_to_string_special_case s z,TestCasting.test_void_to_string_special_casec C sX t t t¡}t t t¡}t||}tjtdd | t d¡d f¡ W 5 Q R X d S )Nz+casting from object to the parametric DType)ÚmatchÚO)
|
r r r Úobjectr r r r§ Ú TypeErrorr )rW Ú object_dtypeZ other_dtyper r r r Ú(test_object_to_parametric_internal_error¢ s
|
ÿz4TestCasting.test_object_to_parametric_internal_errorr c C s® t t t¡}t|t |}| t d¡|f¡\}\}}}|tjksHt|d ksTt||ks`t| t d¡d f¡\}\}}}|tjkst|d kst|| d¡ksªtd S )NrÕ rA )
|
r r r rÖ r r r4 r< r0 r ©rW r rØ r rÁ rg rà r r r r Ú!test_object_and_simple_resolution¬ s ÿ ÿ z-TestCasting.test_object_and_simple_resolutionc C sd t t t¡}tt ||}| |d f¡\}\}}}|tjksBt|d ksNt|t d¡ks`td S )NrÕ ) r r r rÖ r r r4 r: r0 rÚ r r r Ú test_simple_to_object_resolution¾ s ÿ z,TestCasting.test_simple_to_object_resolutionr[ r8 r< c C sH t dg¡}|dk}t jd||d|ks,tt j|d|d|ksDtd S )N)Zfooz<f4)r r! r< ÚV4rZ )r r rV r0 )rW r[ r Úexpectedr r r Ú&test_void_and_structured_with_subarrayÊ s z2TestCasting.test_void_and_structured_with_subarrayÚ expected_offÚarÆ Úi4Úf4r" ©ÚnamesÚformatsÚoffsetsr! é c C s~ t ddgddgddgd¡}tt|t|}| ||f¡\}}}|j|jkr`|tjksntn|tj ksnt||ksztd S )Nrá rÆ râ rã r! rè rä )
|
r r r r r rå r4 r9 r0 r: )rW rL rà rJ r rÁ rg r r r r Ú test_structured_field_offsetsÒ s
|
þ z)TestCasting.test_structured_field_offsets)rJ rL rà )r& ú(1,1)ir )rê r& r )ú(2,1)irë r r& )zi,izi,i,iN)râ r\ r )râ rÝ r )râ ZV10N)rÕ rÝ N)rÕ ZV8N)rÝ r\ r )rÝ rÝ r )r\ rÝ N)rÝ râ Nc C sJ t |¡}t |¡}tt|t|}| ||f¡\}}}||ksFtd S )N)r r r r r r0 )rW rJ rL rà r rg r r r r Ú'test_structured_view_offsets_paramtericì s
|
%
|
|
z3TestCasting.test_structured_view_offsets_paramtericZAllc C s t d gd ¡}t |¡}t |jjd|j¡ | ¡ d|j ksDtz| |¡}W n6 t
|
k
|
r t t
|
¡ | |¡f W 5 Q R X Y nX t || |¡ d S )Nr r r} )r r° r² ÚctypesÚmemsetrÑ r r r0 r± r× r r§ r )rW r Z
|
arr_normalZ arr_NULLsrÞ r r r Ú'test_object_casts_NULL_None_equivalence s
|
z3TestCasting.test_object_casts_NULL_None_equivalenceZ
|
AllIntegerZAllFloatc C s> t jdddgt jd t¡}| |¡}dddg}t|| d S )Nr r iùÿÿÿrc r )r r° Zint8r rk r± r )rW r Znonstandard_boolsÚresrÞ r r r Ú test_nonstandard_bool_to_other) s
|
|
z*TestCasting.test_nonstandard_bool_to_other)TT)*r5 r6 r7 rn r| r r r` ra r r Úfilterwarningsr r r© r4 r8 r r r r r9 r: r< r; rµ r¸ rÅ rÈ rÊ rÒ rÓ rÙ rÛ rÜ rß ré Údictrì Z typecodesrï rñ r r r r rb sì )
|
;
|
!
|
<
|
. ÿ ÿ þ ÿ þéý
|
0 . < 3
|
|
|
|
ÿÿÿÿÿÿùÿ
|
|
ÿâ
|
|
ÿrb )r_ r rD Úenumre rí Únumpyr Znumpy.lib.stride_tricksr Z numpy.testingr Z numpy.core._multiarray_umathr r r r r/ Úreplacer r3 ÚIntEnumr4 rM r rN rb r r r r Ú<module> s( #*
|
