U
|
O ±d|« ã @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlZd dl m
|
Z
|
m Z d dl m Z d dlmZ d dlZd dlmZmZmZmZmZmZmZmZmZ ej ej ej ej ej ej!ej"ej#ej$ej%ej&ej'ej(ej)ej*ej+ej,gZ-ej. /¡ Z0ej1 /¡ Z2e3 dgZ4ej5ej6ej7ej8ej9ej:ej;ej<ej=ej>ej?ej@ejAg ZBG dd dZCd d
|
ZDejEjFe d d d e
|
e eBejGe H¡ ddejGe H¡ dddd ZIejEjFe
|
e eBe H¡ e H¡ dd ZJejE KdejLejMg¡dd ZNG dd dZOG dd dZPdd ZQd d ZRG d d d ZSG d!d" d"ZTG d#d$ d$ZUG d%d& d&ZVesZG d'd( d(ZWG d)d* d*ZXG d+d, d,ZYG d-d. d.ZZG d/d0 d0Z[G d1d2 d2Z\G d3d4 d4Z]e j^d5d6 Z_e
|
e e4e eBe e-d7d8 Z`e
|
e e4e eBe e-d9d: Zae
|
e eBe e-e e-d;d< ZbejE Kd=eB¡ejE Kd>dd?g¡d@dA ZcejE Kd=eB¡ejE Kd>dd?g¡dBdC ZdejE KdDejedE ¡ejEjKdFdGdH dIdH dJdH gdKdLdMgdNdOdP ZfejE KdDejedQ ¡ejEjKdFdRdH dSdH dTdH ejgdUdH ejEjhdVdWdXgdYdZdMd[gdNd\d] ZiejE KdDejed^ ¡d_d` ZjejE KdDejedE ¡ejEjKdFdadH dbdH gd[dcgdNddde Zkdfdgej5dhfdidjej6dhfdkdkej7dhfdldlej8dhfdgdfej:dhfdjdiej9dhfdmdnej<dofdpdqejAdofdrdsej;dofdtduej=dofdvdwej>dofdxdyej?dofdzd{ej@dofg ZlejE Kd|d}d=d~gel¡ejE KdejMejmej)g¡dd Zndd ZoejE Kd|d}d=d~gel¡ejE KdepeqerejLg¡e s¡ d
d ZtdS )é N)Ú_pep440)ÚgivenÚsettings)Ú sampled_from)Únumpy) Úassert_Ú assert_equalÚ assert_raisesÚassert_almost_equalÚassert_array_equalÚIS_PYPYÚsuppress_warningsÚ_gen_alignment_dataÚ assert_warnsc @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestTypesc C s, t D ]"}|d}t|dkd||f qd S )Né úerror with %r: got %r)Útypesr )ÚselfÚatypeÚa© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_scalarmath.pyÚ
|
test_types( s zTestTypes.test_typesc C s t tD ]\}}|d}tjdg|d}t tD ]\\}}|d}tjdg|d}|| } || }
|
t| j|
|
jd|t |¡j|t |¡jf q0qd S )Né ©Údtyper z$error with types (%d/'%c' + %d/'%c'))Ú enumerater ÚnpÚarrayr r Úchar) r Úkr Za_scalarZa_arrayÚlZbtypeZb_scalarZb_arrayZc_scalarZc_arrayr r r Ú test_type_add- s
|
ÿÿzTestTypes.test_type_addc C s@ t tD ]2\}}t dddg|¡}|dddg}t|| qd S )Nr é r )r r r r r )r r! r r Úbr r r Útest_type_createA s zTestTypes.test_type_createc C s t dD ]}t dd¡ qd S )Ni@ r )Úranger Úadd)r Úir r r Ú test_leakG s zTestTypes.test_leakN)Ú__name__Ú
|
__module__Ú __qualname__r r# r& r* r r r r r ' s r c C sN |d }|d }t |tjs tt |tjs0t|jjdksH|jjdkrtjtjtj tj
|
h}| |krt |¡sxt |¡rt d¡ | tjkr¢| ¡ dkr¢t d¡ t ¡ tjdd t dt¡ z| ||}W nH tk
|
r } z(t t|¡ | || W 5 Q R X W 5 d }~X Y nX | ||}t||d d
|
W 5 Q R X W 5 Q R X d S ) Nr Úcz3complex comp ufuncs use sort-order, scalars do not.)éÿÿÿÿr ç à?r r$ z.array**2 can have incorrect/weird result dtypeÚignore©ÚallÚerrorT)Ústrict)Ú
|
isinstancer ZgenericÚAssertionErrorr ÚkindÚoperatorÚgeÚgtÚleÚltÚisnanÚpytestÚxfailÚpowÚitemÚskipÚwarningsÚcatch_warningsÚerrstateÚ simplefilterÚDeprecationWarningÚ ExceptionÚraisesÚtyper ) ÚopÚarr1Úarr2Zscalar1Zscalar2Zcomp_opsÚresÚeZ
|
scalar_resr r r Ú check_ufunc_scalar_equivalenceN s&
|
|
&
|
rQ i' iÐ )Z max_examplesÚdeadliner )r Úshapec C s t | || dS )zÛ
|
This is a thorough test attempting to cover important promotion paths
|
and ensuring that arrays and scalars stay as aligned as possible.
|
However, if it creates troubles, it should maybe just be removed.
|
N)rQ )rL rM rN r r r Ú#test_array_scalar_ufunc_equivalencel s rT c C s, t jd|d}t jd|d}t| || d S )Nr$ r r )r r rQ )rL Údt1Údt2rM rN r r r Ú test_array_scalar_ufunc_dtypesz s rW Úfscalarc C s| t d¡}| d}t ||¡}|| j|ks0t|| j|ksBt|| jt d¡ksZtt d¡| jt d¡ksxtd S )Nr Úfloat64Úfloat32)r Úint8Z result_typer r7 Úint16)rX r) ÚfÚexpectedr r r Ú test_int_float_promotion_truediv s
|
r_ c @ s e Zd Zdd Zdd ZdS )Ú TestBaseMathc C s t jdft jdft jdffD ]t\}}t|d|dD ]Z\}}}}t |¡}t |¡|d< t |¡|d< tt ||¡||d tt |d¡|d |d tt d|¡||d t j|||d t|||d |d t j |j
|
|d
|
d 7 < tt |¡t ||¡|d |t jkr0tt |¡t d|¡|d t |¡|d< t j|d|d t||d |d t |¡|d< t jd||d t||d |d q4q d S ) Né é Úbinary)r rK Úmax_size.©Úerr_msgr$ r ©Úoutr )r rZ rY Úint32r Ú ones_likeZ
|
zeros_liker
|
r( ZarangeÚsizeZsquareÚmultiplyZ
|
reciprocalÚdivide)r ÚdtÚszrh Zinp1Zinp2ÚmsgZexp1r r r Ú test_blocked s: &þ
|
|
|
ÿ
|
|
ÿzTestBaseMath.test_blockedc C s¶ t jdt jddd
t j¡}t jdt jddd
t j¡}t|| |d t j|||d t jt |¡||d t j|t |¡|d t t |¡|¡ t |t |¡¡ d S )Né¸ r é éüÿÿÿr$ rg )r Úzerosr[ ÚviewrY r
|
r( rj )r ÚdÚor r r Útest_lower_alignµ s z TestBaseMath.test_lower_alignN)r+ r, r- rq ry r r r r r` s r` c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestPowerc C s@ t jt jt jfD ]*}|d}|d }t|dkd||f qd S )Nr rs éQ r )r r[ r\ Úfloat16r ©r Útr r% r r r Útest_small_typesà s zTestPower.test_small_typesc C sj t jt jt jt jt jfD ]L}|d}|d }d||f }t |t j¡rVt|dk| qt |d|d qd S )Né3 rs r i:g re )
|
r ri Úint64rZ rY Ú
|
longdoubleÚ
|
issubdtypeÚintegerr r
|
)r r~ r r% rp r r r Útest_large_typesÉ s zTestPower.test_large_typesc C sR dd dD }dd dD }t ||¡D ]N\}}|jtjkrNtttj|| q(t ||¡}t |jj
|
tj k t |d q(dd dD }t ||¡D ]N\}}|jtjkr¸tttj|| qt ||¡}t |jj
|
tj k t |d qd d dD }t ||¡D ]P\}}|jtjkr$tttj|| qüt ||¡}t |jj
|
tj k t |d
|
qüd S ) Nc S s g | ]}t d |¡d qS ©r/ r ©r r ©Ú.0rn r r r Ú
|
<listcomp>× s zETestPower.test_integers_to_negative_integer_power.<locals>.<listcomp>Zbhilqc S s g | ]}t d |¡d qS )r r r r r r r r Ú s Z
|
bhilqBHILQç ð?c S s g | ]}t d |¡d qS r r r r r r r ä s ç ð¿c S s g | ]}t d |¡d qS )r$ r r r r r r r î s r0 ) Ú itertoolsÚproductr r Úuint64r Ú
|
ValueErrorr9 rA r rK rY r
|
)r ÚexpÚbaseÚi1Úi2rO r r r Ú'test_integers_to_negative_integer_powerÓ s, z1TestPower.test_integers_to_negative_integer_powerc C s t jt jt jt jt jt jt jt jt jg }|D ]h}|D ]^}|d}|d}|| }d|||df }t t |¡t j
|
¡rt |dk| q4t |d|d q4q,d S )Nr r$ z(error with %r and %r:got %r, expected %ré re ) r r[ r\ r| rZ rY ri r r r r r r
|
)r ZtypelistÚt1Út2r r% Úresultrp r r r Útest_mixed_types÷ s& þ
|
ÿzTestPower.test_mixed_typesc C sd d}d}d}t |||}tjtjtjfD ]6}tttj |||| tttj t ||¡|| q(d S )Né rs é
|
) rA r ri rZ Ú complex64r Ú TypeErrorr9 r )r r r% r. r^ r~ r r r Útest_modular_power s z TestPower.test_modular_powerN)r+ r, r- r r
r r r r r r r rz  s
|
|
$rz c C s | | | | fS ©Nr )ÚxÚyr r r Úfloordiv_and_mod s r£ c C s | t jd krdS dS d S )NÚUnsignedInteger)r ©r r/ )r Ú typecodes)rn r r r Ú_signs s r§ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestModulusc C s t jd t jd }ttfD ]æ}t ||¡D ]Ô\}}t t|t|¡D ]¶\}}d}||j||||f }t j|d |dd } t j|d |dd }
|
|| |
|
\} } t | |
|
| | |d |d kràt
|
|
|
| k oÔd
|
kn | qHt
|
|
|
| koôd
|
kn | qHq,q d S ) NÚ
|
AllIntegerÚFloatú*op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %séG r r é re r/ r ) r r¦ r£ Údivmodr r r§ r+ r r r © r rn rL rU rV Zsg1Zsg2Úfmtrp r r% ÚdivÚremr r r Útest_modulus_basic s z TestModulus.test_modulus_basicc s8 t tdd}t tdd}|dg | }|| }t t ||¡}t dd |D }tj|tdj\}}tj|tdj\} }
|
t | dk|dk |dk A @ d | ¡} t |
|
dk|dk @ d |
|
¡}
|
t
|
t fD ]r tj d
|
D ]b} d j | f } | | ¡} | | ¡}t fd d t| |D \}}t|| | d t||
|
| d qÎqÀd S )Niÿÿÿr r é c s s | ]}t | V qd S r )r® )r r~ r r r Ú <genexpr>8 s z7TestModulus.test_float_modulus_exact.<locals>.<genexpr>r ç ç rª zop: %s, dtype: %sc s g | ]\}} ||qS r r )r Za_Zb_©rL r r r G s z8TestModulus.test_float_modulus_exact.<locals>.<listcomp>re )Úlistr' r r r r ÚintÚTÚfloatÚwherer£ r® r¦ r+ ÚastypeÚzipr )r ZnlstZplstZdividendZdivisorÚargÚtgtr r% ZtgtdivZtgtremrn rp ÚfaZfbr± r² r r¸ r Útest_float_modulus_exact0 s$ "
|
|
z$TestModulus.test_float_modulus_exactc C sú t jd }ttfD ]â}t ||¡D ]Ð\}}t dd¡D ]º\}}d}||j||||f }t j|d d |dd } t j|d |dd }
|
|| |
|
\} } t| |
|
| | |d |d krÒt |
|
| k oÆd
|
kn | q6t |
|
| koæd
|
kn | q6q"qd S ) Nrª r¥ r« éN g+i¤)+p>r r re r/ r )
|
r r¦ r£ r® r r r+ r r r r¯ r r r Útest_float_modulus_roundoffK s
|
z'TestModulus.test_float_modulus_roundoffc C s´ t jd D ]l}t jd|d}t t jd|d| ¡}t ||¡}t||kd| t | | ¡}t|| kd| q
|
t *}| t d¡ | t d¡ | t d¡ | t d ¡ | t d
|
¡ t jd D ]Ü}t jd|d}t jd|d}t jt j
|
|d}t jt j |d} t ||¡}tt |¡d| t || ¡}tt |¡d| t ||¡}tt |¡d| t tfD ]2}
|
|
|
||\} } tt | ¡ott | ¡ qpqÈW 5 Q R X d S ) Nrª r r r¶ zdt: %sz&invalid value encountered in remainderz'divide by zero encountered in remainderz*divide by zero encountered in floor_dividez$divide by zero encountered in divmodz#invalid value encountered in divmod)r r¦ r Z nextafterr9 Úmodr r ÚfilterÚRuntimeWarningÚinfÚnanr> r£ r® Úisinf) r rn r% r r² ÚsupZfoneZfzerZfinfZfnanrL r± rÆ r r r Ú test_float_modulus_corner_cases] s6
|
z+TestModulus.test_float_modulus_corner_casesc C sJ t ddgt j¡}t ddgt j¡}tjtdd || }W 5 Q R X d S )Nr r$ z,Cannot cast ufunc 'floor_divide' output from©Úmatch)r r r r r? rJ r )r r r% r r r Ú test_inplace_floordiv_handling s ÿz*TestModulus.test_inplace_floordiv_handlingN)r+ r, r- r³ rà rÅ rÍ rÐ r r r r r¨ s
|
#r¨ c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestComplexDivisionc C sú t jddä t jt jfD ]Ð}|d}|d}tt || ¡ |tt jt j}tt || ¡ |tt jt j}tt || ¡ |tt jt j}tt || ¡ |tt jt j}tt || ¡ |d}tt || ¡ qW 5 Q R X d S )Nr1 r2 r¶ r )
|
r rF r Ú
|
complex128r rË ÚcomplexrÉ rÊ r> r} r r r Útest_zero_division s z&TestComplexDivision.test_zero_divisionc C s t jdd t jt jfD ]v}d}|D ]h}|d }|d }|d }|t|d |d |t|d |d }t|j|d t|j|d q&qW 5 Q R X d S )Nr1 r2 ))©r¶ r ©r¶ r ©r r· ©rÕ rÕ ©r r· )rÕ ©r· r ©r r¶ )rÕ ©r· r ©r r¶ )rÖ rÕ rÝ rØ )rÚ rÕ rÙ )rÜ rÕ r× r r r$ )r rF r rÒ rÓ r ÚrealÚimag©r r~ ÚdataZcasesÚnrw Úexr r r r Útest_signed_zeros s
|
,z%TestComplexDivision.test_signed_zerosc C s¶ t jdd t jt jfD ]}t }| d¡ | d¡ |D ]h}|d }|d }|d }|t|d |d |t|d |d }t|j|d t|j |d q<qW 5 Q R X d S )Nr1 r2 )©ç @r rå rÛ )©r ræ rç rÛ r r r$ )
|
r rF r rÒ r¹ ÚappendrÓ r rÞ rß rà r r r Ú test_branches¶ s
|
|
,z!TestComplexDivision.test_branchesN)r+ r, r- rÔ rä ré r r r r rÑ s rÑ c @ s¤ e Zd Zdd Zdd Zdd Zdd Zejj e
|
d d
|
d d Z ejj e e j¡e e j¡kd d
|
ejj e ¡ d¡dd
|
dd Zdd Zdd Zdd ZdS )ÚTestConversionc C s ddddddg}ddd d
|
d d g}d t jt jfD ]&}t j||d }tdd |D | q.t j|d d
t jd }tdd |D |d d
d S )Ng .Ag ¢mBg ÈNgmÁ«Cg .Ág ¢mÂg ÈNgmÁ«Ãi@B l J)£l ÈNÎZoiÀ½ðÿlýÿÿÿ J)£lüÿÿÿ ÈNÎZor c S s g | ] }t |qS r ©rº ©r Ú_mr r r r Ù s z5TestConversion.test_int_from_long.<locals>.<listcomp>r c S s g | ] }t |qS r rë rì r r r r Ü s )r rY r r r r )r r" Úlir» r r r r Útest_int_from_longÔ s z!TestConversion.test_int_from_longc
|
C sÔ dD ]L}t t¡ tjt |¡jd |d}W 5 Q R X t |¡j}t||k qtj d D ]2}tjt |¡j|d}t |¡j}t||k q\tj d D ]4}t
|
|¡ t |¡j¡}t |¡j}t||k qd S )NZbBhHr r r© ) r? ÚwarnsrH r r ÚiinfoÚmaxÚminr r¦ r rK )r ÚcoderO rÁ r r r Útest_iinfo_long_valuesÞ s $ z%TestConversion.test_iinfo_long_valuesc C s2 dd }t jt jt jt jfD ]}tt|| q d S )Nc S s | t | ¡jd d S ©Nr )r rñ rò r r r r Úoverflow_error_funcð s zDTestConversion.test_int_raise_behaviour.<locals>.overflow_error_func)r Úint_ÚuintÚlonglongÚ ulonglongr Ú OverflowError)r r÷ rô r r r Útest_int_raise_behaviourï s z'TestConversion.test_int_raise_behaviourc C sb t t j¡}ttt| t :}| t j¡ t t j¡}ttt| t
|
t |j d W 5 Q R X d S rö ) r r rÉ r rü rº r ÚrecordÚComplexWarningÚ clongdoubler ÚlenÚlog©r r¡ rÌ r r r Ú!test_int_from_infinite_longdoubleö s z0TestConversion.test_int_from_infinite_longdoublezTest is PyPy only (gh-9972)©Úreasonc C sb t t j¡}tt|j t :}| t j¡ t t j¡}tt|j t
|
t |j d W 5 Q R X d S rö ) r r rÉ r rü Ú__int__r rþ rÿ r r r r r r r r Ú)test_int_from_infinite_longdouble___int__ s z8TestConversion.test_int_from_infinite_longdouble___int__z long double is same as doubleÚppczIBM double doublec C sP t t j¡jd }dt d¡| }dd| }t|t jk tt|| d S )Nr i¤ r$ ) r ÚfinfoÚdoubleZmaxexpr r rÉ r rº )r r Zhuge_ldZhuge_ir r r Ú test_int_from_huge_longdouble
|
s
|
z,TestConversion.test_int_from_huge_longdoublec C s4 t d¡}tt|d t d¡}tt|d d S )Nç ø?r g %Àiöÿÿÿ)r r r rº )r r¡ r r r Útest_int_from_longdouble s
|
|
z'TestConversion.test_int_from_longdoublec C s t jd D ]À}tdt jd|dd kd|f tdt jd|dd k d|f t jd D ]j}tt jd|dd t jd|dd kd||f tt jd|dd t jd|dd k d||f q^q
|
dD ]}td t jd|dd k d|f td t jd|dd k d|f td t jd|dd kd|f d
|
D ]}tt jd|dd t jd |dd kd||f tt jd|dd t jd |dd k d||f tt jd|dd t jd |dd kd||f qBqÐd t jd D ] }tdt jd |dd kd|f tdt jd |dd k d|f td t jd |dd kd|f d t jd D ]}tt jd|dd t jd |dd kd||f tt jd|dd t jd |dd k d||f tt jd |dd t jd |dd kd||f qnqòd S ) Nr© r r r r ztype %s failedztype %s and %s failedZBHILQPr/ ZbhilqpZbhlqprª )r r¦ r r )r rU rV r r r Ú&test_numpy_scalar_relational_operators sL "$$
|
ÿ&
|
ÿ
|
"$"$
|
ÿ&
|
ÿ$
|
ÿ
|
"$"$
|
ÿ&
|
ÿ$
|
ÿz5TestConversion.test_numpy_scalar_relational_operatorsc C s t jdd}t ddt¡ tt d¡d k tt d¡d k tt d¡d k tt d¡d k tt d¡d k tt d¡d k W 5 Q R X tt |dk tt
|
t d¡d ¡ d S ) NT)rþ ÚalwaysÚ r ÚtestZNaTr ) rD rE ÚfilterwarningsÚ FutureWarningr r rZ Zstr_Z
|
datetime64r Úequal)r Úwr r r Ú test_scalar_comparison_to_noneF s z-TestConversion.test_scalar_comparison_to_noneN)r+ r, r- rï rõ rý r r? ÚmarkZskipifr r r r
|
r r ÚplatformÚmachineÚ
|
startswithr r r r r r r r rê Ó s
|
|
|
ÿÿ )rê c @ s e Zd Zdd Zdd ZdS )ÚTestReprc C sè t |¡}|j|j }|j}t |¡jd }dD ]²}t jdg| t jd}|dkrt|d }d|d } d| > ||< n2|dkr|d }d|d } d| > ||< ntd | |¡d }
|
t
|
|
|
} |t | } | dkrØ|
|
d
|
k s0t |
|
| q0d S ) Né )ú small denormú
|
small normr r r rb r r Zhmmg0ä.ÿ++) r r
|
ZnexpZnmantr Úitemsizer Zuint8r rv ÚreprÚevalr ) r r~ r
|
Zlast_fraction_bit_idxZlast_exponent_bit_idxZ storage_bytesÚwhichZconstrÚbyteZbytebitÚvalZval_reprZval2r r r Ú_test_type_reprf s(
|
zTestRepr._test_type_reprc C s t jt jfD ]}| |¡ q d S r )r rZ rY r& )r r~ r r r Útest_float_repr s zTestRepr.test_float_reprN)r+ r, r- r& r' r r r r r e s r c @ s e Zd Zdd Zdd ZdS )Ú
|
TestSizeOfc C s* t D ] }|d}tt |¡|jk qd S )Nr )r r ÚsysÚ getsizeofÚnbytes)r rK r¡ r r r Útest_equal_nbytes s z TestSizeOf.test_equal_nbytesc C s t ¡ }tt|jd d S )Nr )r rZ r r Ú
|
__sizeof__)r rw r r r Ú
|
test_error s zTestSizeOf.test_errorN)r+ r, r- r, r. r r r r r( s r( c @ s e Zd Zdd Zdd ZdS )Ú TestMultiplyc C s& t tjd }dh}t tjd | | }|dh8 }ttfD ]ä}|dddg}|D ]@}t |¡ d¡}t|| |t| t|| t|| qR|D ]N}t |¡ ¡ }tt t
|
t j |||t| tt t
|
t j ||t|| q|D ]2}t |¡ ¡ }t tt j || t tt j || qìq<d S )Nr© ú?ZAllÚVr r$ r )Úsetr r¦ r¹ Útupler rK r rº r rH r9 Úmulr r )r Zaccepted_typesZdeprecated_typesZforbidden_typesZseq_typeÚseqZ
|
numpy_typer) r r r Útest_seq_repeat s2 ÿ
|
|
þ
|
þz TestMultiply.test_seq_repeatc C s¤ G dd d}|t d¡tt d¡fD ]t}t|t d¡ t dd¡ tt d¡| t dd¡ t|t d¡ t dd¡ tt d¡| t dd¡ q*d S )Nc @ s e Zd Zdd Zdd ZdS )zCTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLikec S s
|
|| _ d S r ©Úarr)r r8 r r r Ú__init__º s zLTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLike.__init__c S s | j S r r7 )r r r r Ú __array__¼ s zMTestMultiply.test_no_seq_repeat_basic_array_like.<locals>.ArrayLike.__array__N)r+ r, r- r9 r: r r r r Ú ArrayLike¹ s r; r g @)r ÚonesÚ
|
memoryviewr rZ Úfullrø )r r; Zarr_liker r r Ú#test_no_seq_repeat_basic_array_likeµ s z0TestMultiply.test_no_seq_repeat_basic_array_likeN)r+ r, r- r6 r? r r r r r/ s r/ c @ s e Zd Zdd Zdd ZdS )Ú TestNegativec C s& t jdt jdd }tttj| d S ©Nr r )r r< Úbool_r r r9 Úneg©r r r r r Útest_exceptionsÈ s z TestNegative.test_exceptionsc C s t jd t jd }t |}| t¡ |D ]f}t jd|dd }|t jd kr|t |¡j}|t |¡j }t
|
t |¡| q*t
|
t |¡| d q*W 5 Q R X d S )Nr© ÚAllFloatr r r¤ r ) r r¦ r rÇ rÈ r< r rK rñ rò r r9 rC )r r rÌ rn r Ústrò r r r Ú test_resultÌ s
|
zTestNegative.test_resultN©r+ r, r- rE rH r r r r r@ Ç s r@ c @ s e Zd Zdd Zdd ZdS )Ú TestSubtractc C s( t jdt jdd }tttj|| d S rA )r r< rB r r r9 ÚsubrD r r r rE Ú s z TestSubtract.test_exceptionsc C sb t jd t jd }t >}| t¡ |D ](}t jd|dd }tt ||¡d q*W 5 Q R X d S )Nr© rF r r r ) r r¦ r rÇ rÈ r< r r9 rK )r r rÌ rn r r r r rH Þ s
|
zTestSubtract.test_resultNrI r r r r rJ Ù s rJ c @ sH e Zd Zdd Zej dee ¡dd Z ej dee ¡dd Z
|
dS ) ÚTestAbsc C sÌ |d}t ||d |d}||}t |d |d}||}t |d |t |¡j}t |||j t 0}| t¡ |t |¡j}t |||j W 5 Q R X |t |¡j }t |||j d S )Ng ø¿r r¶ r· )
|
r r r
|
rò rÞ r rÇ Ú UserWarningZtinyró )r ZabsfuncZ
|
test_dtyper¡ rO rÌ r r r Ú_test_abs_funcè s
|
|
|
zTestAbs._test_abs_funcr c C sR t jdkrB|tjkrBt t ¡ d¡d ¡t d¡k rBt j
|
dd | t |¡ d S ©NÚcygwinú-r z3.3.0z4absl is computed in double precision on cygwin < 3.3r © r) r r r r ÚparseÚreleaseÚsplitÚVersionr? r@ rN Úabs©r r r r r Útest_builtin_absþ s ÿÿÿýÿzTestAbs.test_builtin_absc C sT t jdkrB|tjkrBt t ¡ d¡d ¡t d¡k rBt j
|
dd | tj |¡ d S rO rR rX r r r Útest_numpy_abs s ÿÿÿýÿzTestAbs.test_numpy_absN) r+ r, r- rN r? r Ú parametrizeÚfloating_typesÚcomplex_floating_typesrY rZ r r r r rL ç s
|
|
rL c @ sF e Zd Zej dejd ¡ejjdej ej
|
gddgddd Z d S )
|
Ú TestBitShiftsÚ type_coder© rL z>>z<<©Úidsc C sÀ t |¡}|jd }dD ]¢}||d fD ]}t |¡ |¡d }| |¡}|||} |dk rr|tjkrrt| d n
|
t| d t j|gd |d}
|
t j|gd |d} ||
|
| } t| | q(qd S )
|
zA Shifts where the shift amount is the width of the type or wider r )r éûÿÿÿrs r r r/ é r N) r r r r r¾ rK r9 Úrshiftr ) r r_ rL rn Únbitsr% ÚshiftZval_sclZ shift_sclZres_sclZval_arrZ shift_arrZres_arrr r r Útest_shift_all_bits s
|
|
|
|
|
|
z!TestBitShifts.test_shift_all_bitsN) r+ r, r- r? r r[ r r¦ r9 rd Úlshiftrg r r r r r^ s
|
ÿr^ c @ s` e Zd Zej dejd ¡dd Zej dejd ¡dd Z ej dejd ¡d d
|
Z
|
d S ) ÚTestHashr_ r© c C s6 t |¡j}tdD ] }t|t||kstqd S )Nr´ )r r rK r' Úhashr7 )r r_ Úscalarr) r r r Útest_integer_hashes7 s z TestHash.test_integer_hashesrF c C s¨ t |¡j}t jt jddfD ]J}||}|jjdkr>t|}nt|}||ksRtt |t |ks tq t tt j
|
t tt j
|
kr¤t |t j
|
t |t j
|
ks¤td S )Nr g @r. ) r r rK ÚpirÉ r8 rÓ r¼ r7 rj rÊ ©r r_ rk r% Z numpy_valr r r Ú test_float_and_complex_hashes= s
|
z&TestHash.test_float_and_complex_hashesÚComplexc C sN t |¡j}t jd t jd ddfD ]$}||}tt|t|ks$tq$d S )Nù ð?y @y @ ð?)r r rK rm rÉ rj rÓ r7 rn r r r Útest_complex_hashesN s z TestHash.test_complex_hashesN) r+ r, r- r? r r[ r r¦ rl ro rr r r r r ri 6 s
|
|
ri c c s. t ¡ }zt | ¡ d V W 5 t |¡ X d S r )r) ÚgetrecursionlimitÚsetrecursionlimit)râ rx r r r ÚrecursionlimitW s
|
|
|
ru c C s@ z&t d || |d W 5 Q R X W n tk
|
r: Y nX d S ©NéÈ r ©ru r ©rx rL Útype_r r r Útest_operator_object_lefta s
|
|
r{ c C s@ z&t d ||d| W 5 Q R X W n tk
|
r: Y nX d S rv rx ry r r r Útest_operator_object_rightl s
|
|
r| c C s0 z| |d|d W n t k
|
r* Y nX d S rö )r )rL Ztype1Ztype2r r r Útest_operator_scalarsw s r} rL r% l c C sX z| t d¡| W n tk
|
r( Y nX z| |t d¡ W n tk
|
rR Y nX d S ©Nr )r r r ©rL r% r r r Útest_longdouble_inf_loop s r c C sX z| t d¡| W n tk
|
r( Y nX z| |t d¡ W n tk
|
rR Y nX d S r~ )r r r r r r r r Útest_clongdouble_inf_loop s r r r© Ú operationc C s || S r r ©ró rò r r r Ú<lambda> ó r c C s | | S r r r r r r r r
c C s || S r r r r r r r ¡ r
ú+rQ Ú*r` c C sT t | ¡j}|t | ¡j}|t | ¡j}tjtdd ||| W 5 Q R X d S )Núoverflow encounteredrÎ ) r r rK rñ ró rò r? rð rÈ )r r rG ró rò r r r Ú&test_scalar_integer_operation_overflow s
|
r ZIntegerc C s | S r r ©ró Úneg_1r r r r r
c C s t | S r )rW r r r r r ® r
c C s | | S r r r r r r r ¯ r
c C s | | S r r r r r r r ° r
zbroken on some platformsr )ZmarksrC rW z//c C sL t | ¡j}|t | ¡j}|d}tjtdd ||| W 5 Q R X d S )Nr/ r rÎ )r r rK rñ ró r? rð rÈ )r r rG ró r r r r Ú#test_scalar_signed_integer_overflow« s
|
|
r r¤ c C sJ t | ¡ d¡}tjtdd | W 5 Q R X t | ¡ d¡}| d S )Nr r rÎ r ©r r rK r? rð rÈ )r r% Úzeror r r Ú%test_scalar_unsigned_integer_overflow½ s
|
r c C s | | S r r ©r% r r r r r È r
c C s | | S r r r r r r r É r
ú%c C sD t | ¡j}|d}|d}tjtdd ||| W 5 Q R X d S )Néd r zdivide by zerorÎ r )r r rG r% r r r r Ú'test_scalar_integer_operation_divbyzeroÆ s
|
r Ú__lt__Ú__gt__TÚ__le__Ú__ge__Ú__eq__Ú__ne__Ú __floordiv__Ú __rfloordiv__FÚ __truediv__Ú __rtruediv__Ú__add__Ú__radd__Ú__mod__Ú__rmod__Ú__mul__Ú__rmul__Ú__pow__Ú__rpow__Ú__sub__Ú__rsub__Ú__op__Ú__rop__ÚcmpÚsctypec sÜ G dd d| }G dd d| } fdd}fdd}t d | f ||i} ||d
|
|d }
|
t |
|
| ks~t |
|
tjks~t||d
|
|d |d
|
d kst|| d
|
|d ks¸t||d
|
| d |d
|
d ksØtd S ) aß
|
This test covers scalar subclass deferral. Note that this is exceedingly
|
complicated, especially since it tends to fall back to the array paths and
|
these additionally add the "array priority" mechanism.
|
|
The behaviour was modified subtly in 1.22 (to make it closer to how Python
|
scalars work). Due to its complexity and the fact that subclassing NumPy
|
scalars is probably a bad idea to begin with. There is probably room
|
for adjustments here.
|
c @ s e Zd ZdS )z+test_subclass_deferral.<locals>.myf_simple1N©r+ r, r- r r r r Ú myf_simple1ò s r c @ s e Zd ZdS )z+test_subclass_deferral.<locals>.myf_simple2Nr¬ r r r r Ú myf_simple2õ s r® c s S r r ©r Úother©r¨ r r Úop_funcø s z'test_subclass_deferral.<locals>.op_funcc s S r r r¯ ©r© r r Úrop_funcû s z(test_subclass_deferral.<locals>.rop_funcÚmyf_opr r$ N)rK r rB r7 ) r« r¨ r© rL rª r r® r² r´ rµ rO r ©r¨ r© r Útest_subclass_deferralå s r· c C s. t d¡} | d dkstd| dks*td S )Nr rq y ð? ð?)r r r7 )r¡ r r r Útest_longdouble_complex s
|
r¸ Úsubtypec
|
s¾ fdd}fdd}t d| f ||dd i}||dt d¡ ksNt|t d¡|dksjt|tjtjhkr| tkrd S krd S t d| f|i}||dt d¡}|| dt d¡} || ksØtt |t | ksìt|t d¡|d}|t d¡| d} || ks"tt |t | ks8t||dt
|
d¡}|| dt
|
d¡} || ksntt |t | kst|t d¡|d}|t
|
d¡| d} || ksºtd S ) Nc s S r r r¯ r± r r r² s z)test_pyscalar_subclasses.<locals>.op_funcc s S r r r¯ r³ r r r´ s z*test_pyscalar_subclasses.<locals>.rop_funcÚmytZ__array_ufunc__r r$ ) rK r rY r7 r9 rÆ ÚfloordivrÓ r| rZ r )
|
r¹ r¨ r© rL rª r² r´ rº rO r^ r r¶ r Útest_pyscalar_subclasses s6 ÿ r¼ )uÚ
|
contextlibr) rD r r9 r Z numpy.compatr r? Z
|
hypothesisr r Zhypothesis.strategiesr Zhypothesis.extrar Zhynpr Z numpy.testingr r r r
|
r r r r r rB r$ ZubyteÚshortZushortZintcZuintcrø rù rú rû Zsingler r ZcsingleZcdoubler r ZfloatingÚ__subclasses__r\ Zcomplexfloatingr] ÚobjectZobjecty_thingsr= r< ÚeqÚner: r; r( r» rÆ r4 rA rK ÚtruedivZ reasonable_operators_for_scalarsr rQ r ZslowZarraysZ scalar_dtypesrT rW r[ r| rZ r_ r` rz r£ r§ r¨ rÑ rê r r( r/ r@ rJ rL r^ ri Úcontextmanagerru r{ r| r} r r r¦ r ÚparamrC r r r Zops_with_namesrY r· r¸ r¼ rº rÓ Z_no_nep50_warningr¼ r r r r Ú<module> s. , ý
|
|
|
ý '
|
þ ÿ
|
.QmH ! 33 !
|
þ
|
þ
|
þ
|
ýý
|
ÿü
|
ú
|
|
þþ
|
ò $
|
