U
|
O ±dä9 ã @ s¦ d dl mZmZmZmZmZmZmZmZm Z m
|
Z
|
m Z m Z m Z d dlmZmZmZmZmZ G dd deZG dd de ZG dd dZG d d
|
d
|
ZG d d d Zd S )é ) ÚlogspaceÚlinspaceÚ geomspaceÚdtypeÚarrayÚsctypesÚarangeÚisnanÚndarrayÚsqrtÚ nextafterÚstackÚerrstate)Úassert_Ú assert_equalÚ assert_raisesÚassert_array_equalÚassert_allclosec @ sL e Zd Zdd Zdd ZeZdd Zdd Zd d
|
ZeZ d d Z
|
d d Z dS )ÚPhysicalQuantityc C s t | |¡S ©N)ÚfloatÚ__new__)ÚclsÚvalue© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_function_base.pyr s zPhysicalQuantity.__new__c C s" t t|t tt|t| S r ©r Ú
|
isinstancer r ©ÚselfÚxr r r Ú__add__ s zPhysicalQuantity.__add__c C s" t t|t tt| t| S r r r r r r Ú__sub__ s zPhysicalQuantity.__sub__c C s" t t|t tt|t| S r r r r r r Ú__rsub__ s zPhysicalQuantity.__rsub__c C s t t|t| S r ©r r r r r r Ú__mul__ s zPhysicalQuantity.__mul__c C s t t| t| S r r$ r r r r Ú__div__ s zPhysicalQuantity.__div__c C s t t|t| S r r$ r r r r Ú__rdiv__" s zPhysicalQuantity.__rdiv__N) Ú__name__Ú
|
__module__Ú __qualname__r r! Ú__radd__r" r# r% Ú__rmul__r& r' r r r r r
|
s r c @ s e Zd ZdZdS )ÚPhysicalQuantity2é
|
N)r( r) r* Z__array_priority__r r r r r- & s r- c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestLogspacec C s t dd}tt|dk t dddd}t|d dk t dddd }t|d dk t ddd
|
d}t|d d dd dddg d S )Nr é é2 éd ©Únuméÿÿÿÿé@B F©Úendpointé é r. ç @@ç Ã@ç jø@ç .A)r r Úlenr ©r Úyr r r Ú
|
test_basic, s
|
zTestLogspace.test_basicc s t ddg}t ddg t| d}tdd t| D dd }t|| t| d
|
d}t fd d|D dd }t|| t| dd d }t||j d S ) Nç ç ð?g @ç @r0 c S s g | ]\}}t ||d qS )r0 ©r ©Ú.0Ú_startÚ_stopr r r Ú
|
<listcomp>: s ÿz6TestLogspace.test_start_stop_array.<locals>.<listcomp>r: ©Zaxisr c s g | ]}t | d dqS )r r0 rF ©rH rI ©Ústopr r rK > s ÿr5 )r r r Úzipr ÚT©r ÚstartÚt1Út2Út3Út4Út5r rN r Útest_start_stop_array6 s$ ÿÿ
|
ÿÿ
|
z"TestLogspace.test_start_stop_arrayc C s^ t dddd}t|jtd t dddd}t|jtd t dddd}t|jtd d S ©Nr r0 Úfloat32©r Úfloat64Úint32)r r r r@ r r r Ú
|
test_dtypeD s zTestLogspace.test_dtypec C s* t d}t d}tt||tdd d S ©NrD g @)r r r ©r ÚaÚbr r r Útest_physical_quantitiesL s z%TestLogspace.test_physical_quantitiesc C sx t d t¡}t d t¡}t||}t|tks6tt|tdd t||d}t|tksbtt|tddd d S ©Nr: r9 rD rE )r Úviewr- r ÚtypeÚAssertionErrorr ©r rb rc Zlsr r r Ú test_subclassQ s
|
zTestLogspace.test_subclassN)r( r) r* rB rY r_ rd rj r r r r r/ * s
|
|
r/ c @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dS )Ú TestGeomspacec C sþ t dd}tt|dk t dddd}t|d dk t dddd }t|d dk t ddd
|
d}t|dd dd d ddg t dddd}t|dddg t|jd t dddd}t|dddg t|jd t dddd}t|dddg t|jd d S )Nr: r> r1 r2 r3 r5 r6 Fr7 r9 r. r; r< r= é é é é r éÿÿÿiöÿÿÿ)r r r? r r Úimagr@ r r r rB ^ s"
|
zTestGeomspace.test_basicc C s d}d}t ||dd}t|d | t ||ddd}t|d | t ||dd}t|d | t|d | t ||ddd}t|d | d S )
|
Ng333333Ó?gÍÌÌÌÌL4@r: r3 r F©r4 r8 rn r5 )r r ©r rS rO rA r r r Ú,test_boundaries_match_start_and_stop_exactlyt s z:TestGeomspace.test_boundaries_match_start_and_stop_exactlyc C s¨ t dd tdddd}W 5 Q R X t|d d tt|d d
|
¡ t|d d t dd tdddd d }W 5 Q R X t|d d tt|d d
¡ d S )NÚignore)Úinvalidéýÿÿÿrn ro r3 r g Àr: r5 g @Frr )r r r r r Úallr@ r r r Útest_nan_interior s z TestGeomspace.test_nan_interiorc C sØ t dddd}t|dddddg t|jd t d d
|
dd}t|d d d d d
|
g t|jd t dddd}t|ddddg t dddd}t|ddddg t dddtd}t|dddg t dd d}t|dd td dtd d g t dd d}t|ddtd dtd d g t d d d}t|d d td dtd d g t dd d}t|dd td dtd d g t d!d"d#}t|d!d$d%d&d'd(d"g t dd)d }t|dd)g t d)dd }t|d)dg d S )*Nù ð?ù 0@é r3 y @y @y @r ù Àù @tÀy (Ày BÀy [Àù ð? ð?y @@ @@ro y $@ $@y Y@ Y@y ð¿ ð?y @À @@y $À $@y YÀ Y@r5 r: rn ©r4 r y @y À rw rm y @ y Ày À Ày @ @r9 y9ð¬ÕÒ¿ÅÍ~¡Ày{xCØøµ @Vßã©]Ày¬é´MuU@èPrôN Àyä¾0ô§@5S¹¬á¿y
|
mµ^>@w@ñ@éûÿÿÿ)r r r ÚrealÚcomplexr r@ r r r Ú test_complex s@ $ $ $ $ ý zTestGeomspace.test_complexc C s t dddd}t|jtd t dddd}t|jtd t dddd}t|jtd t ddtd}t|jtd t ddtd}t|jtd d S ) Nr: r> r[ r\ r] r^ Zfloat_r )r r r r r r@ r r r r_ ¾ s zTestGeomspace.test_dtypec
|
C s¾ t ddgdd}t ddgdd}t ddgd d}t|d
|
|d d }t|d
|
|d d }t|d
|
|d d }td dd }tddd }tddd } t||dd t||dd t|| dd d S )Néx r2 Úint8r\ éÿÿÿrp é° éè Úuint16r r: r| ç ^@ç Y@ç ^Àç YÀç À@r; g{®Gáz?)Zrtolgñhãµøä>)r r r ©
|
r Úlim1Úlim2Zlim3rT rU rV rW rX Zt6r r r Ú test_start_stop_array_scalarÌ s z*TestGeomspace.test_start_stop_array_scalarc s® t ddddddg}t ddd d
|
d d g t| d }tdd t| D d d}t|| t| d d }t fdd|D d d}t|| t| d dd}t||j d S )NrD g @@rz r} r r5 r< ç @r{ r~ y Ã@ Ã@r: r| c S s g | ]\}}t ||d qS ©r| ©r rG r r r rK á s ÿz7TestGeomspace.test_start_stop_array.<locals>.<listcomp>rL r c s g | ]}t | d dqS ©r r| r rM rN r r rK å s ÿ)r r r rP r rQ rR r rN r rY Ü s$ ÿÿ
|
ÿÿ
|
z#TestGeomspace.test_start_stop_arrayc C s* t d}t d}tt||tdd d S r` )r r r ra r r r rd ë s z&TestGeomspace.test_physical_quantitiesc C sx t d t¡}t d t¡}t||}t|tks6tt|tdd t||d}t|tksbtt|tddd d S re )r rf r- r rg rh r )r rb rc Zgsr r r rj ð s
|
zTestGeomspace.test_subclassc C s. t ttdd t ttdd t ttdd d S )Nr r. )r Ú
|
ValueErrorr ©r r r r Ú test_boundsú s zTestGeomspace.test_boundsN) r( r) r* rB rt ry r r_ r rY rd rj r r r r r rk \ s (
|
rk c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd S ) Ú TestLinspacec C sl t dd}tt|dk t dddd}t|d dk t dddd }t|d dk ttt dddd d S )
|
Nr r. r1 rm r2 r3 r5 Fr7 )r r r? r r r@ r r r rB s
|
zTestLinspace.test_basicc C s6 t tddd}t|dgk| tttdddd d S )Nr r: rC g @r3 )Úlistr r r Ú TypeErrorr@ r r r Ú test_corner s zTestLinspace.test_cornerc C sB t dddj}t dddj}t dddj}t|| t|| d S )Nr r: rm )r r r )r rT rU rV r r r Ú test_type s
|
|
zTestLinspace.test_typec C s^ t dddd}t|jtd t dddd}t|jtd t dddd}t|jtd d S rZ )r r r r@ r r r r_ s zTestLinspace.test_dtypec
|
C s² t ddgdd}t ddgdd}t ddgd d}t|d
|
|d d }t|d
|
|d d }t|d
|
|d d }td dd }tddd }tddd } t|| t|| t|| d S )Nr r2 r r\ r
rp r r r r r: r| r r r r r r; )r r r r r r r r s
|
|
z)TestLinspace.test_start_stop_array_scalarc s¦ t ddgdd}t ddgdd t| d}tdd t| D d
|
d }t|| t| d d}t fd d |D d
|
d }t|| t| ddd }t||j d S )Nr r
r r\ r2 rp r| c S s g | ]\}}t ||d qS r ©r rG r r r rK 1 s ÿz6TestLinspace.test_start_stop_array.<locals>.<listcomp>r: rL r c s g | ]}t | d dqS r r rM rN r r rK 5 s ÿr5 )r r r rP r rQ rR r rN r rY - s$ ÿÿ
|
ÿÿ
|
z"TestLinspace.test_start_stop_arrayc C sT t ddd}tdddddg}t ddd}tdd d
|
d d g}t|| t|| d S ) Ny ð? @y @ @r| y ø? @y @ @y @ @rz r. y @ è?y @ à?y @ Ð?y $@ )r r r )r r rT r rU r r r r ; s
|
zTestLinspace.test_complexc C s* t d}t d}tt||tdd d S )NrC rD )r r r ra r r r rd C s z%TestLinspace.test_physical_quantitiesc C sx t d t¡}t d t¡}t||}t|tks6tt|tdd t||d}t|tksbtt|tddd d S )Nr r: rC rD )r rf r- r rg rh r ri r r r rj H s
|
zTestLinspace.test_subclassc C sH G dd d}|t ddd}|t ddd}tt||tdd d S )Nc @ s, e Zd ZdZdd Zedd Zdd ZdS ) z3TestLinspace.test_array_interface.<locals>.Arrayisha
|
A generic object that supports the __array_interface__ and hence
|
can in principle be converted to a numeric scalar, but is not
|
otherwise recognized as numeric, but also happens to support
|
multiplication by floats.
|
|
Data should be an object that implements the buffer interface,
|
and contains at least 4 bytes.
|
c S s
|
|| _ d S r ©Ú_data)r Údatar r r Ú__init__b s z<TestLinspace.test_array_interface.<locals>.Arrayish.__init__c S s dd| j ddS )Nr ú<i4rn )ÚshapeZtypestrr£ Úversionr¡ r r r r Ú__array_interface__e s ÿzGTestLinspace.test_array_interface.<locals>.Arrayish.__array_interface__c S s | S r r )r Úotherr r r r% j s z;TestLinspace.test_array_interface.<locals>.Arrayish.__mul__N)r( r) r* Ú__doc__r¤ Úpropertyr¨ r% r r r r ÚArrayishW s
|
|
|
r¬ r: r¥ r\ r| )r r r )r r¬ ZoneZfiver r r Útest_array_interfaceR s z!TestLinspace.test_array_interfacec
|
C sB t d D ]4}t|d|dd }tttd|dd|d qd S )Nr r r: r| r. F)r8 r )r r r Úanyr )r ZftyperO r r r Útest_denormal_numberst s z"TestLinspace.test_denormal_numbersc C s8 t dD ]*}ttd||d tdt|d td qd S )Nr r r: r\ )Úranger r Úintr )r Újr r r Útest_equivalent_to_arange{ s ÿz&TestLinspace.test_equivalent_to_arangec C sÖ dD ]Ì}dD ]Â}t dd||dd}t|tr8t|dks<t|dkrz|rPdd gndd
|
g}t|d | t|d |d q |dkr¦|s¦t|d dg t|d d q t|d dgd |
t|d s tq qd S ) N)r r: rm )FTr r: T)r8 Zretsteprm rC rD g à?)r r Útupler? rh r r r )r r4 ZeptrA Z y0_expectr r r Ú test_retstep s zTestLinspace.test_retstepc C s< t ddd}t ddd}t||d}t|t dddg d S ) Nr: ÚOr\ rm rn rD g ø?r )r r r rs r r r Ú test_object s zTestLinspace.test_objectc C s: t dddtd}tddddddddgtd}t|| d S ) Nr5 rn rl r r r: rm r\ )r r± r r )r rA Útr r r Útest_round_negative s z TestLinspace.test_round_negativeN)r( r) r* rB r r r_ r rY r rd rj r r¯ r³ rµ r· r¹ r r r r r s
|
"r N)Únumpyr r r r r r r r r
|
r r r r Z numpy.testingr r r r r r r r- r/ rk r r r r r Ú<module> s < 2 %
|
