U
|
P ±dí< ã @ sN d dl Zd dlmZmZmZmZ d dlmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZ dd ZG dd dZG dd d ZG d
|
d d ZG d d d ZG dd dZG dd dZG dd dZG dd dZG dd dZ G dd dZ G dd dZ G d d d Z G d d d Z G d d! d!Z!G d"d# d#Z"G d$d% d%Z#G d&d' d'Z$dS )(é N)Úassert_Ú assert_equalÚassert_array_equalÚ assert_raises) Ú common_typeÚ mintypecodeÚisrealÚ iscomplexÚisposinfÚisneginfÚ
|
nan_to_numÚ isrealobjÚ iscomplexobjÚasfarrayÚ real_if_closec C s t t | ¡| d S ©N)r ÚnpÚall)Úx© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_type_check.pyÚ
|
assert_all s r c @ s e Zd Zdd ZdS )ÚTestCommonTypec C s t jddgddggt jd}t jddgddggt jd}t jddgddggt jd}t jddgddggt jd}t jddgdd ggt jd}t jddgdd ggt jd}tt |t jk tt |t jk tt |t jk tt |t jk tt |t jk tt |t jk d S )
|
Né é é é ©Údtypey ð? @y @ @y @ @y @ @)
|
r ÚarrayÚint32Úfloat16Úfloat32Úfloat64ZcsingleZcdoubler r )ÚselfZai32Zaf16Zaf32Zaf64ZacsZacdr r r Ú
|
test_basic s zTestCommonType.test_basicN©Ú__name__Ú
|
__module__Ú __qualname__r% r r r r r s r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestMintypecodec C sT dD ]}t t|d qt tdd t tdd t tdd t tdd d S )NÚ1bcsuwilÚdÚfÚFÚD©r r ©r$ Úityper r r Útest_default_1! s z TestMintypecode.test_default_1c C s6 dD ]L}t t|d d t t|d d t t|d d t t|d d qt tdd t tdd t tdd t td d t td
|
d t td d t td d t td d t tdd t tdd t tdd t tdd t tdd t tdd t tdd t tdd d S )Nr+ r- r, r. r/ ÚffÚfdZfFZfDZdfÚddZdFZdDZFfZFdZFFZFDZDfZDdZDFZDDr0 r1 r r r Útest_default_2) s* z TestMintypecode.test_default_2c C s t tdd t tdd t tdd t tdd t tdd t tdd t td
|
d t td d t td d d S ) NZfdFr/ ZfdDZfFDZdFDZifdr, ZifFr. ZifDZidFZidDr0 ©r$ r r r Útest_default_3B s z TestMintypecode.test_default_3N)r' r( r) r3 r7 r9 r r r r r* s r* c @ s e Zd Zdd ZdS )Ú TestIsscalarc C sP t t d¡ t t dg¡ t t d¡ t t d¡ t t d¡ d S )Nr )r y @g @)r r Zisscalarr8 r r r r% S s
|
zTestIsscalar.test_basicNr& r r r r r: Q s r: c @ s e Zd Zdd Zdd ZdS )ÚTestRealc C sx t j d¡}t|t |¡ t d¡}t |¡}t|| tt|t j d}t |¡}t || tt|t j d S )Né
|
r ©
|
r ÚrandomÚrandr Úrealr r Ú
|
isinstanceÚndarrayr ©r$ ÚyÚoutr r r Ú test_real] s
|
|
|
|
|
zTestReal.test_realc C s t j d¡dt j d¡ }t|jt |¡ t d¡}t |¡}t|j| tt|t j d}t |¡}t d| tt|t j d S ©Nr< ù ð?ù ð? ð?ç ð?r= rC r r r Ú
|
test_cmplxk s
|
|
|
|
zTestReal.test_cmplxN©r' r( r) rF rK r r r r r; [ s r; c @ s e Zd Zdd Zdd ZdS )ÚTestImagc C sx t j d¡}tdt |¡ t d¡}t |¡}td| tt|t j d}t |¡}t d| tt|t j d S )Nr< r r ©
|
r r> r? r Úimagr r rA rB r rC r r r rF | s
|
|
|
|
|
zTestImag.test_realc C s t j d¡dt j d¡ }t|jt |¡ t d¡}t |¡}t|j| tt|t j d}t |¡}t d| tt|t j d S rG rN rC r r r rK s
|
|
|
|
zTestImag.test_cmplxNrL r r r r rM z s rM c @ s e Zd Zdd Zdd ZdS )Ú TestIscomplexc C s0 t dddg¡}t|}tt j|dd d S )Néÿÿÿÿr r ©Zaxis)r r r r Zsometrue©r$ ÚzÚresr r r Ú test_fail s zTestIscomplex.test_failc C s, t dddg¡}t|}t|dddg d S ©Nù ð¿r r )r r r r rS r r r Ú test_pass s zTestIscomplex.test_passN)r' r( r) rV rY r r r r rP s rP c @ s e Zd Zdd Zdd ZdS )Ú
|
TestIsrealc C s, t dddg¡}t|}t|dddg d S )NrQ r rH r ©r r r r rS r r r rY ¨ s zTestIsreal.test_passc C s, t dddg¡}t|}t|dddg d S rW r[ rS r r r rV s zTestIsreal.test_failN)r' r( r) rY rV r r r r rZ ¦ s rZ c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestIscomplexobjc C s> t dddg¡}tt| t dddg¡}tt| d S ©NrQ r r rX )r r r r ©r$ rT r r r r% µ s zTestIscomplexobj.test_basicc C s t td t td d S )NrJ ù ð? ©r r r8 r r r Ú test_scalar» s z TestIscomplexobj.test_scalarc C s* t tdddg t tdddg d S )Nr r_ Tr r` r8 r r r Ú test_list¿ s zTestIscomplexobj.test_listc C s$ G dd d}| }t t| d S )Nc @ s e Zd Zedd ZdS )z5TestIscomplexobj.test_duck.<locals>.DummyComplexArrayc S s
|
t t¡S r )r r Úcomplexr8 r r r r Å s z;TestIscomplexobj.test_duck.<locals>.DummyComplexArray.dtypeN©r' r( r) Úpropertyr r r r r ÚDummyComplexArrayÄ s rf r` )r$ rf Údummyr r r Ú test_duckà s zTestIscomplexobj.test_duckc sL G dd dt j G fdddG fddd}| }tt| d S )Nc @ s e Zd ZdS )z4TestIscomplexobj.test_pandas_duck.<locals>.PdComplexN)r' r( r) r r r r Ú PdComplexÎ s ri c s* e Zd ZdZdZ ZdZdZe d¡Z
|
dS )z2TestIscomplexobj.test_pandas_duck.<locals>.PdDtypeÚcategoryNÚcz<c16Ú
|
complex128) r' r( r) ÚnameÚnamesÚtypeÚkindÚstrr r Úbaser )ri r r ÚPdDtypeÐ s rs c s e Zd Ze fddZdS )z2TestIscomplexobj.test_pandas_duck.<locals>.DummyPdc s S r r r8 ©rs r r r Ø s z8TestIscomplexobj.test_pandas_duck.<locals>.DummyPd.dtypeNrd r rt r r ÚDummyPd× s ru )r rl r r )r$ ru rg r )ri rs r Útest_pandas_duckË s
|
z!TestIscomplexobj.test_pandas_duckc C s. G dd dt }|dddg}tt| d S )Nc @ s e Zd Zedd ZdS )z8TestIscomplexobj.test_custom_dtype_duck.<locals>.MyArrayc S s t S r )rc r8 r r r r à s z>TestIscomplexobj.test_custom_dtype_duck.<locals>.MyArray.dtypeNrd r r r r ÚMyArrayß s rw r_ y @ y @ )Úlistr r )r$ rw Úar r r Útest_custom_dtype_duckÞ s z'TestIscomplexobj.test_custom_dtype_duckN) r' r( r) r% ra rb rh rv rz r r r r r\ ³ s r\ c @ s e Zd Zdd ZdS )Ú TestIsrealobjc C s> t dddg¡}tt| t dddg¡}tt| d S r] )r r r r r^ r r r r% é s zTestIsrealobj.test_basicNr& r r r r r{ è s r{ c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )Ú TestIsnanc C s. t d¡}t |¡dk}tt j|dd d S ©N©ç ð¿ç rJ r rR )r r Úisnanr r rS r r r Útest_goodvaluesò s
|
zTestIsnan.test_goodvaluesc C s8 t jdd" tt t d¡d ¡dk W 5 Q R X d S )NÚignore©Údivide©rJ r r ©r Úerrstater r r r8 r r r Ú test_posinf÷ s zTestIsnan.test_posinfc C s8 t jdd" tt t d¡d ¡dk W 5 Q R X d S )Nr r ©r r r r r8 r r r Ú test_neginfû s zTestIsnan.test_neginfc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr ©r
Úinvalid©r r r r r8 r r r Útest_indÿ s zTestIsnan.test_indc C s t t d¡dk d S )Nr r ©r r r r8 r r r Ú test_integer s zTestIsnan.test_integerc C s t t d¡dk d S )NrI r r r8 r r r Ú test_complex s zTestIsnan.test_complexc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr r y r r r r8 r r r Ú test_complex1 s zTestIsnan.test_complex1N©
|
r' r( r) r r r r r r r r r r r r| ð s r| c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )Ú TestIsfinitec C s. t d¡}t |¡dk}tt j|dd d S )Nr~ r r rR )r r Úisfiniter r rS r r r r s
|
z TestIsfinite.test_goodvaluesc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr r r r r ©r r r r r r8 r r r r s zTestIsfinite.test_posinfc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr r r r r r r8 r r r r s zTestIsfinite.test_neginfc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S ©Nr r r r r r r8 r r r r s zTestIsfinite.test_indc C s t t d¡dk d S )Nr ©r r r r8 r r r r " s zTestIsfinite.test_integerc C s t t d¡dk d S )NrI r r r8 r r r r % s zTestIsfinite.test_complexc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr r rI r r r r8 r r r r ( s zTestIsfinite.test_complex1Nr r r r r r s r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú TestIsinfc C s. t d¡}t |¡dk}tt j|dd d S r} )r r Úisinfr r rS r r r r 0 s
|
zTestIsinf.test_goodvaluesc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr r r r r ©r r r r r r8 r r r r 5 s zTestIsinf.test_posinfc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr r rJ r r r r8 r r r Útest_posinf_scalar9 s z TestIsinf.test_posinf_scalarc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr r r r r r r8 r r r r = s zTestIsinf.test_neginfc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S )Nr r r r r r r8 r r r Útest_neginf_scalarA s z TestIsinf.test_neginf_scalarc C s: t jddd" tt t d¡d ¡dk W 5 Q R X d S r r r8 r r r r E s zTestIsinf.test_indN) r' r( r) r r r r r r r r r r r - s r c @ s e Zd Zdd ZdS )Ú TestIsposinfc C s` t jddd tt d¡d }W 5 Q R X t|d dk t|d dk t|d dk d S ©Nr r ©r r r r r r r )r r r
|
r r ©r$ Úvalsr r r Ú test_genericL s
|
zTestIsposinf.test_genericN©r' r( r) r¤ r r r r r J s r c @ s e Zd Zdd ZdS )Ú TestIsneginfc C s` t jddd tt d¡d }W 5 Q R X t|d dk t|d dk t|d dk d S r )r r r r r r¢ r r r r¤ V s
|
zTestIsneginf.test_genericNr¥ r r r r r¦ T s r¦ c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )Ú TestNanToNumc C s" t jddd tt d¡d }W 5 Q R X t|d dk oLtt |d ¡ t|d dk t|d d ko~tt |d ¡ tt|t j t jddd tt d¡d d
|
d d d }W 5 Q R X t|d d
|
d g tt |ddg ¡ tt|t j t jddd t d¡d }W 5 Q R X t|dd}t||k t|d dk o\tt |d ¡ t|d dk t|d d kott |d ¡ tt|t j t jddd t d¡d }W 5 Q R X t|dd
|
d d d}t||k t|d d
|
d g tt |ddg ¡ tt|t j d S )Nr r r¡ r r ç _ Âr r g _ Br< é é ©ÚnanÚposinfÚneginfF)Úcopy)r¯ r¬ r r® )
|
r r r r r r r r ro rB )r$ r£ Úresultr r r r¤ ` s< "" ÿ $$ zTestNanToNum.test_genericc C sh t dg}t|t dgt¡ tt|tj t dgdddd}t|t dgt¡ tt|tj d S ©Nr r< r© rª r« )r r r r Úintr ro rB r¢ r r r Ú
|
test_array
s
|
zTestNanToNum.test_arrayc C sT t d}t|dk tt|tj t ddddd}t|dk tt|tj d S r± )r r r ro r Úint_r¢ r r r r s zTestNanToNum.test_integerc C sT t d}t|dk tt|tj t ddddd}t|dk tt|tj d S )NrJ gñ?r< r© rª r« )r r r ro r Zfloat_r¢ r r r Ú
|
test_float s zTestNanToNum.test_floatc C sT t d}t|dk tt|tj t ddddd}t|dk tt|tj d S )NrI r< r© rª r« )r r r ro r Úcomplex_r¢ r r r Útest_complex_good s z TestNanToNum.test_complex_goodc C sZ t jddd d}|t d¡d 7 }W 5 Q R X t|}tt |¡ tt|t j d S )Nr r rI rH r © r r r r r r r ro r¶ ©r$ Úvr£ r r r Útest_complex_bad¥ s z TestNanToNum.test_complex_badc C sZ t jddd d}|t d¡d 7 }W 5 Q R X t|}tt |¡ tt|t j d S )Nr r rI y ð¿ ð?r r¸ r¹ r r r Útest_complex_bad2® s z TestNanToNum.test_complex_bad2c C s t jddd tt d¡d t jdd}W 5 Q R X tt |ddg ¡ t|d d k t|d
|
dg t jdg tt|t j d S ) Nr r r¡ r iç )r¬ r r r r¨ r )
|
r r r r Úinfr r r ro rB r¢ r r r Ú$test_do_not_rewrite_previous_keyword¼ s $z1TestNanToNum.test_do_not_rewrite_previous_keywordN) r' r( r) r¤ r³ r rµ r· r» r¼ r¾ r r r r r§ ^ s % r§ c @ s e Zd Zdd ZdS )ÚTestRealIfClosec C sf t j d¡}t|d }tt| t|| t|d }tt| t|d dd}tt| d S )Nr< y Vç¯Ò<y H¯¼ò×z>gíµ ÷Æ°>)Ztol)r r> r? r r r r r )r$ ry Úbr r r r% É s
|
zTestRealIfClose.test_basicNr& r r r r r¿ Ç s r¿ c @ s e Zd Zdd ZdS )ÚTestArrayConversionc C s\ t t dddg¡}t|jtj tt |jtj ¡ t
|
t t t dddg¡t d¡d d S )Nr r r rJ r ) r r r r Ú __class__rB r Z
|
issubdtyper Zfloatingr Ú TypeError)r$ ry r r r Ú test_asfarrayÖ s ÿz!TestArrayConversion.test_asfarrayN)r' r( r) rÄ r r r r rÁ Ô s rÁ )%Únumpyr Z numpy.testingr r r r Znumpy.lib.type_checkr r r r r
|
r r r r r r r r r* r: r; rM rP rZ r\ r{ r| r r r r¦ r§ r¿ rÁ r r r r Ú<module> s( 42
|
5
|
|
i
|
