U
|
O ±dÍ ã @ sò d dl Zd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
|
Z
|
d dl Z d dl Z d dl Z d dlZd dlmZ d dlmZ d dlZd dlZd dlmZ d dlZd dlZd dlm m Z d dl m Z d dl m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0 d dl1m2Z2m3Z3 d dl4m5Z5 d d l6m7Z7 d d
|
l8m9Z9m8Z8 e:d dfd d Z;G dd dZ<G dd dZ=G dd dZ>G dd dZ?G dd dZ@G dd dZAG dd dZBG d d d ZCG d d d ZDG d d! d!ZEG d"d# d#ZFG d$d% d%ZGG d&d' d'ZHG d(d) d)ZIG d*d+ d+ZJG d,d- d-ZKG d.d/ d/ZLG d0d1 d1ZMG d2d3 d3ZNG d4d5 d5ZOG d6d7 d7ZPG d8d9 d9ZQG d:d; d;ZRG d<d= d=ZSG d>d? d?ZTG d@dA dAZUG dBdC dCZVG dDdE dEZWG dFdG dGZXG dHdI dIZYG dJdK dKZZG dLdM dMZ[G dNdO dOZ\G dPdQ dQZ]G dRdS dSZ^G dTdU dUZ_G dVdW dWZ`dXdY ZadZd[ Zbd\d] ZcG d^d_ d_ZdG d`da daZeG dbdc dcZfG ddde deZgG dfdg dgegZhG dhdi diegZidjdk Zjdldm ZkG dndo doZlG dpdq dqZmG drds dsZnd dtdudvdwdxZoejpjqdye:egdzd{gd|G d}d~ d~ZrG dd dZsG dd dZtG dd dZud d
lvmwZw G dd dZxG dd dZyG dd dezZ{G dd dZ|G dd dZ}dd Z~dd Zdd Zdd Zdd Zdd Zdd ZG dd dZ
G d d¡ d¡ZG d¢d£ d£ZG d¤d¥ d¥Ze)sPG d¦d§ d§ZG d¨d© d©ZG dªd« d«ZG d¬d dZG d®d¯ d¯ZG d°d± d±ZG d²d³ d³Zd d´l m)Z) G dµd¶ d¶ZG d·d¸ d¸ZG d¹dº dºZG d»d¼ d¼Zd½d¾ Zd¿dÀ Zejp qdÁdÂge jejfe jejfg¡ejp qdÃdtejej ejg¡ejp qdÄdtejej ejg¡ejp qdÅej ¡ ¡dÆdÇ ZdÈdÉ ZdÊdË Z G dÌdÍ dÍZ¡dÎdÏ Z¢G dÐdÑ dÑZ£ejp qdÒdÓdÔdÕdÖd×dØdÙdÚdÛdÜdÝdÞdßdàdág¡dâdã Z¤dädå Z¥dædç Z¦dS )èé N)Úcontextmanager)Úpickle)ÚDecimal)Úrational)Úassert_Ú assert_raisesÚ assert_warnsÚ assert_equalÚassert_almost_equalÚassert_array_equalÚassert_raises_regexÚassert_array_almost_equalÚassert_allcloseÚIS_PYPYÚ IS_PYSTONÚ HAS_REFCOUNTÚassert_array_lessÚ runstringÚtemppathÚsuppress_warningsÚ break_cycles)Úrequires_memoryÚ _no_tracing)ÚCommaDecimalPointLocale)Ú repack_fields)Ú timedeltaÚdatetimeÚCc C sþ t |¡}|t t¡kr8|dk r(tdt j| ||dS |dkrF|j}t| dsV| f} t t j
|
| ¡|j }t |d| d t j ¡}|jd d }|| }|dkrª|| }|d| dkrÂ||7 }|||| d
dd
}| d¡ t j| |||d
|
}|S ) zÞ
|
Allocate a new ndarray with aligned memory.
|
|
The ndarray is guaranteed *not* aligned to twice the requested alignment.
|
Eg, if align=4, guarantees it is not aligned to 8. If align=None uses
|
dtype.alignment.Nz$object array alignment not supported©ÚdtypeÚorderÚ__len__é é Údatar éÿÿÿÿ©r )Únpr ÚobjectÚ
|
ValueErrorÚzerosÚ alignmentÚhasattrÚ functoolsÚreduceÚoperatorÚmulÚitemsizeÚemptyÚuint8Ú__array_interface__ÚfillÚndarray) Úshaper r ÚalignÚsizeÚbufÚptrÚoffsetr$ © r= úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_multiarray.pyÚ_aligned_zeros) s*
|
|
|
r? c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej j
|
e d d d d Z dd Z dd Zej dddgdddg¡dd Zdd Zd d Zd d Zd!S )"Ú TestFlagsc C s t d¡| _d S ©Né
|
)r' ÚarangeÚa©Úselfr= r= r> Ú setup_methodN s zTestFlags.setup_methodc C sN t }d| jj_tttd| tttd| d| jj_d| jd< d| jd< d S )NFz self.a[0] = 3zself.a[0:1].itemset(3)Té r )ÚlocalsrD ÚflagsÚ writeabler r) r )rF Zmydictr= r= r> Útest_writeableQ s
|
|
|
zTestFlags.test_writeablec C s° t d¡}G dd dt j}| |¡}|d }d|j_d|j_d|j_t d¡}G dd d}t |¡}|d }d|j_d|j_d|j_d|j_tt d|j_W 5 Q R X d S ) NrB c @ s e Zd ZdS )z3TestFlags.test_writeable_any_base.<locals>.subclassN©Ú__name__Ú
|
__module__Ú __qualname__r= r= r= r> Úsubclass_ s rQ .FTc @ s e Zd Zdd ZdS )z8TestFlags.test_writeable_any_base.<locals>.frominterfacec S s || _ |j| _d S ©N)Úarrr4 ©rF rS r= r= r> Ú__init__l s zATestFlags.test_writeable_any_base.<locals>.frominterface.__init__N)rN rO rP rU r= r= r= r> Ú frominterfacek s rV ) r' rC r6 ÚviewrJ rK Úasarrayr r) )rF rS rQ Zview1Zview2rV r= r= r> Útest_writeable_any_baseZ s"
|
|
|
|
|
z!TestFlags.test_writeable_any_basec C sZ d}t |d¡}tt|jdd t ddg¡}t jj ||¡}|d }tt|jdd d S )Nód ÚBT©Úwrite©ÚvalsÚu1©Zres3ÚS4r_ ) r' Ú
|
frombufferr r) Úsetflagsr ÚcoreÚrecordsÚ
|
fromstring©rF r$ r_ ÚtypesÚvaluesr= r= r> Ú test_writeable_from_readonly{ s z&TestFlags.test_writeable_from_readonlyc C s¾ t d}t |d¡}t|jj |jdd t|jjdk |jdd t|jj t ddg¡}tjj
|
||¡}|d }t|jj |jdd t|jjdk |jdd t|jj d S ) NrZ r[ Fr\ Tr^ ra r_ ) Ú bytearrayr' rc r rJ rK rd r re rf rg rh r= r= r> Útest_writeable_from_buffer s z$TestFlags.test_writeable_from_bufferzPyPy always copies©Úreasonc C sT dd l }t d¡}t|jD ]2}| | ||¡¡}t|jj tt
|
|j t q d S )Nr éè ) r r' rC ÚrangeÚHIGHEST_PROTOCOLÚloadsÚdumpsr rJ rK Ú
|
isinstanceÚbaseÚbytes)rF r rD Úvr_ r= r= r> Útest_writeable_pickle s
|
z TestFlags.test_writeable_picklec C s. ddl m} |d}|jjr t|jjs,t|d }d|j_|jjrHtd|j_|jjs\td|j_|d}|jjrxt|jjrt||fD ]}|d }d|j_d|j_|jjr´ttt d|j_W 5 Q R X t ¡ . t
|
dt ¡ tt d|j_W 5 Q R X W 5 Q R X t t d|j_W 5 Q R X qd S )Nr )Úget_c_wrapping_arrayT.FÚerror) Ú numpy.core._multiarray_testsrz rJ ÚowndataÚAssertionErrorrK r r) ÚwarningsÚcatch_warningsÚ simplefilterÚDeprecationWarningr )rF rz Z arr_writeablerW Z arr_readonlyrS r= r= r> Útest_writeable_from_c_data£ s4
|
|
|
|
z$TestFlags.test_writeable_from_c_datac C sX t d¡}d|j_tjdd0}t d¡ d|d< d|d< tt|dk W 5 Q R X d S )NrB T©ÚrecordÚalwaysr# r" ) r' rC rJ Ú_warn_on_writer r Úfilterwarningsr Úlen)rF rD Úwr= r= r> Útest_warnonwriteË s
|
|
zTestFlags.test_warnonwriteÚflagÚ
|
flag_valuerK )rK TT)r TF)rK FFc sn t d¡ t j|| G fddd}t j|k s<t jd d |k sRtt | ¡jj |ksjtd S )NrB c s e Zd Z jZdS )z5TestFlags.test_readonly_flag_protocols.<locals>.MyArrN)rN rO rP Ú__array_struct__r= ©rD r= r> ÚMyArrß s r r$ r# )
|
r' rC ÚsetattrrJ Ú
|
memoryviewÚreadonlyr~ r4 rX rK )rF r r rK r r= r r> Ú test_readonly_flag_protocolsÕ s
|
z&TestFlags.test_readonly_flag_protocolsc C sÊ t | jjjd t | jjd d t | jjjd t | jjjd t | jjjd t | jjjd t | jjjd t | jjj d t | jjj
|
d t | jjj d t | jjd d t | jjd d d S )NTr FÚXZWRITEBACKIFCOPY) r rD rJ ÚcarrayÚfarrayÚbehavedÚfncÚforcr} rK ÚalignedÚwritebackifcopyrE r= r= r> Útest_otherflagsæ s zTestFlags.test_otherflagsc C sD t jdt d¡d}t|jj t jdt d¡d}t|jj d S )Né z|S4©r rH ©r' r* r r rJ r ©rF rD r= r= r> Útest_string_alignô s zTestFlags.test_string_alignc C s( t jdt ddg¡d}t|jj d S )Nr ©rD Úi4©Úbr¤ r r r¡ r= r= r> Útest_void_alignû s zTestFlags.test_void_alignN)rN rO rP rG rL rY rk rm ÚpytestÚmarkÚskipifr ry r r Ú parametrizer r r¢ r§ r= r= r= r> r@ M s& !
|
(
|
üÿ
|
r@ c @ s e Zd Zdd ZdS )ÚTestHashc C sd t jt jdft jt jdft jt jdft jt jdffD ](\}}}t d|D ]}t
|
t |d| t d| d||f d t
|
t |d|d t d|d d ||d f d t
|
t |d| d t d| d d
|
||f d t |d d}t
|
t |d|d t d|d d ||d f d t
|
t |d| d t d| d d
|
||f d qJq4d S ) Né é é é@ r# r" z
|
%r: -2**%d©Úerr_msgz %r: 2**%dz %r: 2**%d - 1) r' Úint8r3 Úint16Úuint16Úint32Úuint32Úint64Úuint64rq r ÚhashÚmax)rF ÚstÚutÚsÚir= r= r> Útest_int s, ý
|
ÿ"ÿ"
|
ÿ"ÿ"
|
ÿzTestHash.test_intN)rN rO rP rÀ 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 )ÚTestAttributesc C s@ t d¡| _t d¡ dd¡| _t jdt jd ddd¡| _d S ) NrB é r rH é< r r" é )r' rC ÚoneÚreshapeÚtwoÚfloat64ÚthreerE r= r= r> rG s zTestAttributes.setup_methodc C s t | jjd t | jjd t | jjd d| j_t | jjd d| j_t | jj| jjf | jj}t | jjd| |f | jj}t | jjd| d| |f t | jjd t | jjd t | jjd
|
| jj}t | jjd t | jj d | t | jj| jj
|
j t | jj t d ¡ d S ) N©rB ©r rH )r" rH rÄ )rB é r" rH é rÄ r# r" rÌ r )r rÅ r7 rÇ rÉ Ústridesr1 Úndimr9 Únbytesr rv r' rC ©rF Únumr= r= r> Útest_attributes s& z TestAttributes.test_attributesc C s t | jjt tj¡ t | jjt tj¡ t | jjjd t | jjjd t| jjj d dk t | jjj d d t | jjj d d d S )NÚlÚdr ú<>r# r¿ Úf)
|
r rÅ r r' Úint_rÉ Zfloat_Úcharr ÚstrrE r= r= r> Útest_dtypeattr1 s z TestAttributes.test_dtypeattrc C s t d¡}tt|t d S ©Nr )r' rØ r ru Úint)rF Z numpy_intr= r= r> Útest_int_subclassing: s
|
z#TestAttributes.test_int_subclassingc sª | j fdd}t|dddt ddddg¡ tt|ddd tt|ddd tt|d dd t|d dd
|
t dgd ¡ tt|d d t dd g¡ |d
|
d
|
d d S )Nc s t j| t| j | j dS )N)Úbufferr r< rÎ )r' r6 rÝ r1 )r9 r< rÎ ©Úxr= r> Ú
|
make_arrayE s
|
þz3TestAttributes.test_stridesattr.<locals>.make_arrayr r% rÌ r" r# éþÿÿÿr r ©r" rÌ rH éýÿÿÿrB )rÅ r r' Úarrayr r) )rF râ r= rà r> Útest_stridesattrB s z TestAttributes.test_stridesattrc sD | j fdd}t|dddt ddddg¡ t|dddt ddd d
|
dd d g¡ tt|ddd tt|ddd tt|d dd tjj t
|
d¡dd¡ dd }tt| d j j f tjjj tj
|
dtj dd ddd tt| d d d
d d d d
}d|_ d|d d d
_ t d¡}d|_ tt||d d S )Nc
|
s^ z t j| gt | j d}W n* tk
|
rH } z t|W 5 d }~X Y nX | j |_}|S )N)r rß r< )r' r6 rÝ r1 Ú ExceptionÚ RuntimeErrorrÎ )r9 r< rÎ ÚrÚerà r= r> râ V s ÿ
|
z7TestAttributes.test_set_stridesattr.<locals>.make_arrayr r% rÌ r" r# é rH rÄ r é rã ©rB rB ©r r c S s
|
|| _ d S rR ©rÎ )rS rÎ r= r= r> Ú set_stridesh s z8TestAttributes.test_set_stridesattr.<locals>.set_stridesrB r rÊ ©r% )r7 rÎ r r= )rÅ r r' ræ r r) ré ÚlibZ stride_tricksZ
|
as_stridedrC r1 r³ rÎ Ú TypeError)rF râ rñ rD Zarr_0dr= rà r> Útest_set_stridesattrS s* & ÿ
|
z#TestAttributes.test_set_stridesattrc C sB dD ]8}t d|¡}t d|¡}| d¡ d|d< t|| qd S )Nz?bhilqpBHILQPfdgFDGO©rÌ r" r# r# .)r' r2 r5 r )rF Útrá Úyr= r= r> Ú test_fillz s
|
zTestAttributes.test_fillc C sD t jdt jd}t jdt jd}d}||d< | |¡ t|| d S )Nrö r l ÿÿÿÿ .)r' r2 r¹ r5 r )rF rá rø Úvaluer= r= r> Útest_fill_max_uint64 s
|
z#TestAttributes.test_fill_max_uint64c C s~ t jddgdd}| |d ¡ t|d d |d d t jdd d
|
gd}| d ¡ t|d d d g t|d ddg d S )N)r ç )r# ç ð?úi4,f8r r Úf1r# r" ©rD Úf8r¥ )ç @rã rD r r¦ rã )r' ræ r5 r r* r ©rF rá r= r= r> Útest_fill_struct_array s
|
z%TestAttributes.test_fill_struct_arrayc C s> t d¡}|jdd tjtdd | d¡ W 5 Q R X d S )Né Fr\ z .*read-only©Úmatchr )r' r* rd r¨ Úraisesr) r5 r¡ r= r= r> Útest_fill_readonly s
|
z!TestAttributes.test_fill_readonlyN) rN rO rP rG rÓ rÛ rÞ rç rõ rù rû r r r= r= r= r> rÁ s ' rÁ c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d e j e je je je jg¡dd Ze j
|
d e j e je je je jg¡dd ZdS )ÚTestArrayConstructionc C sÎ t d¡}t ||g¡}t|t d¡ t d¡}t d¡}t ||g¡}t|| d|d< t ||d g¡}t|| t d¡}t ||gg¡}t|t d¡ t d¡}t ||g||gg¡}t|t d¡ t d¡}t ||g¡}t|t d¡ t d ¡}t j||d g|d gtd
|
}tt|d t|d ||d g t|d |d t jd td
|
}d |d< d |dd d
f< t ddd gd d dgg¡}t|| t dd gdd gd dgg¡}t||j d S )NrÄ )r" rÄ r" r# )r# r" rÄ )r" r" rÄ )rÄ rÄ )r" rÄ rÄ )rÄ r r rä F©r r" T)r' Úonesræ r r( r ÚboolÚT)rF rÕ rê Útgtr= r= r> Ú
|
test_array s>
|
|
|
|
|
|
|
|
|
|
z TestArrayConstruction.test_arrayc C s t ttj d S rR )r rô r' ræ rE r= r= r> Útest_array_emptyÆ s z&TestArrayConstruction.test_array_emptyc C s t t d¡¡jdkstd S )NrÌ ©rÌ )r' r ræ r7 r~ rE r= r= r> Útest_0d_array_shapeÉ s z)TestArrayConstruction.test_0d_array_shapec C sz t dddg¡}t j|dd}d|d< t|dddg t j|ddd}d|d< t|dddg d |d< t|ddd g d S )
|
Nr# r" rÌ F©ÚcopyÚF©r r r rì ©r' ræ r ©rF rÕ rë r= r= r> Útest_array_copy_falseÌ s z+TestArrayConstruction.test_array_copy_falsec C sÄ t dddgdddgg¡}t j|dd}d|d< d|d< t|dddgdddgg t|dddgdddgg t j|dd d
|
}d |d< d |d< t|ddd gdddgg t|dd dgdddgg d S ) Nr# r" rÌ Tr ©r r# iùÿÿÿr r r rH rì r r r= r= r> Útest_array_copy_true× s z*TestArrayConstruction.test_array_copy_truec C s¢ t d¡d d d
}tt |¡jj tt |¡jj tt |¡jj tt |¡jj t d¡d d d
d d d
f }tt |¡jj tt |¡jj d S )NrB r" rî )r' r r ÚascontiguousarrayrJ Ú c_contiguousÚ f_contiguousÚasfortranarray©rF rÕ r= r= r> Útest_array_contä s z%TestArrayConstruction.test_array_contÚfuncc
|
C sn t t¡ |ddd W 5 Q R X t t¡ | W 5 Q R X t t¡ |ddddddd d
|
W 5 Q R X d S ) NrÌ z bad dtyper r# r" r rH rÄ rì r )r¨ r rô ©rF r# r= r= r> Útest_bad_arguments_errorî s z.TestArrayConstruction.test_bad_arguments_errorc C s$ |t jkr|dd n
|
|dd d S )NrÌ )r( r ©r' ræ r$ r= r= r> Útest_array_as_keywordü s
|
z+TestArrayConstruction.test_array_as_keywordN)rN rO rP r r r r r r" r¨ r© r« r' ræ rX Z
|
asanyarrayr r r% r' r= r= r= r> r
|
s0 '
|
üÿ
|
üÿr
|
c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestAssignmentc C sÐ t d¡ dd¡}t d¡|d< t|dddgdddgg t d¡ dd¡|d< t|dddgdddgg t d¡d d d
ddd¡|d< t|dd dgdddgg d
|
d }tt||t d ¡ ddd¡ d S ) NrÄ r" rÌ .r r# r% rH r c S s || d< d S )N.r= ©rD r¦ r= r= r> Úassign! s z;TestAssignment.test_assignment_broadcasting.<locals>.assigné )r' rC rÆ r r r) ©rF rD r* r= r= r> Ú test_assignment_broadcasting s
|
"z+TestAssignment.test_assignment_broadcastingc sH G dd d}t d¡ fdd}tttf|| tt|dg d S )Nc @ s e Zd ZdS )z0TestAssignment.test_assignment_errors.<locals>.CNrM r= r= r= r> r ( s r r# c s | d< d S rÜ r= ©rx r r= r> r* , s z5TestAssignment.test_assignment_errors.<locals>.assign)r' r* r ÚAttributeErrorrô r) )rF r r* r= r r> Útest_assignment_errors& s
|
|
z%TestAssignment.test_assignment_errorsc sr ddl m t fdd}t dg¡}t d¡}|d ||d< W 5 Q R X t|d d t t d¡g¡ d S ) Nr ©Úset_string_functionc
|
3 s2 fdddd z
|
dV W 5 ddd X dS )z% replace ndarray.__str__ temporarily c s S rR r= rà ©r¾ r= r> Ú<lambda>9 ó zLTestAssignment.test_unicode_assignment.<locals>.inject_str.<locals>.<lambda>F)ÚreprNr= r3 r1 r3 r> Ú
|
inject_str6 s
|
z:TestAssignment.test_unicode_assignment.<locals>.inject_strÚtestÚdoneÚbadu åäö)Znumpy.core.numericr2 r r' ræ r )rF r7 Za1dZa0dr= r1 r> Útest_unicode_assignment2 s
|
|
z&TestAssignment.test_unicode_assignmentc C sv t dg¡}t dg¡}G dd d}tttj|dg tttj|dg tttj|d| tttj|d| d S )Nr9 s donec @ s e Zd Zdd Zdd ZdS )z?TestAssignment.test_stringlike_empty_list.<locals>.bad_sequencec S s d S rR r= rE r= r= r> Ú __getitem__N r5 zKTestAssignment.test_stringlike_empty_list.<locals>.bad_sequence.__getitem__c S s t d S rR ©ré rE r= r= r> r! O r5 zGTestAssignment.test_stringlike_empty_list.<locals>.bad_sequence.__len__N©rN rO rP r< r! r= r= r= r> Ú bad_sequenceM s r? r )r' ræ r r) r/ Úsetitem)rF Úur¦ r? r= r= r> Útest_stringlike_empty_listH s z)TestAssignment.test_stringlike_empty_listc C sà t jt jfD ]Î}t t d¡d¡ |¡}t t d¡d¡ |¡}t |g¡}t|d | ||d< t|d | ||d< t|d | |d |d< t|d | |d |d< t|d | t t |¡g¡}t|d | q d S )Nr r# r% ©r ..)r' Ú
|
longdoubleZ longcomplexZ nextafterÚastyperæ r )rF r ZtinybZtinyaZtiny1drS r= r= r> Útest_longdouble_assignmentW s z)TestAssignment.test_longdouble_assignmentc C s8 t jddd}t jdgdd|d d
< t|d d d S )Nr# ÚS20r z1.12345678901234567890r r s 1.1234567890123457)r' r* ræ r r¡ r= r= r> Útest_cast_to_stringw s z"TestAssignment.test_cast_to_stringN) rN rO rP r- r0 r; rB rF rH r= r= r= r> r( s r( c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestDtypedescrc C s< t d¡}t|t t j¡ t d¡}t|t t j¡ d S )Nr¤ r )r' r r r¶ rÈ )rF Úd1Úd2r= r= r> Útest_construction s
|
|
z TestDtypedescr.test_constructionc C s8 t t d¡t d¡k t t dg¡t dg¡k d S )Nú<i4ú>i4©rD rM ©rD rN )r r' r rE r= r= r> Útest_byteorders s z TestDtypedescr.test_byteordersc C sB ddg}t d|f¡}tt|d t d|¡}tt|d d S )N)rD ú<i2©r¦ rR r¤ z+(numpy.int32, [('a', '<i2'), ('b', '<i2')])r zFarray([0, 0, 0, 0], dtype=(numpy.int32, [('a', '<i2'), ('b', '<i2')])))r' r r rÚ r* r6 )rF ÚfieldsZdt_intZarr_intr= r= r> Útest_structured_non_void s ÿz'TestDtypedescr.test_structured_non_voidN)rN rO rP rL rQ rU r= r= r= r> rI s rI c @ sl e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd ZdS )Ú TestZeroRankc C s t d¡t dt¡f| _d S )Nr rá )r' ræ r( rÕ rE r= r= r> rG s zTestZeroRank.setup_methodc C sN | j \}}t|d d t|d d t|d j|k t|d j|k d S )N.r rá )rÕ r r rv ©rF rD r¦ r= r= r> Útest_ellipsis_subscript s
|
|
z$TestZeroRank.test_ellipsis_subscriptc C sV | j \}}t|d d t|d d tt|d |jjk tt|d tk d S )Nr= r rá )rÕ r r Útyper rÚ rW r= r= r> Útest_empty_subscript¡ s
|
|
z!TestZeroRank.test_empty_subscriptc C sN | j \}}ttdd | ttdd | ttdd | ttdd | d S )Nc S s | d S rÜ r= rà r= r= r> r4 ª r5 z5TestZeroRank.test_invalid_subscript.<locals>.<lambda>c S s | d S rÜ r= rà r= r= r> r4 « r5 c S s | t g t¡ S rR ©r' ræ rÝ rà r= r= r> r4 ¬ r5 c S s | t g t¡ S rR r[ rà r= r= r> r4 r5 ©rÕ r Ú
|
IndexErrorrW r= r= r> Útest_invalid_subscript¨ s
|
|
z#TestZeroRank.test_invalid_subscriptc C s6 | j \}}d|d< t|d d|d< t| ¡ d d S )Né* .Ú ©rÕ r ÚitemrW r= r= r> Ú"test_ellipsis_subscript_assignment¯ s
|
|
|
z/TestZeroRank.test_ellipsis_subscript_assignmentc C s6 | j \}}d|d< t|d d|d< t| ¡ d d S )Nr_ r= r` ra rW r= r= r> Ú test_empty_subscript_assignment¶ s
|
|
|
z,TestZeroRank.test_empty_subscript_assignmentc C sF | j \}}dd }tt||dd tt||dd tt||dd d S )Nc S s || |< d S rR r= ©rá r¿ rx r= r= r> r* À s z>TestZeroRank.test_invalid_subscript_assignment.<locals>.assignr r_ r` r= )rÕ r r] r) )rF rD r¦ r* r= r= r> Ú!test_invalid_subscript_assignment½ s
|
|
z.TestZeroRank.test_invalid_subscript_assignmentc C sÈ | j \}}t|tj jd t|dtjf jd t|tjdf jd t|dtjf jd t|tjdtjf jd t|dtjtjf jd t|tjtjdf jd t|tjfd jd d S ©N©r# .©r# r# rB )
|
r# r# r# r# r# r# r# r# r# r# ©rÕ r r' Únewaxisr7 rW r= r= r> Ú test_newaxisÇ s
|
zTestZeroRank.test_newaxisc C s@ | j \}}dd }tt||tjdf tt||tjfd d S )Nc S s | | d S rR r= ©rá r¿ r= r= r> Ú subscriptÕ s z4TestZeroRank.test_invalid_newaxis.<locals>.subscriptr é2 ©rÕ r r] r' rk )rF rD r¦ rn r= r= r> Útest_invalid_newaxisÒ s
|
z!TestZeroRank.test_invalid_newaxisc C s t d¡}d|d< t|d d t jd|d}d|d< t|d d t t¡ t jddd W 5 Q R X t t¡ t jddd W 5 Q R X d S )Nr= rH )rß rÄ ©r" rð )r' r6 r r¨ r r) ©rF rá rø r= r= r> Útest_constructorÛ s
|
z TestZeroRank.test_constructorc C s" t d¡}ttt j|dg| d S )Nr" r# )r' ræ r r) Úaddr r= r= r> Ú test_outputé s
|
zTestZeroRank.test_outputc C s t d¡}|j}|j}t|t d¡ tt|t jk t|jj d t|jj
|
d t|t d¡ tt|t jk t|jj d t|jj
|
d d S )Nù ð?r Tr# ) r' ræ ÚrealÚimagr r rY r6 rJ Ú
|
contiguousr )rF rá ZxrÚxir= r= r> Útest_real_imagí s
|
zTestZeroRank.test_real_imagN)rN rO rP rG rX rZ r^ rc rd rf rl rq rt rv r| r= r= r= r> rV s
|
rV c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestScalarIndexingc C s t ddg¡d | _d S ©Nr r# )r' ræ rÕ rE r= r= r> rG ÿ s z TestScalarIndexing.setup_methodc C s( | j }t|d d t|d jd d S )N.r r= ©rÕ r r7 r¡ r= r= r> rX s z*TestScalarIndexing.test_ellipsis_subscriptc C s( | j }t|d d t|d jd d S )Nr= r r r¡ r= r= r> rZ s z'TestScalarIndexing.test_empty_subscriptc C s* | j }ttdd | ttdd | d S )Nc S s | d S rÜ r= rà r= r= r> r4 r5 z;TestScalarIndexing.test_invalid_subscript.<locals>.<lambda>c S s | t g t¡ S rR r[ rà r= r= r> r4 r5 r\ r¡ r= r= r> r^ s z)TestScalarIndexing.test_invalid_subscriptc C s" | j }dd }tt||dd d S )Nc S s || |< d S rR r= re r= r= r> r* s zDTestScalarIndexing.test_invalid_subscript_assignment.<locals>.assignr r_ )rÕ r rô r, r= r= r> rf s z4TestScalarIndexing.test_invalid_subscript_assignmentc C sÄ | j }t|tj jd t|dtjf jd t|tjdf jd t|dtjf jd t|tjdtjf jd t|dtjtjf jd t|tjtjdf jd t|tjfd jd d S rg rj r¡ r= r= r> rl s z TestScalarIndexing.test_newaxisc C s< | j }dd }tt||tjdf tt||tjfd d S )Nc S s | | d S rR r= rm r= r= r> rn ' s z:TestScalarIndexing.test_invalid_newaxis.<locals>.subscriptr ro rp )rF rD rn r= r= r> rq $ s z'TestScalarIndexing.test_invalid_newaxisc C sx t d¡}|dd
|d d
< t|ddddg t d¡}|d d
|dd
< t|ddddg t d¡}|d d d
|d d
< t|ddddg t d¡ dd¡}|d d
d d d
f |d d d
d d
f< t|dddgdddgg t d¡ dd¡}|d d
d d d
f |d d d
d d d
f< t|dddgdddgg t d¡}|dd
|d d
< t|dddddg t d¡}|d d
|dd
< t|dddddg t d¡}|dd
|dd d
< t|dddddg t d¡}|dd d
|dd
< t|dddddg t d¡}|d dd
|dd d
< t|dddddg t d¡}|dd d
|d dd
< t|dddddg d S ) Nr r# r% r" rÌ r rÄ rH )r' rC r rÆ r¡ r= r= r> Útest_overlapping_assignment- sB
|
|
|
(*
|
|
|
|
|
|
z.TestScalarIndexing.test_overlapping_assignmentN) rN rO rP rG rX rZ r^ rf rl rq r r= r= r= r> r} þ s r} c @ sV e Zd ZdZdd Zdd Zdd Zej dej
|
e d d
|
ej
|
d d d
|
g¡d d Z dd Z ejje ej¡jdkdddd Zdd Zejjdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Z d&d' Z d(d) Z d*d+ Z d,d- Z d.d/ Z!d0d1 Z"d2d3 Z#d4d5 Z$d6d7 Z%d8d9 Z&ej d:e'd;d<d=g¡ej d>ej(ej)d?d@ g¡dAdB Z*dCS )DÚ TestCreationz'
|
Test the np.array constructor
|
c C s" G dd d}t ttj| d S )Nc @ s e Zd ZdddZdS )z+TestCreation.test_from_attribute.<locals>.xNc S s d S rR r= )rF r r= r= r> Ú __array__d s z5TestCreation.test_from_attribute.<locals>.x.__array__)N©rN rO rP r r= r= r= r> rá c s rá )r r) r' ræ r r= r= r> Útest_from_attributeb s z TestCreation.test_from_attributec C s\ t jd t jd }ddg}t jddgtd}|D ]$}d| }tt j||d||d q2d S )NÚ
|
AllIntegerÚFloatZ123é{ r zString conversion for %sr± )r' Ú typecodesræ rÝ r )rF ri ZnstrÚresultrY Úmsgr= r= r> Útest_from_stringi s z TestCreation.test_from_stringc C sä t jg dd}|jdks tt jddgdd}|jdks<tt t¡ t jddgdd W 5 Q R X t t¡ t jddgdd W 5 Q R X t jddgdd d¡}|jdks²tt t¡ t jddgdd d¡ W 5 Q R X d S )NÚVr ÚV8s 1234ÚV4s 12345ÚO)r' ræ r r~ r¨ r rô rE rT r= r= r> Ú test_voidq s zTestCreation.test_voidÚidxrS ©Úidr= Úscalarc C s t jt jddd| t jddd| gdd}t|t jddgdd t t¡4 t jt jddd| t jddd| gdd W 5 Q R X d S ) Nr# úi,ir r" r ri ©r" r" zi,i,i)r' ræ r r¨ r rô )rF r rS r= r= r> Ú test_structured_void_promotion
s "þ "þz+TestCreation.test_structured_void_promotionc C sp t d¡jdkrd}nt d¡jdkr,d}nd S ttt j|t jd ttt j|t jd ttt j|t jd d S )NÚintpéÿÿÿ)éµ r ì ÿÿÿÿ )ì 4s j r r ) r' Úiinfor» r r) r2 r³ r* r )rF r7 r= r= r> Útest_too_big_error s z TestCreation.test_too_big_errorr z%malloc may not fail on 32 bit systemsrn c C s8 t tjjj tjt tj¡jtj d W 5 Q R X d S ©Nr )
|
r r' re Ú _exceptionsZ_ArrayMemoryErrorr2 r r r» r3 rE r= r= r> Útest_malloc_fails¢ s z TestCreation.test_malloc_failsc C sî t jd t jd }|D ]Ð}t jd|d}tt |¡d t| ¡ d t| ¡ t jddd}tt |¡d t| ¡ d t| ¡ t jddd}tt |¡d t| ¡ d t| ¡ t jdd d}tt |¡d qd S )
|
Nr
ÚAllFloat©é r r r" z(2,4)i4Z4i4z(2,4)i4, (2,4)i4)r' r r* r Ú count_nonzeroÚsumr Úany©rF ri ÚdtrÕ r= r= r> Ú
|
test_zeros© s zTestCreation.test_zerosc C s@ t jd t jd }|D ]"}t jd|d}t| ¡ ~qd S )Nr
r¢ )i àr )r' r r* r r§ r¨ r= r= r> Útest_zeros_big¿ s
|
zTestCreation.test_zeros_bigc C s2 t jdtd}t|dgd tt |¡d d S )Nr£ r r r¤ )r' r* r( r r r¥ r! r= r= r> Útest_zeros_objË s zTestCreation.test_zeros_objc C s( t jddtdfgd}t|d d d S )NrB Úkr" r r )r' r* r( r r! r= r= r> Útest_zeros_obj_objÑ s z TestCreation.test_zeros_obj_objc C s® t jd D ]@}|dkrq
|
t jd|d}tt |¡| tt |¡j|j q
|
t jddd}tt |¡| tt |¡j|j t jddd}tt |¡| tt |¡j|j t jddd}tt |¡| tt |¡j|j t jddd}tt |¡| tt |¡j|j t jdd d}tt |¡| tt |¡j|j t jdd
|
d}tt |¡| tt |¡j|j t jdd d}tt |¡| tt |¡j|j d S ) NÚAllr ©rÌ rÌ r ZS5ÚU5rM rN z<M8[s]z>M8[s]zf4,f4)r' r r* r Ú
|
zeros_liker r )rF ÚcrÕ r= r= r> Útest_zeros_like_like_zerosÕ s6 z'TestCreation.test_zeros_like_like_zerosc C s, t dddD ]}tj|dd}t| q d S )NrH éd ÚUr )rq r' r2 rÚ )rF r¿ rÕ r= r= r> Útest_empty_unicode÷ s z TestCreation.test_empty_unicodec C sÖ t t ddd g¡jt t t ddd dg¡jt t t dd dg¡jt t t dd gd ¡jt t t ddgddgg¡jt t t ddgddgg¡jt t t dddgdddgdddgg¡jt d S )Nr r" éP rÌ r# rw rý )r r' ræ r r( ÚcomplexrE r= r= r> Ú test_sequence_non_homogeneousý s z*TestCreation.test_sequence_non_homogeneousc C sb G dd d}G dd d}t | g¡}t|jdk t|jt t¡k ttt j| g dS )a) Should not segfault.
|
|
Class Fail breaks the sequence protocol for new style classes, i.e.,
|
those derived from object. Class Map is a mapping type indicated by
|
raising a ValueError. At some point we may raise a warning instead
|
of an error in the Fail case.
|
|
c @ s e Zd Zdd Zdd ZdS )z5TestCreation.test_non_sequence_sequence.<locals>.Failc S s dS ©Nr# r= rE r= r= r> r! s z=TestCreation.test_non_sequence_sequence.<locals>.Fail.__len__c S s
|
t d S rR ©r) ©rF Úindexr= r= r> r< s zATestCreation.test_non_sequence_sequence.<locals>.Fail.__getitem__N©rN rO rP r! r< r= r= r= r> ÚFail s rÀ c @ s e Zd Zdd Zdd ZdS )z4TestCreation.test_non_sequence_sequence.<locals>.Mapc S s dS r» r= rE r= r= r> r! s z<TestCreation.test_non_sequence_sequence.<locals>.Map.__len__c S s
|
t d S rR )ÚKeyErrorr½ r= r= r> r< s z@TestCreation.test_non_sequence_sequence.<locals>.Map.__getitem__Nr¿ r= r= r= r> ÚMap s r rh N)r' ræ r r7 r r( r r) )rF rÀ r rD r= r= r> Útest_non_sequence_sequence s z'TestCreation.test_non_sequence_sequencec C s: G dd d}t | | | g¡}t|jt t¡ d S )Nc @ s e Zd Zdd Zdd ZdS )z4TestCreation.test_no_len_object_type.<locals>.Point2c S s d S rR r= rE r= r= r> rU % s z=TestCreation.test_no_len_object_type.<locals>.Point2.__init__c S s |dkr |S t d S )Nr ©r] ©rF Úindr= r= r> r< ( s z@TestCreation.test_no_len_object_type.<locals>.Point2.__getitem__N)rN rO rP rU r< r= r= r= r> ÚPoint2$ s rÇ )r' ræ r r r( )rF rÇ rÕ r= r= r> Útest_no_len_object_type" s z$TestCreation.test_no_len_object_typec C s, G dd d}t | ¡}tt|d d S )Nc @ s e Zd Zdd Zdd ZdS )z/TestCreation.test_false_len_sequence.<locals>.Cc S s t d S rR rÄ )rF r¿ r= r= r> r< 3 s z;TestCreation.test_false_len_sequence.<locals>.C.__getitem__c S s dS )Nr_ r= rE r= r= r> r! 5 s z7TestCreation.test_false_len_sequence.<locals>.C.__len__Nr> r= r= r= r> r 2 s r r )r' ræ r r ©rF r rD r= r= r> Útest_false_len_sequence0 s z$TestCreation.test_false_len_sequencec C sT G dd d}t d¡}tt | |d d
< W 5 Q R X t | ¡t| k d S )Nc @ s$ e Zd Zdd Zdd Zdd ZdS )z/TestCreation.test_false_len_iterable.<locals>.Cc S s t d S rR ©rè r r= r= r> r< > s z;TestCreation.test_false_len_iterable.<locals>.C.__getitem__c S s t dS )Nr= )ÚiterrE r= r= r> Ú__iter__@ s z8TestCreation.test_false_len_iterable.<locals>.C.__iter__c S s dS )Nr" r= rE r= r= r> r! B s z7TestCreation.test_false_len_iterable.<locals>.C.__len__N)rN rO rP r< rÍ r! r= r= r= r> r = s r r" )r' r2 r r) ræ ÚlistrÉ r= r= r> Útest_false_len_iterable; s
|
|
|
z$TestCreation.test_false_len_iterablec C s4 G dd d}|dddg}t tt |¡d d S )Nc @ s$ e Zd Zdd Zdd Zdd ZdS )z0TestCreation.test_failed_len_sequence.<locals>.Ac S s
|
|| _ d S rR )Ú_data)rF r$ r= r= r> rU N s z9TestCreation.test_failed_len_sequence.<locals>.A.__init__c S s t | | j| S rR )rY rÐ )rF rb r= r= r> r< P s z<TestCreation.test_failed_len_sequence.<locals>.A.__getitem__c S s
|
t | jS rR )r rÐ rE r= r= r> r! R s z8TestCreation.test_failed_len_sequence.<locals>.A.__len__N)rN rO rP rU r< r! r= r= r= r> ÚAM s rÑ r# r" rÌ )r r r' ræ )rF rÑ rÕ r= r= r> Útest_failed_len_sequenceK s z%TestCreation.test_failed_len_sequencec C sn t d¡}t t j¡j}dD ]L}t |¡}|j}t j|d|| f|d tt t j|d|| d f|d q d S )Nrµ )r rG r¦ ©r )rß rÎ r7 r r# )
|
r' r* r r r» r r1 r6 r r) )rF r: Z max_bytesr r1 r= r= r> Útest_array_too_bigY s
|
|
ÿ ÿz TestCreation.test_array_too_bigc C s2 t jtdd t |¡}W 5 Q R X tj|tdS )Nz.*detected shape wasr r )r¨ r r) r' ræ r( )rF ÚseqrD r= r= r> Ú_ragged_creationh s z TestCreation._ragged_creationc C s | dgddg¡}t|jd t|jt | ddgdg¡}t|jd t|jt | dddgg¡}t|jd t|jt d S ©Nr# r" rÌ r )rÖ r r7 r r( r¡ r= r= r> Útest_ragged_ndim_objecto s z$TestCreation.test_ragged_ndim_objectc C s | ddgdgdgg¡}t|jd t|jt | dgddgdgg¡}t|jd t|jt | dgdgddgg¡}|jdkst|jtkstd S r× )rÖ r r7 r r( r~ r¡ r= r= r> Útest_ragged_shape_object} s z%TestCreation.test_ragged_shape_objectc C sº t d d g¡}t dddg¡ |d< |d< t |¡jdks>tt |g¡jdksTtt d d g¡}t dddg¡|d< t ddddg¡|d< t |¡jdks tt |g¡jdks¶td S )Nr# r" rÌ r rr ©r# r" r )r' ræ r7 r~ )rF ÚouterZ outer_raggedr= r= r> Útest_array_of_ragged_array s z'TestCreation.test_array_of_ragged_arrayc C s@ t tdggg¡}t dtdg¡}t dgtdgg¡}d S r» )r' ræ r r¡ r= r= r> Útest_deep_nonragged_object s z'TestCreation.test_deep_nonragged_objectr zO,OzO,(3)Oz(2,3)OÚfunctionc C s t t j| |d¡S r )r' Ú
|
empty_liker2 ©r7 r r= r= r> r4 ¡ r5 zTestCreation.<lambda>c C s@ |d|d}t d ¡ ¡ }||jt| }| ¡ |ks<td S )NrÌ r )r' ræ ÚtobytesrÐ r r~ )rF rÞ r rS Úexpectedr= r= r> Ú test_object_initialized_to_None s z,TestCreation.test_object_initialized_to_NoneN)+rN rO rP Ú__doc__r r r r¨ r© r« ÚparamÚEllipsisr r rª r' r r r1 r¡ rª Úslowr« r¬ r® r´ r· rº rà rÈ rÊ rÏ rÒ rÔ rÖ rØ rÙ rÜ rÝ r( r6 r2 rã r= r= r= r> r ^ sN ÿ
|
ÿ
|
|
" þr 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 d Zej d!ed"d#gd$d%gd&d'ed"d#gd$d%gd(d)gd*g¡ej d+d,d-g¡d.d/ Zd0d1 Zd2S )3ÚTestStructuredc C s t jddgd}t d¡ dddd¡|d< t|jd |d d d
|
dd¡ |jd d }t|d j |d j t|jd j |j ¡ d j d S ) N)rÌ rH )rD ©r¤ r r rà rÌ rH r" rD r# r r r& )
|
r' r* rC rÆ r r Ú transposer r r7 rW r= r= r> Útest_subarray_field_access s z)TestStructured.test_subarray_field_accessc C sÄ t jjdddgdddgddggfdddgdddgddggfgddtfd gd
|
}| ¡ }t||kd d g t||kd d g d |d _t||kd d g t||kd d g tdD ]F}|d j|d _d|d j|< t||kd d g t||kd d g q°tdD ]T}tdD ]D}| ¡ }d|d j ||f< t||kd d g t||kd d g q q t j
|
dgdggdgd
|
}t j
|
dddgdgd
|
}t||kd d d gd d d gg t||kd d d gd d d gg t j
|
dgdggdgd
|
}t j
|
dddgdgd
|
}t||kd d d gd d d gg t||kd d d gd d d gg t j
|
ddgfgddgfggdgd
|
}t j
|
ddgfddgfddgfgdgd
|
}t||kd d d gd d d gg t||kd d d gd d d gg t j
|
ddgfgddgfggdgdd}t j
|
ddgfddgfddgfgdgd
|
}t||kd d d gd d d gg t||kd d d gd d d gg t j dddgd
|
}t j dddgd
|
}t t¡ ||k W 5 Q R X t j dddgd
|
}t j dddgd
|
}t t¡ ||k W 5 Q R X d S )Nr# r" rÌ rD r r¦ r )rD )Úf4rÌ )r³ ré r TFr³ rH rB rÓ rh r )rD r rh )rD r rr r r )rD )rì rÚ ©r¦ Úi1)rD )rì rr )rD )rì ©r" r# )r' ÚrecZ fromrecordsr( r r r¦ rq rD r³ ræ r* r¨ r rô )rF rD r¦ r¿ Újrá rø r= r= r> Útest_subarray_comparison» sZ 6 þ
|
$( &( z'TestStructured.test_subarray_comparisonc C sd t ddg¡}t|| t ddg¡}t|| t ddg¡}t|| t ddg¡}t|| d S )Nr )rD ú<f8ri )rD ró rh rï )r# r r# )r' r* r r¡ r= r= r> Ú&test_empty_structured_array_comparisonõ s
|
|
|
z5TestStructured.test_empty_structured_array_comparisonc C sô t jddgddgd}t jddgddgd}t||kd d
|
g t||kd
|
d g t jddgd dgd}t jddgdd gd}t||kd d
|
g t||kd
|
d g t jddgd dgd}t jddgddgd}t||kd d
|
g t||kd
|
d g d S )N©rH r_ ©rB r# ©rD ú>i8©r¦ ró r )rH é+ ©rD z<i8©r¦ ú>f8FT)rD rý ©r¦ rø )rD z10>f8)r¦ z5<f8)rD z10<i8)r¦ z5>i8©r' ræ r rW r= r= r> Ú*test_structured_comparisons_with_promotion s z9TestStructured.test_structured_comparisons_with_promotionc C s¸ t jddgd}t d¡}t t¡ ||k W 5 Q R X t jddgd}t |j|j¡s\tt t¡ ||k W 5 Q R X t jddd}t jddd}t t¡ ||k W 5 Q R X d S )NrÌ )rD rî r ))ÚtitlerD rî ZV7r )r' r* r¨ r rô Úcan_castr r~ rs r= r= r> Ú test_void_comparison_failures s
|
z,TestStructured.test_void_comparison_failuresc C s t jdgdgd}tt j|jdgdd | dg¡}t|| ¡ ¡ t|d d |d d t jd d
|
gdd gd}t jd d
|
gdd gd}tt j|j|jd d t||kddg tt j|j|jd d |j|jd d}t||kddg dd g}tt j|j|dd |j|dd}t|t jd d
|
g|dkddg ddg}tt j|j|dd |j|dd}t|t jd d
|
g|dkddg ddg}tt j|j|dd t t
|
|j|dd dd g}tt j|j|d d t t
|
|j|d d ddg}tt j|j|dd t t
|
|j|dd tt j|j|jdd t t
|
|j|jdd dD ]F}dg}tt j|j||d dd dg}tt j|j||d qRd S )Nrh rO r rP Úunsafe©ÚcastingrD r rõ rö rù rü ÚequivTrû Úsafe)r¦ z>f4Ú same_kindr÷ )r¦ ú<f4)rD ú>i2rS Úno)r r r r )r³ r¤ ) r' ræ r r r rE r ÚbyteswapÚ newbyteorderr rô )rF rD r¦ r³ r÷ r r= r= r> Ú test_casting* sP ÿÿ
|
zTestStructured.test_castingc C s@ t jg dddgd}|ddg t ddd g¡}|d
|
d g d S ) N©rD r× ©r¦ r× )r³ r r rD r¦ rÌ )rÑ r¿ )r[ z|Or[ rÑ )r' ræ r* )rF rD Zdat2r= r= r> Ú test_objviewd s zTestStructured.test_objviewc C sv t dg¡}t dd|fg¡}t d|¡}t jddd|d d < t jd|d|d d< t|d d t jddd d S )
|
N)Úelemr¤ rH )Úfieldr¤ rB Ústructr# rB r¤ r r r )r' r r* r r )rF Z struct_dtr© rá r= r= r> Ú test_setfieldm s z TestStructured.test_setfieldc sd t jddgd}t d¡|d d< t|d d t d¡ t jddgd fdd }tt| d S )
|
Nr# )rá r r rÌ r rá )rá r rH c s t d¡ d d< d S )NrÌ r rá ©r' rC r= ©r³ r= r> Ú
|
testassign s z7TestStructured.test_setfield_object.<locals>.testassign)r' r* rC r r r) )rF r¦ r r= r r> Útest_setfield_objectv s z#TestStructured.test_setfield_objectc C sX t dtfdg¡}t jd|d}t|d ddddg t|d jd dd d
|
d g|d< t|d ddddg t|d ddddg d |d |d dk< t|d ddddg t|d ddddg d |d< t|d ddddg t|d ddddg t jd|d jd}t|jd t|d | tt jdddgdd jd tt jdddjd tt jdddd jd |d d¡}t|jd t|ddgddgg t|d d
j|j t t
|
t |t t j d| t ¡ }t ||¡ | d¡ t |¡}t|jd t|| tdd2}t ||¡ t |¡}t|jd t|| W 5 Q R X d S )NÚI)ÚSZS0r r r r5 r rD r¦ r³ rÕ ZhellorÑ )rD úS0,S0)r¦ r` rÌ r zS0,u1Úf0r ©ræ z.npy)Úsuffix)r' r rÝ r* r r1 r6 r2 rÆ r Úevalr6 Údictræ ÚioÚBytesIOÚsaveÚseekÚloadr )rF r© rá rø Zxxr¦ ÚyyÚtmpr= r= r> Útest_zero_width_string sR
|
ÿÿÿ
|
|
|
|
z%TestStructured.test_zero_width_stringc C s( t jddd}|d }t|j|k d S )NrÌ zi4,f4r r )r' r* r rv rW r= r= r> Útest_base_attr¾ s z TestStructured.test_base_attrc s¢ dd t ddg¡}t d|¡ t jdgddgd}t jdgddgd}t jdgddgd}t jd gdgdt jdgd
|
d gd} d gd gdgd¡}t jddg|d}t || t || t || tt fdd t || d|d d
< t t jddg|d t jdgdddgd}|d dg |dd g< t|d ¡ d t jdgddgd}|d d g |d d g< t|d ¡ d d S ) Nc S s | ¡ }||d< ||dd
< |S r~ r )rS rx r³ r= r= r> r Ä s z2TestStructured.test_assignment.<locals>.testassign)ÚfooÚi8)Úbarr- r" rä r )Úbazr- rr )r, r )r. r r. r- r ©ÚnamesÚformatsÚoffsetsc s
|
S rR r= r= ©rS r Zv4r= r> r4 × r5 z0TestStructured.test_assignment.<locals>.<lambda>r )r# r ©r# r" rÌ )r/ rì r, r )r" r# rÌ rÚ r£ r¥ r¦ rD rï ) r' r r ræ rW r r rô rb ) rF r© Úv1Úv2Zv3Zv5r ZansrD r= r4 r> Útest_assignmentà s2 ÿz TestStructured.test_assignmentc C s4 t t" t d¡ dd¡}| d¡ W 5 Q R X d S )Né rH rÌ )r r) r' rC rÆ ZitemsetrT r= r= r> Útest_scalar_assignmentè s
|
z%TestStructured.test_scalar_assignmentc C sJ t jddd}t|ddg d |d ddg t|d |d d d S )Nr# z (2)3S,(2)3Urà r rÿ r r= )r' r2 r r r= r= r> Ú test_structuredscalar_indexingí s "z-TestStructured.test_structuredscalar_indexingc sZ t jddddgd tt fdd tt fdd tt fd d d
|
d g d S ) Nr )r) r¿ ©r³ r¿ ©rÕ r¿ r c s ddg S )NrD r³ r= r= r r= r> r4 õ r5 z7TestStructured.test_multiindex_titles.<locals>.<lambda>c s ddg S )NrD r= r= r r= r> r4 ö r5 c s ddg S )Nr¦ r= r= r r= r> r4 ÷ r5 r¦ r³ )r' r* r rÁ r) rE r= r r> Útest_multiindex_titlesó s
|
z%TestStructured.test_multiindex_titlesc C s4 d}d}t jdg||gd}t jdg||gd}ttt j|| ttt j|j|j ttt j|j|j tt |j|ddg j¡t |j
|
d¡ tt |j|j¡d t| |j¡j|j tt d d
|
d g¡d tt d d d g¡d tt d d d
|
g¡d tt j d d d
|
gddd ||d d
< t dg¡}t dg¡}tt ||¡t d g¡ tt ||¡t d g¡ ttt j|t dg¡ tt dd¡t d d g¡ t d|fg¡}t d|fg¡}tt ||¡t dt d g¡fg¡ t dgdgdgd¡} t j d | d}
|
tt |
|
|
|
g¡jt d g¡ d S ) Nrû rþ rÚ r rD r¦ ÚNTzf8,i8)r r )rÿ r- )rÿ r )r r- Fr r )r` r¤ )r` r- )r` ÚV3rþ zi8,f4r` r rá r¤ r r0 rÌ )rá r¤ )r' ræ r rô Ú concatenateÚ result_typer Ú promote_typesr r r r rE r ) rF rÑ r[ ÚabÚbaÚdt1Údt2Zdt1nestZdt2nestr© rD r= r= r> Ú)test_structured_cast_promotion_fieldorderú sJ ÿÿÿ ÿ ÿz8TestStructured.test_structured_cast_promotion_fieldorderÚ
|
dtype_dictrD r¦ r¤ r× rµ )r1 r2 r1 r r+ r0 r8 TFc C s| t j||d}| dd ¡ | dd ¡ t j||d}t ||¡}|j|jksPt|j|jks`tt ||¡}||ksxtd S )N©r8 r1 r3 )r' r ÚpoprC r1 r~ rT )rF rI r8 r râ Úresr= r= r> Ú test_structured_promotion_packs) s z.TestStructured.test_structured_promotion_packsc C s t jdgdd}|d }|jjr$tt |¡j|ks8tt j||jdjd ksRtt t
|
|¡¡}|jjsntt |¡jd kstd S )Nr# r r r ) r' ræ rJ r} r~ rX rv r r rs rt )rF rS r r= r= r> Ú test_structured_asarray_is_view? s z.TestStructured.test_structured_asarray_is_viewN)rN rO rP rë rò rô r r r r r r r* r+ r8 r: r; r> rH r¨ r© r« r" rM rN r= r= r= r> rè ¬ s2 : : 8%/ÿþ rè c @ s e Zd Zdd Zdd Zdd Zdd Zejj d d
|
Z
|
d d Z d d Z dd Z ejjdddd Zejjdddd ZdS )ÚTestBoolc C sr t d¡}t d¡}t||k t d¡}t d¡}t||k tt dg¡d |k tt d¡d |k d S )Nr Fr# Tr= )r' Úbool_r ræ )rF Úa0Úb0Za1Úb1r= r= r> Útest_test_interningP s
|
|
|
|
z TestBool.test_test_interningc C sÐ t jdtd}t| ¡ |j t|d d d
¡ |d d d
j t|d d d
¡ |d d d
j t jdtd}t| ¡ |j t|d d d
¡ |d d d
j t|d d d
¡ |d d d
j d S )Née r r" rã sÈ ÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿ)r' r r r r¦ r9 rc r! r= r= r> Útest_sumZ s $$$zTestBool.test_sumc sº dd t |D }t d| D ] fdd|D }tj|td}t |¡}tt |¡| | tj ¡}|d9 }tt |¡| |d9 }tt |¡| d||d k< tt |¡| q d S )
|
Nc S s g | ] }d | qS rr r= ©Ú.0r¿ r= r= r> Ú
|
<listcomp>f s z0TestBool.check_count_nonzero.<locals>.<listcomp>r" c s g | ]} |@ d kqS rÓ r= ©rX rá ©r¿ r= r> rY h s r rÌ r éÿ r )
|
rq r' ræ r Úbuiltinsr¦ r r¥ rW r3 )rF ÚpowerÚlengthZpowersrÔ rD r³ Úavr= r[ r> Úcheck_count_nonzeroe s
|
z TestBool.check_count_nonzeroc C s | dd¡ d S )Nr+ é ©ra rE r= r= r> Útest_count_nonzerot s zTestBool.test_count_nonzeroc C s | dd¡ d S )Nrb rc rE r= r= r> Útest_count_nonzero_ally s z TestBool.test_count_nonzero_allc C s t dD ]}tjdtd|d d
}d|d |
< tt |¡t | ¡ ¡ tj dtd|d d
}d|d |
< tt |¡t | ¡ ¡ qd S )Nrì )é r r# TF)
|
rq r' r* r r r¥ r] r¦ Útolistr )rF ÚorD r= r= r> Ú test_count_nonzero_unaligned s z%TestBool.test_count_nonzero_unalignedc C s t dD ]r}t d||f¡}tt|d tt|d d t| t¡d tt| t¡tj t|d t¡tj k qt ddD ]|}dD ]r}t |||f¡}tt|d tt|d d t| t¡d tt| t¡tj t|d t¡tj
|
k qqd S ) NrÌ r5 Fr= r# r )ó aó 0ó T) rq r' ræ r r rE r ru r6 ÚFalse_ÚTrue_)rF r Únrx Úvalr= r= r> Ú_test_cast_from_flexible s z!TestBool._test_cast_from_flexiblec C s | tj¡ d S rR )rq r' ÚvoidrE r= r= r> Útest_cast_from_void s z TestBool.test_cast_from_voidz See gh-9847rn c C s | tj¡ d S rR )rq r' Úunicode_rE r= r= r> Útest_cast_from_unicode s z TestBool.test_cast_from_unicodec C s | tj¡ d S rR )rq r' Úbytes_rE r= r= r> Útest_cast_from_bytes¤ s z TestBool.test_cast_from_bytesN)rN rO rP rT rV ra rd r¨ r© rç re ri rq rs Zxfailru rw r= r= r= r> rO O s
|
|
|
|
rO c @ s e Zd ZeefddZdd Zdd Zdd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zejjejdk ddd d Zd S ) ÚTestZeroSizeFlexiblec C sJ t |¡}|t jkr&t j| |dfdS t d|dfg¡}t j| |dd S )Nr r rá )r' r rr r* rà r= r= r> Ú_zerosª s
|
|
|
zTestZeroSizeFlexible._zerosc C sN | dt¡}t|jd | dtj¡}t|jd | dt¡}t|jd d S )NrB r )ry rw r r1 r' rr rÚ ©rF Úzsr= r= r> Ú test_create´ s z TestZeroSizeFlexible.test_createc K sf t tjtfD ]T}| d|¡}t||}tt|}|D ]*}|f d|i| ||fd|i| q4q d S )NrB Úkind)rw r' rr rÚ ry Úgetattr) rF ÚnameÚkindsÚkwargsr© r{ Z sort_methodZ sort_funcr} r= r= r> Ú_test_sort_partition¼ s
|
|
z)TestZeroSizeFlexible._test_sort_partitionc C s | j ddd d S )NÚsortÚqhs©r ©r rE r= r= r> Ú test_sortÆ s z TestZeroSizeFlexible.test_sortc C s | j ddd d S )NÚargsortr r
r rE r= r= r> Ú test_argsortÉ s z!TestZeroSizeFlexible.test_argsortc C s | j ddgdd d S )NÚ partitionÚ introselectr" ©r Úkthr rE r= r= r> Útest_partitionÌ s z#TestZeroSizeFlexible.test_partitionc C s | j ddgdd d S )NÚ argpartitionr r" r r rE r= r= r> Útest_argpartitionÏ s z&TestZeroSizeFlexible.test_argpartitionc C s6 t tjtfD ]$}| d|¡}| d¡ | d¡ q d S )NrB r9 rî )rw r' rr rÚ ry Úresize©rF r© r{ r= r= r> Ú test_resizeÒ s
|
z TestZeroSizeFlexible.test_resizec C sP t tjtfD ]>}| d|¡}t| |¡jt |¡ t| |df¡jd q d S )NrB r# rÓ ) rw r' rr rÚ ry r rW r r7 r r= r= r> Ú test_viewÙ s z TestZeroSizeFlexible.test_viewc C s$ | dt¡}t|t | ¡ ¡ d S rA )ry rÝ r r rs rt rz r= r= r> Ú
|
test_dumpsã s z TestZeroSizeFlexible.test_dumpsc C s^ t dtjd D ]H}ttjtfD ]6}| d|¡}tj||d}t |¡}t
|
|j |j q qd S )Nr" r# rB ©Úprotocol) rq r rr rw r' rr rÚ ry rt rs r r )rF Úprotor© r{ ÚpZzs2r= r= r> Ú test_pickleç s
|
z TestZeroSizeFlexible.test_picklec C s8 t g ¡ dd¡}t |¡}t |¡}|jdks4tdS )z]Checking if an empty array pickled and un-pickled will not cause a
|
segmentation faulti?B r N)r' ræ rÆ r rt rs r9 r~ )rF rS Zpk_dmpZpk_loadr= r= r> Útest_pickle_emptyð s
|
|
z&TestZeroSizeFlexible.test_pickle_emptyrH úrequires pickle protocol 5rn c C sR t d¡}g }tj||jdd}tj||d}d|d< |d dksNt|d d S )NrB rH )Úbuffer_callbackr ©Úbuffersr% r )r' rC r rt Úappendrs r~ )rF ræ r Ú bytes_stringZarray_from_bufferr= r= r> Ú test_pickle_with_buffercallbackù s
|
|
ÿz4TestZeroSizeFlexible.test_pickle_with_buffercallbackN)rN rO rP Ú staticmethodrÚ ry r| r r r r r r r r r r r¨ r© rª r rr r¢ r= r= r= r> rx © s"
|
|
ÿrx c @ s, e Zd ZdddgZdd Zdd Zdd Zd
|
d Zd d Zdd Z dd Z
|
dd Z dd Z dd Z dd Zej dejejejejejejejejg¡dd Zej dej ej ej ej ejejejejg¡d d Z ej dejejejg¡ej d d d!g¡d"d# Z!d$d% Z"ej dej#ej$g¡d&d' Z%d(d) Z&d*d+ Z'ej dd,d-g¡d.d/ Z(d0d1 Z)d2d3 Z*d4d5 Z+d6d7 Z,d8d9 Z-d:d; Z.d<d= Z/ej dd>ej d?g¡d@dA Z0dBdC Z1dDdE Z2dFdG Z3dHdI Z4ej dJej5dKdLej6gejdMej5dKdLej6gejdMe 5dKdLej6g¡g¡dNdO Z7dPdQ Z8dRdS Z9dTdU Z:dVdW Z;dXdY Z<dZd[ Z=d\d] Z>d^d_ Z?d`da Z@ej dejAdb ¡dcdd ZBej dejAdb ¡dedf ZCdgdh ZDdidj ZEej dkejAdl ¡dmdn ZFej dkejAdl ¡dodp ZGdqdr ZHdsdt ZIdudv ZJdwdx ZKdydz ZLd{d| ZMej dkejAdl ¡d}d~ ZNdd ZOej dejPejQf¡dd ZRej dejPejQf¡ej dd¡d
d ZSdd ZTdd ZUdd ZVdd ZWdd ZXdd ZYdd ZZdd Z[dd Z\dd Z]dd Z^dd Z_dd Z`d¡d¢ Zad£d¤ Zbd¥d¦ Zcd§d¨ Zdd©dª Zed«d¬ Zfdd® Zgd¯S )°Ú TestMethodsZ quicksortZheapsortZstablec C s t dddgdddgdddgg¡}t dddgdddgdddgg¡}t dgdgdgg¡}dD ]p}t|j||dt j||d d
df d d
f |d tt j|||d||d d
df d d
f j|d q^t|j|dd tt j||dd t|jddd tt j|ddd d S ©NTF©r N©ÚaxisÚwherer ©r¨ ©r© )r' ræ r Úall)rF rD Úwh_fullZwh_lowerÚ_axr= r= r> Útest_all_where s. þ þþ$ÿ"ÿzTestMethods.test_all_wherec C s t dddgdddgdddgg¡}t dddgdddgdddgg¡}t dgdgdgg¡}dD ]p}t|j||dt j||d d
df d d
f |d tt j|||d||d d
df d d
f j|d q^t|j|dd tt j||dd t|jddd tt j|ddd d S r¥ )r' ræ r r§ )rF rD r Z wh_middler® r= r= r> Útest_any_where" s. þ þþ$ÿ"ÿzTestMethods.test_any_wherec C sØ dddddgg}t d¡ dd¡}|jdd gdd
|
}t|| d d gddgg}|jdd dd dgd d
|
}t|| d gdgg}t d¡ dd¡}|jdd gd d
|
}t|| t d¡ dd¡}| dd g¡}t|d d S © NrH rÄ rì r rí rB r" r r# rª rÌ ©r' rC rÆ Úcompressr ©rF r rS Úoutr= r= r> Ú test_compress6 s
|
|
|
zTestMethods.test_compressc
|
C s dt jdtd }dt jdtd }dt jdtd }dt jdtd }t dddg¡}| ||f¡}t|dddg | ||f¡}t|dddgdddgg | ||f¡}t|dddgdddgg t d¡}tt|jt d¡|g t d¡}t jt d¡dd d
|
g|d } || ks t t|d d t
|
d ¡}t jdddg|d d
|d d
|d d
g|dd
dd}t|t dddg¡ d S )Nr" r r rÌ rä r r# rB r rÍ ©rµ r= rH r Úwrap©rµ Úmode) r' r rÝ ræ Úchooser r rô rr r~ rC )
|
rF rá rø Úx2Úy2rÆ rÑ Zonedrµ Úretr= r= r> Ú test_chooseI s(
|
|
|
>zTestMethods.test_choosec C sô dddddddg}ddddgddd d
|
gddddgg}t jt jt jt jt jt jt jt jfD ]}t ||¡}t ||¡}|d krt
|
t |j t
|
t |j dd qXt |j d d d t|j d d t ddddg|¡ t|j dd t dddg|¡ qXd S )Nr# r" rB r rÄ rH r rÌ rì rí )Ú1r¦ rª r i g ro é$ éT é´ r% é ib iX )r' r´ rµ r¶ r· Úfloat32rÈ Ú complex64Ú
|
complex128ræ r ÚArithmeticErrorÚprodr r )rF rE Zba2ÚctyperD Za2r= r= r> Ú test_prodg s( " ÿ ÿ ÿzTestMethods.test_prodc C s^ t ddddddg¡}| d¡}| ddddddg¡}t|ddddddddddg
|
| d¡}t|ddddddddddddg |jddgdd }t|dddgdddgdddgg |jdddgdd }t|ddddddgddddddgg |jddd }t|dddgdddgdddgdddgg |jddd }t|ddddddgddddddgg d S )
|
Nr# r" rÌ r rH rÄ rä r rª )r' ræ rÆ Úrepeatr )rF ÚmÚm_rectrÑ r= r= r> Ú test_repeaty sL
|
ÿ
|
ÿ þÿ ýÿzTestMethods.test_repeatc C s t dddgdddgddd gd
|
d d gg¡}ddddddgddd d
|
d d gg}t| dd¡| ddddgddddgd d
|
d d gg}t| dd¡| dd
|
ddgddd d gdddd gg}t|jd dd| dddd
|
gdddd gddd d gg}t|jjd dd| d S )Nr# r" rÌ r rH rÄ rì r rí rB r r+ ©rÌ r r r& r )r' ræ r rÆ r )rF rS r r= r= r> Ú test_reshape s * """zTestMethods.test_reshapec C s dd }|t ddg¡ddg |t d¡d |t ddg¡d d
|
gd |t d d g¡ddgd |t dg¡dg |t dg¡dgd d S )Nc W sD t | j| | t | ¡}| j|d|i}t || ||ks@td S ©Nrµ )r Úroundr' r² r~ )rS râ Z
|
round_argsrµ rL r= r= r> Ú check_round¨ s
|
|
|
z+TestMethods.test_round.<locals>.check_roundg333333ó?ç ø?r# r" gffffff(@g /@rB r r% gÍÌÌÌÌL(@g
ëQ¸ /@y @ ø?y @ @y )@ /@y $@ 4@r& )rF rÔ r= r= r> Ú
|
test_round§ s zTestMethods.test_roundc C sx t dgdgdggg¡}t| ¡ dddg t|jdddgdgdgg tt|jdd t|jdddddgg d S )Nr# r" rÌ rÓ rª rh rr )r' ræ r Zsqueezer r) r¡ r= r= r> Ú test_squeeze¸ s
|
zTestMethods.test_squeezec sj t ddgddgg¡ t ¡ ddgddgg tt fdd tt fdd tt fdd d S ) Nr# r" rÌ r c s
|
d¡S rÜ ©rê r= r r= r> r4 Â r5 z,TestMethods.test_transpose.<locals>.<lambda>c s dd¡S rÜ rØ r= r r= r> r4 Ã r5 c s ddd¡S )Nr r# r" rØ r= r r= r> r4 Ä r5 )r' ræ r rê r r) rE r= r r> Útest_transpose¿ s
|
zTestMethods.test_transposec C sº d}t t jddg¡}t |¡}t||d d d
| d}t jdt jd}| jt jt jt jddddddg 7 _| jt jddt jt jddddg 7 _t |¡}t||d d d
| d S )Nz Test real sort order with nansr# r r% z!Test complex sort order with nansrí r ) r' ræ Únanr r r* rÇ rx ry )rF r rD r¦ r= r= r> r Æ s
|
&&
|
zTestMethods.test_sortr c C sx t jd|d}|d d d
¡ }| jD ]L}d| }| ¡ }|j|d t||| | ¡ }|j|d t||| q&d S )NrU r r% úscalar sort, kind=%s©r} ©r' rC r Ú
|
sort_kindsr r ©rF r rD r¦ r} r r³ r= r= r> Útest_sort_unsignedß s
|
z TestMethods.test_sort_unsignedc C sz t jdd|d}|d d d
¡ }| jD ]L}d| }| ¡ }|j|d t||| | ¡ }|j|d t||| q(d S )NéÎÿÿÿé3 r r% rÛ rÜ rÝ rß r= r= r> Útest_sort_signedî s
|
z TestMethods.test_sort_signedÚpartrx ry c C sÒ t jt jt jt jt jt ji| }t jdd|d}|d d d
¡ }|d |¡}|d |¡}t
|
||d t
|
||d | j D ]P}d||f } | ¡ }
|
|
|
j |d t |
|
|| | ¡ }
|
|
|
j |d t |
|
|| q|d S ) Nrá râ r r% ù ð? ð?r# z#complex sort, %s part == 1, kind=%srÜ )r' ÚsingleÚcsingleÚdoubleÚcdoublerD Ú clongdoublerC r rE r rÞ r r ) rF rä r ZcdtyperD r¦ ÚaiÚbir} r r³ r= r= r> Útest_sort_complexý s. ýü
|
z TestMethods.test_sort_complexc C s\ dD ]R}t jd D ]B}t jdddg|| d}| ¡ }| ¡ d |¡}t||| qqd S )NrÖ ÚComplexù ð? @ù @ @ù @ ð?r z$byte-swapped complex sort, dtype={0})r' r ræ r r Úformatr )rF Ú
|
endiannessr© rS r³ r r= r= r> Ú test_sort_complex_byte_swapping s
|
z+TestMethods.test_sort_complex_byte_swappingc C s t jdd tdD |d}|d d d
¡ }| jD ]L}d| }| ¡ }|j|d t||| | ¡ }|j|d t||| q4d S )Nc S s g | ]}d t | qS )Úaaaaaaaa©ÚchrrW r= r= r> rY # s z0TestMethods.test_sort_string.<locals>.<listcomp>rU r r% úkind=%srÜ )r' ræ rq r rÞ r r rß r= r= r> Útest_sort_string s
|
z TestMethods.test_sort_stringc C s t jdtd}ttd|d d
< |d d d
}dD ]L}d| }| ¡ }|j|d t||| | ¡ }|j|d t||| q4d S )N©rU r rU r% ©ÚqÚhrÍ rø rÜ )r' r2 r( rÎ rq r r r )rF rD r¦ r} r r³ r= r= r> Útest_sort_object. s z TestMethods.test_sort_objectc C s t dtfdtfg¡}t jdd tdD |d}|d d d
}dD ]L}d | }| ¡ }|j|d
|
t||| | ¡ }|j|d
|
t||| qDd S ) Nr× r¿ c S s g | ] }||fqS r= r= rW r= r= r> rY ? s z4TestMethods.test_sort_structured.<locals>.<listcomp>rU r r% rû rø rÜ ) r' r ÚfloatrÝ ræ rq r r r )rF r© rD r¦ r} r r³ r= r= r> Útest_sort_structured< s z TestMethods.test_sort_structuredú datetime64[D]útimedelta64[D]c C st t jdd|d}|d d d
}dD ]L}d| }| ¡ }|j|d t||| | ¡ }|j|d t||| q"d S )Nr rU r r% rû rø rÜ )r' rC r r r rß r= r= r> Útest_sort_timeJ s zTestMethods.test_sort_timec C s¢ t ddgddgg¡}t ddgddgg¡}t ddgddgg¡}| ¡ }|jdd t||d | ¡ }|jdd t||d | ¡ }| ¡ t||d d S ) NrÌ r" r# r rª ztest sort with axis=0ztest sort with axis=1ztest sort with default axis)r' ræ r r r )rF rD r¦ r³ rÕ r= r= r> Útest_sort_axisX s zTestMethods.test_sort_axisc C sh t g ¡}d|_t|j |jD ]$}d |¡}tt j||d|| q d}tt j|d d| ¡ | d S )N©rÌ r" r# r z#test empty array sort with axis={0}rª z$test empty array sort with axis=None) r' ræ r7 rq rÏ rò r r Úravel)rF rD r¨ r r= r= r> Útest_sort_size_0h s
|
|
z TestMethods.test_sort_size_0c C s\ G dd d}t j| gd td}| jD ],}d| }| ¡ }|j|d t||| q*d S )Nc @ s e Zd Zdd ZdS )z0TestMethods.test_sort_bad_ordering.<locals>.Boomc S s dS )NTr= ©rF Úotherr= r= r> Ú__lt__v s z7TestMethods.test_sort_bad_ordering.<locals>.Boom.__lt__N)rN rO rP r
|
r= r= r= r> ÚBoomu s r rµ r rø rÜ )r' ræ r( rÞ r r r )rF r rD r} r r³ r= r= r> Útest_sort_bad_orderingr s
|
z"TestMethods.test_sort_bad_orderingc C s t dD ]4}tjjddtjd}| d¡}|d d d
¡ qt dg¡}t dD ]4}tjjddtjd}| |¡}|d d d
¡ qRd S )Nr é i ©r9 r r r% )rp r¤ rh )rq r' ÚrandomÚrandintr3 rW r r )rF r¿ ÚrandrS r© r= r= r> Útest_void_sort s
|
|
zTestMethods.test_void_sortc s t jdt ¡ dgtd}| jD ]}tt|j|d q G dd d t fddt d D ¡
|
d
|
¡}t j |¡ | jD ]}tt|j|d qvd S ) Nr r# r rÜ c @ s, e Zd Zdd Ze Z Z Z Z ZZ dS )z,TestMethods.test_sort_raises.<locals>.Raiserc _ s t dd S )NzSOMETHING ERRORED©rô ©Úargsr r= r= r> Úraises_anything s z<TestMethods.test_sort_raises.<locals>.Raiser.raises_anythingN)
|
rN rO rP r Ú__eq__Ú__ne__r
|
Ú__gt__Ú__ge__Ú__le__r= r= r= r> ÚRaiser s r c s g | ]} |gqS r= r= )rX ro ©r r= r> rY s z0TestMethods.test_sort_raises.<locals>.<listcomp>rB r% ) r' ræ r Únowr( rÞ r rô r rq rÆ r Úshuffle)rF rS r} r= r r> Útest_sort_raises s
|
"
|
z TestMethods.test_sort_raisesc C sz t d¡}| ¡ }|}|jdkrR|jd }|d || ||< |d< |d d
}qtt |¡| t|t |¡ | d S )Né@B rÌ r" rã )r' rC r r9 r r r )rF rÕ Zdorá Úmidr= r= r> Útest_sort_degraded s
|
|
|
z TestMethods.test_sort_degradedc C s dd }dd }t jddd}|| ¡ || d¡ || d¡ || d ¡ t jddd}|| ¡ || d¡ || d¡ || d ¡ d S )
|
Nc S s* t | jj t | jj t | jj d S rR ©r rJ Úfortranr r ©rS r= r= r> Úassert_fortran¬ s z-TestMethods.test_copy.<locals>.assert_fortranc S s, t | jj t | jj t | jj d S rR r$ r& r= r= r> Úassert_c± s z'TestMethods.test_copy.<locals>.assert_cr r r& r rÑ )r' r2 r )rF r' r( rD r= r= r> Ú test_copy« s zTestMethods.test_copyr zi,Oc C sX t jd|d}t |jjd|j¡ | i ¡}d|d< t t ¡ t
|
|| W 5 Q R X d S )Nr r r r_ ) r' r2 ÚctypesÚmemsetr$ rÐ Ú __deepcopy__r¨ r r~ r )rF r rD r¦ r= r= r> Útest__deepcopy__Ä s
|
z TestMethods.test__deepcopy__c C sH G dd d}t jd| dgdd}t t¡ | i ¡ W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z:TestMethods.test__deepcopy__catches_failure.<locals>.MyObjc _ s t d S rR r= ©rF r r r= r= r> r, Ó s zGTestMethods.test__deepcopy__catches_failure.<locals>.MyObj.__deepcopy__N)rN rO rP r, r= r= r= r> ÚMyObjÒ s r/ r# rÌ r r )r' ræ r¨ r ré r, )rF r/ rS r= r= r> Ú test__deepcopy__catches_failureÑ s z+TestMethods.test__deepcopy__catches_failurec sÖ t dddg¡}t dddg¡}t ddd g¡}t jj|||gd
|
d jd gd t jt dddg¡ t jt dddg¡ t jt d ddg¡ jdgd t jt dddg¡ t jt dddg¡ t jt ddd g¡ jdgd t jt dddg¡ t jt dddg¡ t jt ddd g¡ t t
|
d fdd t j dkrnd}nd}dd|fg}t jddddg|d jdd t d d d d d g t t jddddg|d d S )!Né r¯ é ZmyÚfirstr gÍÌÌÌÌÌ@g @çÍÌÌÌÌÌ@zid,word,number)r1 r r& ÚwordÚnumberZ duplicatec s j ddgdS )Nr r& )r r= ©rê r= r> r4 ñ r5 z-TestMethods.test_sort_order.<locals>.<lambda>Úlittler rR )r r± Zcol2)rD r# )r¦ r\ )r³ rÌ )rÕ é r r# rÌ r\ r9 ) r' ræ rð Ú
|
fromarraysr r r r5 r6 r r) ÚsysÚ byteorder)rF Úx1r¼ Úx3ZstrtypeZmydtyper= r7 r> Útest_sort_orderÚ s>
|
ÿ ÿ ÿzTestMethods.test_sort_orderc sr t jt jt jfD ]l}t jd|d}|d d d
¡ }| jD ]@}d||f }t| ¡ j|d|| t| ¡ j|d|| q:q|d d }|d d }| jD ]<}d| }t| ¡ j|d|| t| ¡ j|d|| q|d }|d }| jD ]<}d| }t| ¡ j|d|| t| ¡ j|d|| qðd D ]Z}t j d
|
D ]H} t j
|
d d d g|| d}
|
d | ¡}t|
|
¡ t jt |
|
t j d| q@q2d t
|
fddtdD ¡}|d d d
¡ }t d¡} | d d d
} | jD ]>}d| }t| ¡ j|d| | t| ¡ j|d| | qÞd t j
|
fddtdD t jd}|d d d
}t d¡} | d d d
} | jD ]>}d| }t| ¡ j|d| | t| ¡ j|d| | qpt jdtd}ttd|d d
< |d d d
}t d¡} | d d d
} | jD ]>}d| }t| ¡ j|d| | t| ¡ j|d| | qþt dtfdtfg¡} t j
|
dd tdD | d}|d d d
}t d¡} | d d d
} | jD ]>}d| }t| ¡ j|d| | t| ¡ j|d| | qt jddd d}|d d d
}t d¡} | d d d
} d D ]>}d | }t| ¡ j|d| | t| ¡ j|d| | qt jddd d}|d d d
}t d¡} | d d d
} d D ]>}d | }t| ¡ j|d| | t| ¡ j|d| | qt
|
d!d"gddgg¡}t
|
ddgddgg¡}t
|
ddgddgg¡} t| ¡ jdd#| t| ¡ jdd#| t| ¡ ¡ | t
|
g ¡}d$|_t|j |jD ]2}d% |¡}tt j||d#t j|t j d| qpd&}tt j|d d#t j| ¡ t j d| t d'¡} t d'¡}t|jd(d| t jd'td}t|jd(d| t
|
d)d td'D ¡}t|jd(d| t j
|
d*d td'D t jd}t|jd(d| d S )+NrU r r% z!scalar argsort, kind=%s, dtype=%srÜ rw r# zcomplex argsort, kind=%srÖ rî rï rð rñ z'byte-swapped complex argsort, dtype={0}rõ c s g | ]} t | qS r= rö rW r3 r= r> rY ' s z,TestMethods.test_argsort.<locals>.<listcomp>zstring argsort, kind=%sc s g | ]} t | qS r= rö rW r3 r= r> rY 2 s zunicode argsort, kind=%srú zobject argsort, kind=%sr× r¿ c S s g | ] }||fqS r= r= rW r= r= r> rY H s z!structured array argsort, kind=%sr r rû zdatetime64 argsort, kind=%sr z timedelta64 argsort, kind=%srÌ r" rª r z&test empty array argsort with axis={0}z'test empty array argsort with axis=Nonerµ rÍ c S s g | ]}d qS ©Z aaaaaaaaar= rW r= r= r> rY s c S s g | ]}d qS r@ r= rW r= r= r> rY
s ) r' r¶ r· rÅ rC r rÞ r r r ræ rò r r rq rt r2 r( rÎ r rÿ rÝ r7 rÏ r² r r* r¹ )rF r rD r¦ r} r rë rì ró r© rS rê Úrrr³ r¨ r= r3 r> r ÿ sÖ
|
|
|
|
ÿ
|
|
"
|
|
|
|
|
|
|
|
|
|
ÿ ÿ
|
|
zTestMethods.test_argsortc C s8 t d¡}d d¡}tt|j|d tt|j|d d S )NrB ó äÚUTF8rÜ )r' rC Údecoder r) r r ©rF rÕ r r= r= r> Útest_sort_unicode_kind s
|
|
z"TestMethods.test_sort_unicode_kindrD r r# r c C s d|j }|j|dd}t|t d¡| d|j }|j|dd}t|t dd¡| |jdd tjd
|
dtjgd d }t ||d ¡}t|d d S )Nz/Test real (%s) searchsorted with nans, side='l'Úleft©ÚsiderÌ z/Test real (%s) searchsorted with nans, side='r'Úrightr# r r. r rÅ r r% r" )r Ú searchsortedr r' rC ræ rÚ )rF rD r r¦ rá rø r= r= r> Útest_searchsorted_floats s
|
|
z$TestMethods.test_searchsorted_floatsc C s t jdt jd}| jddddddt jt jt jg 7 _| jddddt jt jddt jg 7 _d}|j|dd}t|t d¡| d}|j|d d}t|t dd
|
¡| d }t j dd gd d}| t j d d d¡}t|d| d}t j dd gdd}| t j d dd¡}t|d| d S )Nrí r r r# z-Test complex searchsorted with nans, side='l'rG rH z-Test complex searchsorted with nans, side='r'rJ rB z.Test searchsorted with little endian, side='l'é rM z+Test searchsorted with big endian, side='l'rN )
|
r' r* rÇ rx rÚ ry rK r rC ræ )rF rD r r¦ r= r= r> Útest_searchsorted_complex£ s" && z%TestMethods.test_searchsorted_complexc C sî t d¡}| dddgd¡}t|dddg | dddgd¡}t|dddg t d¡}| dddgd¡}t|dddg | dddgd¡}t|dddg t d¡}| dddgd¡}t|dddg | dddgd¡}t|dddg d S )Nr r# r" rG rJ )r' r rK r rW r= r= r> Ú test_searchsorted_n_elements» s
|
|
|
z(TestMethods.test_searchsorted_n_elementsc C s¦ t d¡}t |j|j d d¡}|dd
|j¡}||d d
< | |d¡}t|| | |d¡}t||d | |d¡}t|| | |d¡}t||d d S )NrB r# r3 rG rJ ) r' rC r2 r1 r9 rW r rK r )rF rD r Ú unalignedr¦ r= r= r> Ú!test_searchsorted_unaligned_arrayÏ s
|
|
|
z-TestMethods.test_searchsorted_unaligned_arrayc C sR t d¡}| dddgd¡}t|dddg | dddgd¡}t|dddg d S )NrH rÄ r rG rJ )r' rC rK r rW r= r= r> Útest_searchsorted_resettingà s
|
|
z'TestMethods.test_searchsorted_resettingc C s d tjd tjd tjd df¡}|D ]Ô}|dkr8d}|dkrZtjd |d
|
}t d ¡}ntjd d |d
|
}t d ¡}| |d ¡}t|| | |d¡}t||d tjd d|d}| |d ¡}t|tjt |tj
|
d
|
| |d ¡}t|tjd tj
|
d
|
q(d S )Nr` r
r¢ ÚDatetimeú?OÚMúM8[D]ú?r" r r rH rG rJ r# r5 ©r7 rß r ) Újoinr' r rC rK r r6 r r* r r )rF ri r© rD rµ r¦ rë r= r= r> Ú test_searchsorted_type_specificè s* ÿ
|
|
z+TestMethods.test_searchsorted_type_specificc s¦ t jddddddddd d
|
d d d dgt jd t t ¡}t fdd D | t fdd D |d t fdd|D | t fdd|D |d d S )Nz'P:\20x_dapi_cy3\20x_dapi_cy3_20100185_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100186_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100187_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100189_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100190_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100191_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100192_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100193_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100194_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100195_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100196_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100197_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100198_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100199_1r c s g | ]} |d ¡qS ©rG ©rK ©rX rx r r= r> rY
|
s z9TestMethods.test_searchsorted_unicode.<locals>.<listcomp>c s g | ]} |d ¡qS ©rJ r\ r] r r= r> rY
|
s r# c s g | ]} | d ¡qS r[ r\ rW r r= r> rY
|
s c s g | ]} | d ¡qS r^ r\ rW r r= r> rY
|
s )r' ræ rt rC r r rÅ r= r r> Útest_searchsorted_unicode
|
s, óò z%TestMethods.test_searchsorted_unicodec C sì t dddddg¡}t |¡}ttt j|dt jdtdd ttt j|dd
|
gd ttt j|dddddgd ttt j|ddddddd gd ttt j|ddddddgd ttt j|dd ddddgd ttt j|dddd ddgd d S ) NrH r" r# rÌ r r )r# rä r ©Úsortergñ?rÄ r% )r' ræ r r rô rK r( r) )rF rD r¾ r= r= r> Ú%test_searchsorted_with_invalid_sorter
|
s
|
ÿ z1TestMethods.test_searchsorted_with_invalid_sorterc C s t j d¡}| ¡ }t |¡}t ddd¡}t| |¡|j||d t dddddgd ¡}| ¡ }dddddg}ddd d
|
d g}t|j|d |d | dd d
|
d dg}t|j|d|d | t d¡}|
|
¡ }t j |¡ | ¡ }t |j |j d d¡}|dd
|j¡}||d d
< | |d |¡}t|| | |d|¡}t||d ||d d
< | |d |¡}t|| | |d|¡}t||d d t jd t jd t jd df¡} | D ]8}
|
|
|
dkrÊd}
|
|
|
dkr
|
t jddg|
|
d}t jddgt jd}t ddg¡} nLt jdddddg|
|
d}t jdddddgt jd}t jdddddgt jd} | |d |¡}t|| | |d|¡}t|| d t jdd |
|
d } | |d |d d
¡}t|t jt|t jd | | d |¡}t|t jdt jd q¶t dddddg¡}t j d t jd} d | dd d
< dddddg| d d d
< | d d d
}t jdddddgt jd} | |d |¡}t|| | |d|¡}t|| d d S ) Ni, r r# r r` r" rÌ rH é( rà r¸ rG )rI ra rµ rJ rB r3 r` r
r¢ rS rT rU rV rW r r r5 rX rÊ r% )r' r r r r Úlinspacer rK ræ rC r r r2 r1 r9 rW r rY r r´ r r6 r r* r )rF rD r¾ r¦ r râ Úkeysr rP ri r© rµ rë Zsrtr= r= r> Ú test_searchsorted_with_sorter*
|
sz
|
|
|
|
ÿ
|
|
|
|
|
z)TestMethods.test_searchsorted_with_sorterc C s¬ G dd dt j}t d¡ |¡}t dd¡ |¡}t d¡ |¡}tt| |d¡| tt| |d¡| tt| |d|¡| tt| |d|¡| d S )Nc @ s e Zd ZdS )z4TestMethods.test_searchsorted_return_type.<locals>.ANrM r= r= r= r> rÑ y
|
s rÑ rH r# rÌ rG rJ )r' r6 rC rW r ru rK )rF rÑ rD r¦ r¾ r= r= r> Ú test_searchsorted_return_typew
|
s z)TestMethods.test_searchsorted_return_typer¯ c C s2 t d¡j|d}tt|jd tt|jd d S ©NrB r iõÿÿÿ)r' rC rE r r) r ©rF r rÕ r= r= r> Ú test_argpartition_out_of_range
|
s z*TestMethods.test_argpartition_out_of_rangec C s2 t d¡j|d}tt|jd tt|jd d S rh )r' rC rE r r) r ri r= r= r> Útest_partition_out_of_range
|
s z'TestMethods.test_partition_out_of_rangec C s8 t d¡}tt|jd t jdtd}tt|jd d S ©NrB ç "@r )r' rC r rô r r( ©rF rÕ Zd_objr= r= r> Útest_argpartition_integer
|
s
|
z%TestMethods.test_argpartition_integerc C s8 t d¡}tt|jd t jdtd}tt|jd d S rl )r' rC r rô r r( rn r= r= r> Útest_partition_integer
|
s
|
z"TestMethods.test_partition_integerÚ kth_dtyper
c C s~ t jd|dd }t g ¡}d|_t|j |jD ]&}d |¡}tt j|||d|| q2d}tt j||d d| ¡ | d S )Nr r r= r z(test empty array partition with axis={0}rª z)test empty array partition with axis=None) r' ræ r7 rq rÏ rò r r r ©rF rq r rD r¨ r r= r= r> Útest_partition_empty_array£
|
s
|
|
z&TestMethods.test_partition_empty_arrayc C s t jd|dd }t g ¡}d|_t|j |jD ]2}d |¡}tt j|||dt j|t j d| q2d}tt j||d dt j|
|
¡ t j d| d S )Nr r r= r z+test empty array argpartition with axis={0}rª z,test empty array argpartition with axis=None) r' ræ r7 rq rÏ rò r r r² r r rr r= r= r> Ú test_argpartition_empty_array¯
|
s
|
|
ÿ ÿz)TestMethods.test_argpartition_empty_arrayc C sò t d¡}ttt j|ddd ttt j|ddd ttt j|ddd tt|jdddd tt|jdddd dD ]x}t g ¡}tt j|d|d| tt j|d|d| t d¡}tt j|d|dd | t|t j|d|d t j|d|d t d d
|
d g¡}|
|
¡ }t t d ¡|¡ t|| ddgddgddgfD ]Ô}t |¡}t |¡}tt j|d|dd |d tt j|d|dd |d t|t j|d|d t j|d|d t|t j|d|d t j|d|d t |j D ] }||d
jd|d qìt|| qBd ddgddd gddd gdd dgdddgdddgdddgdddgfD ]}t |¡}t |¡}tt j|d|dd |d tt j|d|dd |d tt j|d|dd |d t|t j|d|d t j|d|d t|t j|d|d t j|d|d t|t j|d|d t j|d|d t |j D ] }||d
jd|d qLt|| q\t d¡}tt j|d|d| t|t j|d|d t j|d|d t d¡}tt j|d |dd d tt j|d
|
|dd
|
d
|
t|t j|d |d t j|d |d t|t j|d
|
|d t j|d
|
|d t d¡d d d
}tt j|d|dd d tt j|d|dd d t|t j|d|d t j|d|d t|t j|d|d t j|d|d tt j|d|dt j|d|d tt j|d|dt j|d|d t|t j|d|d t j|d|d t d¡}t ||j d ¡}|j d d } tt || ¡| | t d¡}t ||j d d ¡}|j d d } tt || ¡| | t d¡}d|d< tt |d¡d d tt |d¡d d t|t |d¡ d d t|t |d¡ d d t j|d< tt |t |d¡ d ¡ tt t |d¡d ¡ t d¡d }t t d¡d ¡}t j |¡ t |j D ]$}tt j|||d| || q¶t|t j|d|d t j|d|d t|t j|d|d t j|d|d t |j D ] }||d
jd|d q2t|| t dddd dd d d d d d d d d d d d d d d d d d g¡}dd d d g}tt j|||d| d t|t j|||d | d t ddg¡}|jd|d tt|jd tt j|jd dd! ttt j|d tt jt j|ddd! tt|jd tt j|jd dd! ttt j|d tt jt j|ddd! t d¡ d"¡}|jdd|d |jdd|d t j|dd|d t j|dd|d t j|dd |d t j|d d |d |jdd|d |jdd|d t j|dd|d t j|dd|d t j|dd |d t j|d d |d tt|jddd! tt|jd#dd! tt|jdd d! ttt j|d dd! ttt j|d#d d! tt|jddd! tt|jd#dd! ttt j|d dd! ttt j|d#d d! d$d% t jt jt jfD }
|
|
|
D ]¾\} } t} t}t j| | d&}t j |¡ t t j| | d&d'¡}tt jj| t |¡}t |j D ]V}t j|||d}t|| | t |d |
|| t || ||d d
| ||t j|||d t j||d|d}| |d d
|f t j|g|j d | d& ||d d
d |
f j |d d
|f k ¡ d(||d d
|f |d d
d |
f j f d) ||d d
|d d
f j |d d
|f k ¡ d*||d d
|f |d d
|d d
f j f d) | ||t |j d ¡d d
d f t j||d|df t j||d|d}| ||d d
f t j|g|j d | d& ||d |
d d
f ||d d
f k ¡ d(|||d d
f |d |
d d
f f d) |||d d
d d
f ||d d
f k ¡ d*|||d d
f |d d
|d d
f f d) | ||t j||d|dt |j d ¡d d d
f f |
|
¡ }|j||d t|t j|||d |
|
¡ }|j|d|d t|t j||d|d |
|
¡ }|j|d|d t|t j||d|d
|
q
|
q*qrd S )+NrB r" r# rÜ Znonsenser )r¨ r} )r rÍ é rH rc rÌ ro é1 é/ r% rÄ r® iúÿÿÿé) iðÿÿÿé r! iAB r )r" r% rì rí é r )r rÌ rì rì rª ©r" rH r c S s g | ]}d D ] }||fq qS ))rí r® r= )rX r© r¾ r= r= r> rY Z s ÿz.TestMethods.test_partition.<locals>.<listcomp>r ©r r# z %d: %r <= %r©r z %d: %r < %r)!r' rC r rô r r) r ræ r r r r rq r9 r ZrollrÚ r Úisnanr r Ú AxisErrorrÆ r¶ rÅ rÆ ÚtileÚmaprê r r7 r r¬ )rF rÕ r r Zokthrê r r¿ rá r" Útdr© r¾ ZaaeÚatrJ Úd0r Údcr= r= r> r ½
|
s
|
|
|
|
ÿ
|
|
|
ÿÿ ÿ
|
|
|
ÿÿÿ
|
ÿ
|
ÿÿÿÿÿÿÿ
|
|
|
|
"ÿÿ
|
& ÿ
|
,**ÿ..ÿ ÿ,((ÿ,,ÿÿzTestMethods.test_partitionc C st d}t |¡D ]`}t|||
|| d| d t||d
|| k ¡ d|||d
|| f d |d }qd S )Nr zkth %dr± z kth %d, %r not greater equal %dr} r# )r' r r r r¬ )rF rÕ r Úprevr r= r= r> Úassert_partitioned s ÿz TestMethods.assert_partitionedc C sØ t d¡}d}tt|j| tt|j| t d¡ d¡}tt|j|dd tt|j|dd ttt j||dd ttt j||d d t dd d
|
dg¡}t |d ¡}| |d ¡ | |t |d ¡ d ¡ t |t |d ¡ t ||t |d ¡ t d¡}t j
|
|¡ | t |j ¡ t t d¡| t j
|
|¡ t t d¡|| t |j ¡ t d¡}t j
|
|¡ t ddd dg¡}t j
|
|¡ t ||¡}| ||¡ |t ||¡ }| ||¡ t j
|
|¡ t t ||¡| t |t ||¡ | t d¡d d d
}| t |dgd ¡dg¡ | t |dgd ddg ¡dgd ddg ¡ | |t |dgd ¡ dg¡ | |t |dgd ddg ¡ dgd ddg ¡ t d¡}t j
|
|¡ t t d¡d¡}tt j
|
j | t |¡}d}t j||dd}|t |jd ¡d d
d f |j|ddf }t || t |jd D ] }| ||d d
f |¡ q<t j||dd}|t j||ddt |jd ¡d d d
f f }t || t |jd D ] }| |d d
|f |¡ q´d S )Nrb )r r# r" i éç rB r{ r rª r# rÌ r r" ©r rÌ )rå r% r rã r r% rH rÄ r¤ r+ r| )r# rÄ rì r% )r' rC r r) r r rÆ ræ r r r r rq r9 r r rê r7 ) rF rÕ r r re rJ r Úpar¿ r= r= r> Útest_partition_iterative sz
|
|
|
ÿ ÿ
|
|
ÿ
|
ÿ
|
z$TestMethods.test_partition_iterativec C sJ t jdddgdddgd}t j|dd gd
|
}tt j|t|jdd gd
|
| t|t j|t|jdd gd
|
| t|jD ]L}tt j||dd gd
|
| || t|t j||dd gd
|
| || qzt d d d dg¡}t |¡}tt |t|j¡| t|jD ]>}tt ||¡| || t|t ||¡ | || qd S )N)ÚGalahadg333333û?é& )ÚArthurgÍÌÌÌÌÌü?rx )ÚLancelotgffffffþ?r )r z|S10)Úheightró )ÚagerM r r r r& r r Zzebrar ) r' ræ r r r rq r9 r r )rF rÕ r r r= r= r> Útest_partition_cdtypeÙ s: ÿþÿþÿþÿ ÿ
|
z!TestMethods.test_partition_cdtypec C s< t d¡}d d¡}tt|jd|d tt|jd|d d S )NrB rB rC r" rÜ )r' rC rD r r) r r rE r= r= r> Útest_partition_unicode_kindò s
|
|
z'TestMethods.test_partition_unicode_kindc C s t ddD ]}t d|d D ]z}t |¡}tj |¡ |tj dd¡ }tj |j¡}d|||d g}t |¡| }tt ||¡| |d||f d q q
|
d S )NrB rÍ r# r" r zdata: %r
|
kth: %rr± )
|
rq r' rC r r r r9 r r r )rF rñ r¿ rÕ r r r r= r= r> Útest_partition_fuzzø s
|
|
ÿz TestMethods.test_partition_fuzzc C sJ t jd|dd }dddddd g}t ||¡}| t |¡| dg¡ d S )
|
Nr# r r= rÄ rì rÌ r" rí r )r' ræ r r )rF rq r rÕ r r= r= r> Útest_argpartition_gh5524 s z$TestMethods.test_argpartition_gh5524c
|
C s$ t dddgdddggt j¡}t ddgddggddgddgggt j¡}t ddddddgt j¡}t ddddddgt j¡}t ddddddddgt j¡}t ddddddddgt j¡}t| ¡ | t| d ¡| t| d ¡|j ¡ t| ¡ | t| d ¡| t| d ¡|j ¡ d S )
|
Nr# r" rÌ r rH rÄ rì r r )r' ræ r¶ r Úflattenr )rF Zx0r= Zy0Zy0fÚy1Zy1fr= r= r> Ú test_flatten s * zTestMethods.test_flattenr# c C s t ddgddgg¡}t ddgddgg¡}t ddgddgg¡}t d¡ dd¡}t dd d
|
d gd d d dgd
|
d ddgd dddgg¡}t ddddddgdddddd gddd d d d gddd d!d"d#gddd d"d$d%gdd d d#d%d&gg¡}t jt jt jt jfD ]}| |¡} t || | | t || j
|
| | t || | j
|
| t || j
|
| j
|
| t || j
|
¡ | | t || | j
|
¡ | t || j
|
¡ | j
|
¡ | qøt jt jt jt jfD ]b}| |¡} | |¡}
|
t ||
|
|
|
| t ||
|
j
|
|
|
| t ||
|
|
|
j
|
| t ||
|
j
|
|
|
j
|
| q¬t jt jt jt jfD ]}| |¡} t || d d d'
d d
f | j
|
|| d d d'
d d
f ¡ | j
|
¡ t || d d
d d d'
f | j
|
|| d d
d d d'
f ¡ | j
|
¡ t || | d d d'
d d
f j
|
|| | d d d'
d d
f j
|
¡ t || | d d
d d d'
f j
|
|| | d d
d d d'
f j
|
¡ t || d | j d d(
d d
f | d d d(
d d
f j
|
|| d | j d d(
d d
f ¡ | d d d(
d d
f j
|
¡ t || d d d(
d d
f | d | j d d(
d d
f j
|
|| d d d(
d d
f ¡ | d | j d d(
d d
f j
|
¡ q$t jt jt jt jfD ]H}| |¡} | |¡} | |¡} t || | j
|
| t || j
|
| | qPd S ))Nr# r rí é÷ÿÿÿrÄ r rÄ é7 é éë iE ià iõ i' iÿ i ië iø i i@ id i i¬ iD il i i¼ iä i iÄ ið i iô i$ iT iX i iÄ r% r" ) r' ræ rC rÆ rÅ rÈ rÆ rÇ rE r r r r7 )rF r# rD r¦ r³ rÕ ZddtZdtdÚetZeafZebfZedfZeddtfZedtdfr= r= r> Ú test_arr_mult s
|
|
|
|
ýÿûÿ
|
|
|
|
$þ $þ þ þ6>þ6>þ
|
|
|
zTestMethods.test_arr_multZifdFDc C s* t jd|d dd¡}t dd¡}|||}||| ¡ }t|| ||j|j}||j ¡ |j}t|| t |¡}t jd|j d d d ddd
|
|¡}| dd¡}|d
|
}t |j d d
|
|j d
|
k |||}|| ¡ | ¡ }t|| ||j|j}||j ¡ |j ¡ }t|| d S ) Nç @r r" r rý r| r r# r´ r r$ ) r' rC rÆ Ú broadcast_tor r r r r* r1 rW r r4 )rF r# r rD r¦ Zret1Zret2r© r= r= r> Ú test_no_dgemvj s&
|
|
|
|
*
|
|
zTestMethods.test_no_dgemvc C sÖ t ddgddgg¡}t ddgddgg¡}t ddgddgg¡}tt ||¡| |¡ tt t ||¡|¡| |¡ |¡ t |¡}| ||¡ t|t ||¡ t |¡}|j||d t|t ||¡ d S )Nr# r rí r )r¦ rµ )r' ræ r Údotr² ©rF rD r¦ r³ r= r= r> Útest_dot s &
|
|
zTestMethods.test_dotc C s6 d}t jddd}ttt j|| ttt j|| d S ©Nrý ri r r )r' ræ r rô r¢ ©rF r³ rÑ r= r= r> Útest_dot_type_mismatch s z"TestMethods.test_dot_type_mismatchc
|
C s t j d¡ dd t jd D }|D ]}t j dd¡ |¡}td|d}t j dd¡|d< t ||¡}t j|||d }t||t |d
|
t
|
t t j|||d d d
d t
|
t t j|||j d q$d S ) Nr# c S s g | ]}|d krt |¡qS )ZUSVM©r' r )rX Úcoder= r= r> rY ¦ s ÿz8TestMethods.test_dot_out_mem_overlap.<locals>.<listcomp>r¯ rÌ r° r .r· r± r" ) r' r Úseedr r rE r? r¢ r r6 r r) r )rF Údtypesr rD r¦ rø rá r= r= r> Útest_dot_out_mem_overlap¡ s z$TestMethods.test_dot_out_mem_overlapc C s` G dd dt j}t d¡ |¡}t d¡ |¡}t d¡}t j|||d t j|||d d S )Nc @ s e Zd ZdS )z,TestMethods.test_dot_matmul_out.<locals>.SubNrM r= r= r= r> ÚSub¹ s r r r· )r' r6 r rW r¢ Úmatmul)rF r rD r¦ rµ r= r= r> Útest_dot_matmul_out· s
|
z TestMethods.test_dot_matmul_outc C sN G dd d}t ttj| | t ttj| | t ttj| | d S )Nc @ s e Zd Zdd ZdS )z@TestMethods.test_dot_matmul_inner_array_casting_fails.<locals>.Ac _ s t d S rR ©ÚNotImplementedErrorr. r= r= r> r Æ s zJTestMethods.test_dot_matmul_inner_array_casting_fails.<locals>.A.__array__Nr r= r= r= r> rÑ Å s rÑ )r r± r' r¢ r® Úinner©rF rÑ r= r= r> Ú)test_dot_matmul_inner_array_casting_failsà s z5TestMethods.test_dot_matmul_inner_array_casting_failsc C s t d¡ ddd¡}t ||¡}t j|||d}t||k t|| t d¡ ddd¡}t j|||d d d
df d}t|j|jk t|| d S )Nrf r" rÌ r· r% .)r' rC rÆ r® r r rv r£ r= r= r> Útest_matmul_outÎ s
|
zTestMethods.test_matmul_outc C s¼ t d¡ d¡}t| ¡ dddg t| d¡dddg t| d¡dddg t| d ¡d
|
d g tt j|jddd tt j|jddd tt j|jddd tt|jddd t d ¡ d¡}t| ¡ ddgddgg t| d¡ddgddgg t| d¡dgdgg t| d ¡d
|
gdgg tt|jddd t| ddd¡ddgd
|
dgg t| ddd¡ddgddgg t|jdddddgdgg t| ddd¡ddgd
|
dgg d S )Nr+ rÐ r rH rB r# rÄ r r% r rí )Úaxis1Úaxis2r ©r" r" r" rì r" rÌ ©r< r¶ r· )r' rC rÆ r Údiagonalr r r) rW r= r= r> Ú test_diagonalÚ s& zTestMethods.test_diagonalc C s t d¡ ¡ }t|jj t|jj t t d¡¡}t|jj t|jj t t d¡¡}t|jj t|jj d S ©NrÌ )r' Úeyerº r rJ rK r} Zdiagr¡ r= r= r> Ú test_diagonal_view_notwriteableñ s z+TestMethods.test_diagonal_view_notwriteablec C sP t d¡}tr tt |¡dk tdD ] }| ¡ q(trLtt |¡dk d S )N©rµ rµ ro rµ )r' r* r r r; Ú getrefcountrq rº )rF rD r¿ r= r= r> Útest_diagonal_memleakþ s
|
|
z!TestMethods.test_diagonal_memleakc C sR t jg t jd}t d¡}tdD ]}t j|||d q"trNtt |¡dk d S )Nr ç @rµ r· ro ) r' ræ rÈ rq r¢ r r r; rÀ )rF rD rá Ú_r= r= r> Útest_size_zero_memleak s
|
z"TestMethods.test_size_zero_memleakc C s6 t d¡ d¡}t| ¡ d t| d¡d t| d¡d t| d¡d t d ¡ d
|
¡}t| ¡ d d g t| d¡d d g t| d¡d d g t| d¡ddg t| ddd¡d d g t| ddd ¡ddg t| ddd ¡d dg t|jddd ddd g t d¡}|j|d}||ks2td S )Nr+ rÐ ru r r# rf r% r¤ r r¸ rÄ r" rÌ r rH rí r r¹ r· )r' rC rÆ r Útraceræ r~ )rF rD r¦ rµ r¾ r= r= r> Ú
|
test_trace s"
|
zTestMethods.test_tracec C sB G dd dt j}t d¡ d¡ |¡}| ¡ }tt|| d S )Nc @ s e Zd ZdS )z0TestMethods.test_trace_subclass.<locals>.MyArrayNrM r= r= r= r> ÚMyArray+ s rÇ r r¸ )r' r6 rC rÆ rW rÅ r ru )rF rÇ r¦ r÷ r= r= r> Útest_trace_subclass( s z TestMethods.test_trace_subclassc C s t jd }t jd }|| d D ]}t jddddddg|d}t jd |d}| dddgdddg¡ t|| t jd
|
|d}| dddgdddg¡ t|| d d¡ q d D ]}t jd dd dd dg|d}t jd |d}| dddgdgd ¡ t|| t jd
|
|d}| dddgdgd ¡ t|| d d¡ q¬t d ¡}d |j_t t
|
|jdddgdddg dd dg}t t t j|dd gd d S )Nr
r¢ r r r# rÌ rH r rÄ rä r" rW FT) r' r ræ r* Úputr rÆ rJ rK r r) rô )rF ÚicodesÚfcodesr© r rD Z bad_arrayr= r= r> Útest_put2 s.
|
|
|
|
|
|
zTestMethods.test_putc
|
C sì t ddgddgg¡}t| ¡ ddddg t| ¡ jj t| d¡ddddg t|jddddddg t|jddddddg t|jddddddg t|jddjj t|jd dddddg t|jd djj t| ¡ | d
|
¡ t jddgddggdd}t| ¡ ddddg t|jddddddg t|jd dddddg t|jddjj t|jd djj t| ¡ | d
|
¡ t|jdd|jd
|
dd t ddgddgg¡d d d
|
d d
f }t| ¡ ddddg t|jddddddg t|jddddddg t|jddddddg t|jd dddddg t|jd djj t d ¡d d d
}t| d ¡jj t| d¡jj t| d¡jj t d ¡d d d
}| dddd¡ d
|
d ¡}t
|
|j }d|d< ||_ t|jd djj t| d ¡t ddd¡ t d¡}| dddd¡ d
|
d ¡}t
|
|j }d|d< ||_ tt |jd d|¡ t|jd dt d¡ t d¡d d d
|
dd¡}t|jddjj t|jd djj t| d¡ddddg t| d ¡ddddg t dgg¡}d|_ t d¡j dkrftt | d ¡|¡ t| d ¡j |jjf dD ]8}t d¡}t| |¡dg tt | |¡|¡ qjt d¡d d d
dddd¡}|dd d d
f }t| d ¡ddddd dddg t| d¡ddddd dddg t| d¡ddddd dddg t| d¡dd ddddddg |d d d
df }t| d ¡dddddd dd g t| d¡dddddd dd g t| d¡dddddd dd g t| d¡dddddd dd g d S ) Nr r# r" rÌ r r r& rÑ ÚKr% rB r® rã r ru r r )r i° ©r )r r rÑ rÍ r¯ .r+ r rÄ é rÄ r2 )r' ræ r r r rJ r} rÆ rC ÚswapaxesrÎ rÎ Úmay_share_memoryr r r1 )rF rD rÎ r r¦ r= r= r> Ú
|
test_ravelZ s (
|
|
|
|
zTestMethods.test_ravelc C sà G dd dt j}t d¡ |¡}tt| d¡| tt| d¡| tt| d¡| tt| d¡| t d¡d d d
|¡}tt| d¡| tt| d¡| tt| d¡| tt| d¡| d S ) Nc @ s e Zd ZdS )z6TestMethods.test_ravel_subclass.<locals>.ArraySubclassNrM r= r= r= r> Ú ArraySubclass¶ s rÓ rB r r rÑ rÍ r" )r' r6 rC rW r ru r )rF rÓ rD r= r= r> Útest_ravel_subclassµ s z TestMethods.test_ravel_subclassc C s² t d¡ dddd¡ ¡ }t |j¡}t|jd | ¡ }tt j |j
|
dd tt j |j
|
dd tt j |j
|
dd tt j |j
|
dd t d dD ] }t d dD ]
|
}t ||fD ]ø\}}|
|
||¡}t |j} |j| | |< |j| | |< t|j| t|||f d
|
d |jD \}
|
} } } d d |jD \}}}}t||| || || || f |||
|
|| || || f t|||f t|jd t|||f |dkr°|}q°qqd S ) NrÄ r# r" rÌ r ZOWNDATAéûÿÿÿr éüÿÿÿc S s g | ] }|d qS rh r= ©rX Údimr= r= r> rY Ú s z-TestMethods.test_swapaxes.<locals>.<listcomp>c S s g | ] }|d qS rh r= r× r= r= r> rY Û s )r' rC rÆ r Úindicesr7 r rJ r r rÐ rq Ú enumeraterÎ r rÚ )rF rD r r¦ r¿ rñ r Úsrcr³ r7 Zi0rî Úi2Zi3Zj0Zj1Zj2Zj3r= r= r> Ú test_swapaxesÅ s2
|
þ zTestMethods.test_swapaxesc s´ t dddg¡ ¡ }t j|j t j|j t| ¡ t|t ¡ t dddgd¡ ¡ }t j|j t j|j t| ¡ t|t ¡ t dddg¡ ¡ }t | t| ¡ t|t ¡ t dd d
|
g¡ ¡ }t | t| ¡ t|t ¡ t dddd gt¡ ¡ }t|d d D t| ¡ t|t ¡ t ddd d gt¡ tt fdd tt fdd d S )Nù ð? ð¿rå ù 7@ 7@r r# r" rÌ rý r ç @c S s g | ] }| ¡ qS r= ©Ú conjugate)rX r r= r= r> rY s z.TestMethods.test_conjugate.<locals>.<listcomp>r× c s ¡ S rR )Úconjr= r r= r> r4 r5 z,TestMethods.test_conjugate.<locals>.<lambda>c s ¡ S rR rá r= r r= r> r4 r5 )
|
r' ræ rã r rx ry râ r( r rô )rF Úacr= r r> Útest_conjugateå s<
|
|
zTestMethods.test_conjugatec C sB t dddg¡}t |¡}| |¡}||ks0tt|| ¡ d S )NrÞ rå rß )r' ræ rß râ r~ r )rF rD rµ rL r= r= r> Útest_conjugate_out
|
s
|
|
|
z TestMethods.test_conjugate_outc
|
C s® ddddddddd d
|
d d d dddg}|D ]}t jd|d}t jdg|d}t jdggggg|d}d |¡}t|}t||| t|}t||| t|} t| || q(d S )Nrî rÜ r¤ r- r` Úu2Úu4Úu8r× rÕ Úgr ÚDÚGrW r rì r z
|
dtype: {0})r' ræ rò r¹ r )
|
rF r« r© rD r¦ r³ r ZapZbpÚcpr= r= r> Útest__complex__ s2 ý
|
zTestMethods.test__complex__c C sÈ ddddddddd d
|
d d d dddg}|D ]$}t jdddg|d}ttt| q(t ddg¡}t jd|d}ttt| t jddg|d}ttt| t d¡}ttt| t dgd¡}ttt| d S )Nrî rÜ r¤ r- r` rç rè ré r× rÕ rê r rë rì rW r r# r" rÌ r r rí )rý rÌ )gü©ñÒMb`?rì z1+1jr¶ )r' ræ r rô r¹ r )rF r« r© rD r¦ r³ rÕ rë r= r= r> Ú test__complex__should_not_work% s4 ý
|
z*TestMethods.test__complex__should_not_workN)hrN rO rP rÞ r¯ r° r¶ r¿ rË rÏ rÑ rÖ r× rÙ r r¨ r© r« r' r3 rµ r· r¹ Úfloat16rÅ rÈ rD rà r³ r´ r¶ r¸ rã rí rô rv rt rù rþ r r r r r r r r# r) r- r0 r? r rF ræ rÚ rL rN rO rQ rR rZ r_ rb rf rg r rj rk ro rp rs rt 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= r= r> r¤ sÞ
|
þ
|
ÿÿ
|
|
|
|
|
|
%
|
ý
|
M
|
|
|
|
QD
|
|
M
|
|
([ % r¤ c @ s e Zd Zdd ZdS )ÚTestCequenceMethodsc C s8 t dt d¡ dd¡k t dt d¡ dd¡k d S )Nç @g 0@r ç 4@)r r' rC rÆ rE r= r= r> Útest_array_contains< s z'TestCequenceMethods.test_array_containsN)rN rO rP rô r= r= r= r> rñ ; s rñ c @ sf e Zd Zdd Zdd Zej dddg¡dd Zd
|
d Z d d Z
|
dd Z dd Z dd Z dd ZdS )Ú TestBinopc C s t t dg¡t ddg¡ ddg t ddg¡d d d
}t ||t ddg¡ ddg t dg¡}t dg¡}|| }|| }|| }|| }t|| t|d ||d | | || }t|| t|d t dg¡}t dg¡}|| | }t|d t|d t|d d S ) Nç à?rý r r" ç Ð?rH rÌ g«ªªªªª @)r r' ræ r r
|
)rF rÕ rD r¦ r³ r= r= r> Ú test_inplaceA s0 ÿÿ
|
|
|
|
|
|
zTestBinop.test_inplacec s¶ t jdtft jdtft jdtft jdtft jdtft jdtft jdtft j dt
|
ft j dt
|
ft j dt
|
ft j dt
|
ft jdt
|
ft jdt
|
ft jdtfdG dd dt fdddd d
|
d d d dd }dfdd }d fdd }||tddd ||tddddd ||tddddd ||t jddddddd ||td|dddd ||t jd|dddd ||td dddd ||t jd dddddd d S ) NTF)ru Úsubr0 ÚtruedivÚfloordivÚmodÚdivmodÚpowÚlshiftÚrshiftÚandÚxorÚorr® c @ s e Zd ZdS )z7TestBinop.test_ufunc_binop_interaction.<locals>.CoercedNrM r= r= r= r> ÚCoerced s r c s d S rR r= rE )r r= r> Ú
|
array_impl s z:TestBinop.test_ufunc_binop_interaction.<locals>.array_implc S s dS )NÚforwardr= r r= r= r> Úop_impl s z7TestBinop.test_ufunc_binop_interaction.<locals>.op_implc S s dS )NÚreverser= r r= r= r> Úrop_impl s z8TestBinop.test_ufunc_binop_interaction.<locals>.rop_implc S s dS )Nzin-placer= r r= r= r> Úiop_impl s z8TestBinop.test_ufunc_binop_interaction.<locals>.iop_implc _ s d||||fS )NÚ__array_ufunc__r= )rF ÚufuncÚmethodr r r= r= r> Úarray_ufunc_impl s z@TestBinop.test_ufunc_binop_interaction.<locals>.array_ufunc_implÚ__main__c s¢ d i}|dk r||d< D ].}|d |¡< |d |¡< |d |¡< q |dk r\||d< | ||d}td |}t|tjrt d
|
d ¡ d d ¡ |¡S | S d S ) Nr FÚ__array_priority__ú__{0}__ú__r{0}__ú__i{0}__r )rv Úclass_namespacerN z(type('MyType', (base,), class_namespace)rÌ rì r" )rò r! Ú
|
issubclassr' r6 rC rÆ rW )rv Úarray_priorityÚ array_ufuncZalleged_moduler ÚopZeval_namespaceZMyType)r r
|
r Úopsr r= r> Úmake_obj s& þÿ z8TestBinop.test_ufunc_binop_interaction.<locals>.make_objc s8 ¡ D ](\ \}}}d |||f }tjdd|d dd¡g} |rX| | d d ¡ | D ]Ò}
|
t|
|
d ¡} fdd } |rt| | t| nl|r®t| | d d
|
| nRt | tj
|
rìt | j tj
|
j krì| | | } t | j| jk| nttf| | |d t|
|
d ¡}|rD|| } t| d d
|
|d t| d ||d nVt | tj
|
rt | j tj
|
j kr| || } t | j| jk| nttf|| |d |rt |
|
tj
|
rt|
|
d ¡}|rØt| | t|d n®|r<|| } t| d d
|
| t| d || t t | d d tk| t | d d d |
|
k| nJt | tj
|
rrt | j tj
|
j krrt || |
|
k| nttf|| |d tt d }|d kr¬tt d d }|d krÀtt }t|| |
|
d| t | tj
|
s|rút||
|
| d| n |rt||
|
| d d
|
| |r\t|| |
|
d d
|
| q\qd S )Nz-op: %s, ufunc: %s, has_inplace: %s, dtype: %srÌ rì r r" r r c s& dkr t t| t | d S | S d S )Nrý r )r ru Útuple)r ©r r= r> Ú first_out_arg¸ s zLTestBinop.test_ufunc_binop_interaction.<locals>.check.<locals>.first_out_argr r± r r# r r% rµ rà r r )Úitemsr' rC rÆ r r~ rò r ÚNotImplementedru r6 rY r r Ú __class__r rô r r/ r] )ÚobjZbinop_override_expectedZufunc_override_expectedZinplace_override_expectedÚ check_scalarr Z has_inplacer r² Z
|
check_objsrS Ú
|
arr_methodr rL Z arr_rmethodZ arr_imethodZop_fn)r r r r> Úcheck s®
|
ÿ
|
ÿ ÿÿ ÿ
|
ÿÿÿ ÿ
|
ÿ ÿÿ þ
|
|
|
ÿÿz5TestBinop.test_ufunc_binop_interaction.<locals>.checki À)r r# )r" )r r )r )FFr )T)r' ru rÿ ÚsubtractÚmultiplyÚ true_divideZ floor_divideÚ remainderrý r^ rÝ Z
|
left_shiftZ right_shiftZ bitwise_andZ bitwise_xorZ
|
bitwise_orr® rè r( r6 )rF r r r$ r= )r r r
|
r r r r> Ú test_ufunc_binop_interactionf sd
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ò ÿ ÿ^ÿÿ ÿ
|
ÿ ÿÿz&TestBinop.test_ufunc_binop_interactionÚpriorityNú runtime errorc s\ G fddd}G dd dt j}t d¡ |¡| }|jdksHt|d dksXtd S ) Nc s$ e Zd Ze fddZdd ZdS )zBTestBinop.test_ufunc_binop_bad_array_priority.<locals>.BadPriorityc s dkrt d S )Nr+ z#RuntimeError in __array_priority__!r= rE ©r* r= r> r % s zUTestBinop.test_ufunc_binop_bad_array_priority.<locals>.BadPriority.__array_priority__c S s dS )Nr r= r r= r= r> Ú__radd__+ s zKTestBinop.test_ufunc_binop_bad_array_priority.<locals>.BadPriority.__radd__N)rN rO rP Úpropertyr r- r= r, r= r> Ú BadPriority$ s r/ c @ s e Zd ZdZdS )zBTestBinop.test_ufunc_binop_bad_array_priority.<locals>.LowPriorityiüÿÿN©rN rO rP r r= r= r= r> Ú LowPriority. s r1 rÌ r r r )r' r6 rC rW r7 r~ )rF r* r/ r1 rL r= r, r> Ú#test_ufunc_binop_bad_array_priority s
|
|
z-TestBinop.test_ufunc_binop_bad_array_priorityc C s¢ G dd d}| }t |dg¡}td|ko2d|k t j|dgdd}td|koXd|k t|d d t j|dgdd}td|kod|k t|d d d S ) Nc @ s e Zd Zdd ZdS )zDTestBinop.test_ufunc_override_normalize_signature.<locals>.SomeClassc _ s |S rR r= ©rF r r ÚinputsÚkwr= r= r> r ; s zTTestBinop.test_ufunc_override_normalize_signature.<locals>.SomeClass.__array_ufunc__N©rN rO rP r r= r= r= r> Ú SomeClass: s r7 r# ÚsigÚ signaturezii->i)r8 )r9 )r' ru r r )rF r7 rD r5 r= r= r> Ú'test_ufunc_override_normalize_signature8 s z1TestBinop.test_ufunc_override_normalize_signaturec C s" G dd d}| }t d¡}tt |¡d tt ||¡d tt j||dd tt j||fdd tt ||¡d tt j||dd tt j||fdd tt ||¡d tt |d |¡d tt |||¡d tt j||d fdd tt j|||fdd tt j|d |fdd tt j|||fdd tt j|||fdd tt t j||d W 5 Q R X ttt j||fd tt ||¡d tt ||¡d tt |||¡d tt |||¡d tt j|||dd tt j|||fdd tt j|||dd d S ) Nc @ s e Zd Zdd ZdS )z4TestBinop.test_array_ufunc_index.<locals>.CheckIndexc _ sN t |D ]\}}|| kr| S qt |d D ]\}}|| kr.|f S q.d S rÒ )rÚ )rF r r r4 r5 r¿ rD rñ r= r= r> r M s
|
zDTestBinop.test_array_ufunc_index.<locals>.CheckIndex.__array_ufunc__Nr6 r= r= r= r> Ú
|
CheckIndexL s r; r r rÓ r· rh r# ) r' rC r ÚsinÚmodfr rô r) ru )rF r; rD Údummyr= r= r> Útest_array_ufunc_indexH s8
|
|
|
z TestBinop.test_array_ufunc_indexc C s G dd dt j}t dg¡ |¡}t dg¡}t dg¡}t |||¡ t|d d tt|| d|d< t j|||d t|d d tt|| d S )Nc @ s e Zd Zdd ZdS )z-TestBinop.test_out_override.<locals>.OutClassc _ s> d|kr:| ¡ }| d¡ t||}||||d d d< d S )Nrµ r .)r rK r~ )rF r r r4 r5 Ztmp_kwr# r= r= r> r { s
|
|
|
z=TestBinop.test_out_override.<locals>.OutClass.__array_ufunc__Nr6 r= r= r= r> ÚOutClassz s r@ r rH rÄ rÍ r· )r' r6 ræ rW r& r r ru )rF r@ rÑ r[ r r= r= r> Útest_out_overridex s zTestBinop.test_out_overridec C sÆ G dd dt j}t jdt jd |¡}t|d d tt |d W 5 Q R X tt |d W 5 Q R X tt |d W 5 Q R X tt |d
|
W 5 Q R X tt |d W 5 Q R X d S ) Nc @ s e Zd Zdd ZdS )z:TestBinop.test_pow_override_with_errors.<locals>.PowerOnlyc _ s |t jk rtdS )NúPOWER!)r' r^ r± r3 r= r= r> r s
|
zJTestBinop.test_pow_override_with_errors.<locals>.PowerOnly.__array_ufunc__Nr6 r= r= r= r> Ú PowerOnly s rC ç @r ç @rB rö r r# r% r" )r' r6 ræ rÈ rW r r r± )rF rC rD r= r= r> Ú test_pow_override_with_errors s
|
|
|
|
|
z'TestBinop.test_pow_override_with_errorsc s G fddd dd }t d d dg¡}t|d |d| t|d |d | t|d |d| t|d
|
|d
|
| t|d |d| d S ) Nc sB e Zd Zd ddZdd Zdd Z fdd Z fd
|
d ZeZdS ) z8TestBinop.test_pow_array_object_dtype.<locals>.SomeClassNc S s
|
|| _ d S rR ©rÒ rÑ r= r= r> rU ¥ s zATestBinop.test_pow_array_object_dtype.<locals>.SomeClass.__init__c S s t dd S )Nz __mul__ should not be called©r~ r r= r= r> Ú__mul__© s z@TestBinop.test_pow_array_object_dtype.<locals>.SomeClass.__mul__c S s t dd S )Nz __div__ should not be calledrH r r= r= r> Ú__div__¬ s z@TestBinop.test_pow_array_object_dtype.<locals>.SomeClass.__div__c s | j | dS )NrG rG )rF Úexp©r7 r= r> Ú__pow__¯ s z@TestBinop.test_pow_array_object_dtype.<locals>.SomeClass.__pow__c s t | r| j|jkS d S rR )ru rÒ r rL r= r> r ² s
|
z?TestBinop.test_pow_array_object_dtype.<locals>.SomeClass.__eq__)N) rN rO rP rU rI rJ rM r Ú__rpow__r= rL r= r> r7 ¤ s
|
r7 c s t fdd|D ¡S )Nc s g | ] }| qS r= r= rZ ©rK r= r> rY ¹ s zJTestBinop.test_pow_array_object_dtype.<locals>.pow_for.<locals>.<listcomp>r& )rK rS r= rO r> Úpow_for¸ s z6TestBinop.test_pow_array_object_dtype.<locals>.pow_forr# r" rÌ rö r r% rÿ )rF rP Úobj_arrr= rL r> Útest_pow_array_object_dtype¢ s z%TestBinop.test_pow_array_object_dtypec C s@ G dd dt j}t d¡ |¡}tt |
|
W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z2TestBinop.test_pos_array_ufunc_override.<locals>.Ac _ s t ||dd |D |S )Nc S s g | ]}| tj¡qS r= )rW r' r6 rW r= r= r> rY Æ s ÿzVTestBinop.test_pos_array_ufunc_override.<locals>.A.__array_ufunc__.<locals>.<listcomp>)r~ )rF r r r4 r r= r= r> r Å s
|
ÿÿzBTestBinop.test_pos_array_ufunc_override.<locals>.A.__array_ufunc__Nr6 r= r= r= r> rÑ Ä s rÑ r, )r' r6 ræ rW r rô )rF rÑ Ztstr= r= r> Ú test_pos_array_ufunc_overrideà s
|
z'TestBinop.test_pos_array_ufunc_override)rN rO rP rø r) r¨ r© r« r2 r: r? rA rF rR rS r= r= r= r> rõ @ s % 9
|
0!rõ c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestTemporaryElidec C sF ddl m} t d¡}||\}}|| t|| t||| d S )Nr )Ú incref_elideé )r| rU r' r r )rF rU rÕ ÚorigrL r= r= r> Útest_extension_incref_elideÐ s
|
|
z.TestTemporaryElide.test_extension_incref_elidec C sV ddl m} ddddt d¡g}||}t|d t d¡ t||d |d d S )Nr )Úincref_elide_lr# rV r )r| rY r' r r )rF rY rÔ rL r= r= r> Ú!test_extension_incref_elide_stackß s
|
z4TestTemporaryElide.test_extension_incref_elide_stackc C sx t jdt jd}t|| dd jt d¡ || d }t|jt d¡ t || d¡}t|jt d¡ || d }t|jt d¡ || d }t|jt t j¡ t jdt jd}t|| | t j¡ jt d¡ | t j¡}t|| | j|j t jdt j d}t|| | j|j t j
|
t j t j fD ]<}t jd|d}t |d }t|jt d |jd ¡ q6d S )
|
Ni@ r r" éÞ r r r rV zf%d)r' r r¸ r r r' rÅ rE rÈ rD rÆ rÇ rê Úabsr1 )rF rÕ rê r× rÔ r© r³ r= r= r> Útest_temporary_with_castî s( " z+TestTemporaryElide.test_temporary_with_castc C sB t jdtd}t jdtd}d| | }t|d t|jd d S )N)éÐ r# r r^ r# )r^ r^ )r' r rÝ r r r7 )rF rÕ r¦ rê r= r= r> Útest_elide_broadcast s
|
|
z'TestTemporaryElide.test_elide_broadcastc C s$ t ¡ }tt||@ t jk d S rR )r' rP r rY r¡ r= r= r> Útest_elide_scalar s z$TestTemporaryElide.test_elide_scalarc C s t jdt jd}|jd d S )NrV r r" )r' r2 rÈ ry r¡ r= r= r> Útest_elide_scalar_readonly s z-TestTemporaryElide.test_elide_scalar_readonlyc C s, t t t d¡d¡j¡d }t|d d S )Nr# rV rü r )r' rX r r* Úflatr )rF rê r= r= r> Útest_elide_readonly' s z&TestTemporaryElide.test_elide_readonlyc C s2 t d¡d d d
}|j ¡ d }~t|d d S )Ni r" r# )r' r rb r r rW r= r= r> Útest_elide_updateifcopy, s z*TestTemporaryElide.test_elide_updateifcopyN) rN rO rP rX rZ r] r_ r` ra rc rd r= r= r= r> rT Í s !
|
rT c @ sl e Zd Zdd Zej dejej g¡dd Z
|
ej dejej g¡ejj e oVe jjdkddd d
|
Zd S ) ÚTestCAPIc C sP ddl m} t|d t|d t|dd t|d t|d d S ) Nr )ÚIsPythonScalars foobarr# r" r¸ r rD )r| rf r )rF rf r= r= r> Útest_IsPythonScalar4 s z TestCAPI.test_IsPythonScalarÚ converterc C sb |ddkst |ddks t |t d¡dks6t |ddksFt |t dg¡dks^t d S ) NrB rÊ r% rò r )r r )r )r~ r' ræ ©rF rh r= r= r> Ú test_intp_sequence_converters< s
|
z&TestCAPI.test_intp_sequence_converters)rì rÌ r z PyPy bug in error formattingrn c C s t jtdd |t W 5 Q R X t jtdd |d W 5 Q R X t jtdd |dg W 5 Q R X t jtdd |d W 5 Q R X d S )Nz5expected a sequence of integers or a single integer, r z?expected a sequence of integers or a single integer, got '32.0'g @@z2'float' object cannot be interpreted as an integerzMaximum allowed dimensionì )r¨ r rô r( r) ri r= r= r> Ú$test_intp_sequence_converters_errorsJ s ÿÿÿÿz-TestCAPI.test_intp_sequence_converters_errorsN)rN rO rP rg r¨ r© r« Ú_multiarray_testsZrun_scalar_intp_converterZ run_scalar_intp_from_sequencerj rª r r; ÚimplementationÚversionrl r= r= r= r> re 3 s ÿÿ
|
ÿÿÿre c @ s e Zd Zdd ZdS )ÚTestSubscriptingc C s> t dddg¡}tt|d t j tt|d t jk d S )Nr# r" rÌ r rC )r' ræ r ru rØ rY r6 r r= r= r> Útest_test_zero_ranka s z$TestSubscripting.test_test_zero_rankN)rN rO rP rq r= r= r= r> rp ` s rp c @ s¨ e Zd Zejjejdkdddd Zdd Z ejjejdk ddd d
|
Z
|
d d Z d d Z dd Z dd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd!S )"Ú TestPicklingrH zSthis tests the error messages when trying toprotocol 5 although it is not availablern c C s t d¡}d S rA r )rF ræ r= r= r> Ú$test_correct_protocol5_error_messageh s z1TestPickling.test_correct_protocol5_error_messagec C s® t }tj|ddfgdt fdtfdtfgd}tjdgdtfdtfdtfgd}tdtjd D ]H}t tj
|
||d ¡}t tj
|
||d ¡}t |j |j t |j |j q`d S )
|
Nr# r rD r¦ r³ r ©Zxxxr# r r" r ) r( r' ræ rÝ rÿ rÚ rq r rr rs rt r r )rF Z my_objectZarr_with_objectZarr_without_objectr Zdepickled_arr_with_objectZ depickled_arr_without_objectr= r= r> Ú#test_record_array_with_object_dtypen s,
|
þþ ÿ ÿÿÿz0TestPickling.test_record_array_with_object_dtyper c C s` t jdddgdddggdd}g }tj|d|jd }t|d
|
ksDttj||d }t|| d S ) Nr# r" rÌ r rH rÄ r r& )r r r r ) r' ræ r rt r r r~ rs r )rF Zf_contiguous_arrayr r¡ Z depickled_f_contiguous_arrayr= r= r> Útest_f_contiguous_array s ÿÿz$TestPickling.test_f_contiguous_arrayc C sr t d¡ dd¡d d
d d
f }|jjr.t|jjr:ttdtj d D ]"}t
|
tj ||d¡}t || qJd S )Nr+ rÌ r r" r# r ) r' rC rÆ rJ r r~ r rq r rr rs rt r )rF Znon_contiguous_arrayr Z depickled_non_contiguous_arrayr= r= r> Útest_non_contiguous_array s " ÿz&TestPickling.test_non_contiguous_arrayc
|
C sÆ t dtjd D ]°}t ddgddgddgg¡}|t |¡tjdgd tdffd
|
tfd tfgd g}d d |D }|D ]&}t |t
|
tj ||d¡d| d qt~~~t |D ]}| d ks¬t q¬qd S )Nr" r# rí rì r rÌ r rt rD r¦ r³ r c S s g | ]}t |¡qS r= )ÚweakrefÚref©rX rD r= r= r> rY ¬ s z/TestPickling.test_roundtrip.<locals>.<listcomp>r ú%rr± )rq r rr r' ræ rê rÚ rÝ rÿ r rs rt r r~ )rF r r ZDATAÚrefsrD ry r= r= r> Útest_roundtrip¢ s& ÿý þzTestPickling.test_roundtripc C s t j|ddS )NÚlatin1)Úencoding)r rs ©rF r! r= r= r> Ú_loads· s zTestPickling._loadsc C s4 d}t jddddgt jd}| |¡}t|| d S )Ns} cnumpy.core._internal
|
_reconstruct
|
qcnumpy
|
ndarray
|
qK
UbRq(K
cnumpy
|
dtype
|
qUi1K KRq(U|NNJÿÿÿÿJÿÿÿÿtbUtb.r# r" rÌ r r ©r' ræ r³ r r ©rF r¾ rD r r= r= r> Útest_version0_int8¼ s
|
z TestPickling.test_version0_int8c C s4 d}t jddddgt jd}| |¡}t|| d S )Ns cnumpy.core._internal
|
_reconstruct
|
qcnumpy
|
ndarray
|
qK
UbRq(K
cnumpy
|
dtype
|
qUf4K KRq(U<NNJÿÿÿÿJÿÿÿÿtbU ? @ @@ @tb.rý r rà rò r ©r' ræ rÅ r r r r= r= r> Útest_version0_float32 s
|
z"TestPickling.test_version0_float32c C s2 d}t ddiddig¡}| |¡}t|| d S )Ns cnumpy.core._internal
|
_reconstruct
|
qcnumpy
|
ndarray
|
qK
UbRq(K
cnumpy
|
dtype
|
qUO8K KRq(U|NNJÿÿÿÿJÿÿÿÿtb]q(}qUaKs}qUbKsetb.rD r# r¦ r" ©r' ræ r r r r= r= r> Útest_version0_objectÈ s
|
z!TestPickling.test_version0_objectc C s4 d}t jddddgt jd}| |¡}t|| d S )Ns cnumpy.core._internal
|
_reconstruct
|
qcnumpy
|
ndarray
|
qK
UbRq(KK
cnumpy
|
dtype
|
qUi1K KRq(KU|NNJÿÿÿÿJÿÿÿÿtbUtb.r# r" rÌ r r r r r= r= r> Útest_version1_int8Ï s
|
z TestPickling.test_version1_int8c C s4 d}t jddddgt jd}| |¡}t|| d S )Ns cnumpy.core._internal
|
_reconstruct
|
qcnumpy
|
ndarray
|
qK
UbRq(KK
cnumpy
|
dtype
|
qUf4K KRq(KU<NNJÿÿÿÿJÿÿÿÿtbU ? @ @@ @tb.rý r rà rò r r
r r= r= r> Útest_version1_float32Õ s
|
z"TestPickling.test_version1_float32c C s2 d}t ddiddig¡}| |¡}t|| d S )Ns cnumpy.core._internal
|
_reconstruct
|
qcnumpy
|
ndarray
|
qK
UbRq(KK
cnumpy
|
dtype
|
qUO8K KRq(KU|NNJÿÿÿÿJÿÿÿÿtb]q(}qUaKs}qUbKsetb.rD r# r¦ r" r r r= r= r> Útest_version1_objectÛ s
|
z!TestPickling.test_version1_objectc C s0 d}t jdgddgd}| |¡}t|| d S )NsØ cnumpy.core.multiarray
|
_reconstruct
|
p0
|
(cnumpy
|
ndarray
|
p1
|
(I0
|
tp2
|
S'b'
|
p3
|
tp4
|
Rp5
|
(I1
|
(I1
|
tp6
|
cnumpy
|
dtype
|
p7
|
(S'V6'
|
p8
|
I0
|
I1
|
tp9
|
Rp10
|
(I3
|
S'|'
|
p11
|
N(S'a'
|
p12
|
g3
|
tp13
|
(dp14
|
g12
|
(g7
|
(S'V4'
|
p15
|
I0
|
I1
|
tp16
|
Rp17
|
(I3
|
S'|'
|
p18
|
(g7
|
(S'i1'
|
p19
|
I0
|
I1
|
tp20
|
Rp21
|
(I3
|
S'|'
|
p22
|
NNNI-1
|
I-1
|
I0
|
tp23
|
b(I2
|
I2
|
tp24
|
tp25
|
NNI4
|
I1
|
I0
|
tp26
|
bI0
|
tp27
|
sg3
|
(g7
|
(S'V2'
|
p28
|
I0
|
I1
|
tp29
|
Rp30
|
(I3
|
S'|'
|
p31
|
(g21
|
I2
|
tp32
|
NNI2
|
I1
|
I0
|
tp33
|
bI4
|
tp34
|
sI6
|
I1
|
I0
|
tp35
|
bI00
|
S'\x01\x01\x01\x01\x01\x02'
|
p36
|
tp37
|
b.)r# rÚ )rD rî r )r¦ rî r" r r r r= r= r> Útest_subarray_int_shapeá s
|
z$TestPickling.test_subarray_int_shapec C sZ t jdggdd}|jdd}|j d¡|_|jdd t t |¡¡}t |j|j d S ) Nz 2015-02-24T00:00:00.000000000zdatetime64[ns]r rÍ r& r T©Úinplace)
|
r' ræ r r r r r rs rt r )rF ÚoriginalZoriginal_byte_reversedÚnewr= r= r> Útest_datetime64_byteorderç s z&TestPickling.test_datetime64_byteorderN)rN rO rP r¨ r© rª r rr rs ru rv rw r} r r r r r r r r r r= r= r= r> rr g s( ÿ
|
ÿ
|
rr c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestFancyIndexingc C sl t d¡}d|d d
dgf< t|t dgg¡ t d¡}d|d d
d d
dgf< t|t dggg¡ d S )Nri r r ©r# r# r# ©r' r r ræ r r= r= r> Ú test_listô s
|
|
zTestFancyIndexing.test_listc C sh t d¡}d|d d
df< t|t dgg¡ t d¡}d|d d
d d
df< t|t dggg¡ d S )Nri r rÓ r r r r= r= r> Ú
|
test_tupleý s
|
|
z TestFancyIndexing.test_tuplec C s@ t ddddg¡}t ddddgt¡}t|| t dg¡ d S )Nr# r" rÌ r r ©r' ræ r r ©rF rá rÍ r= r= r> Ú test_mask s zTestFancyIndexing.test_maskc C s¾ t ddddgddddgg¡}t d dgt¡}t d dd d gdd d d ggt¡}t d dd d gd d d d ggt¡}t|| t ddddgg¡ t|| t ddg¡ t|| t dg¡ d S )
|
Nr# r" rÌ r rH rÄ rì r r r )rF rá rÍ Úm2Úm3r= r= r> Ú
|
test_mask2
|
s z TestFancyIndexing.test_mask2c C sJ t ddddg¡}t ddddgt¡}d||< t|t ddddg¡ d S )Nr# r" rÌ r r rH r r r= r= r> Útest_assign_mask s z"TestFancyIndexing.test_assign_maskc C s t ddddgddddgg¡}t d dgt¡}t d dd d gdd d d ggt¡}t d dd d gd d d d ggt¡}| ¡ }d
|
||< t|t ddddgd
|
d
|
d
|
d
|
gg¡ | ¡ }d
|
||< t|t dd
|
ddgd
|
dddgg¡ | ¡ }d
|
||< t|t dd
|
ddgddddgg¡ d S ) Nr# r" rÌ r rH rÄ rì r r rB )r' ræ r r r )rF ZxorigrÍ r r rá r= r= r> Útest_assign_mask2 s $$z#TestFancyIndexing.test_assign_mask2N) rN rO rP r r r r r r r= r= r= r> r ó s r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestStringComparec sØ t dddg¡ t dddg¡t k fdddD t k fdddD t k fd ddD t k fd
|
ddD t k fd ddD t k fd ddD d S ) NÚThisÚisÚexampleÚwasc s g | ]} | | kqS r= r= rW ©Úg1Úg2r= r> rY - s z1TestStringCompare.test_string.<locals>.<listcomp>©r r# r" c s g | ]} | | kqS r= r= rW r¤ r= r> rY . s c s g | ]} | | kqS r= r= rW r¤ r= r> rY / s c s g | ]} | | kqS r= r= rW r¤ r= r> rY 0 s c s g | ]} | | k qS r= r= rW r¤ r= r> rY 1 s c s g | ]} | | kqS r= r= rW r¤ r= r> rY 2 s r rE r= r¤ r> Ú test_string* s z TestStringCompare.test_stringc s t ddddg¡}d t| k fdd|D t| k fdd|D t| k fdd|D t| k fd d|D t| k fd
|
d|D t| k fd d|D d S ) NÚspamZspaZspammerzand eggsc s g | ] }| kqS r= r= rZ ©r¦ r= r> rY 7 s z0TestStringCompare.test_mixed.<locals>.<listcomp>c s g | ] }| kqS r= r= rZ rª r= r> rY 8 s c s g | ] }| k qS r= r= rZ rª r= r> rY 9 s c s g | ] }| kqS r= r= rZ rª r= r> rY : s c s g | ] }| kqS r= r= rZ rª r= r> rY ; s c s g | ] }| kqS r= r= rZ rª r= r> rY < s r )rF r¥ r= rª r> Ú
|
test_mixed4 s z TestStringCompare.test_mixedc sØ t dddg¡ t dddg¡t k fdddD t k fdddD t k fd ddD t k fd
|
ddD t k fd ddD t k fd ddD d S ) Nr r¡ r¢ r£ c s g | ]} | | kqS r= r= rW r¤ r= r> rY A s z2TestStringCompare.test_unicode.<locals>.<listcomp>r§ c s g | ]} | | kqS r= r= rW r¤ r= r> rY B s c s g | ]} | | kqS r= r= rW r¤ r= r> rY C s c s g | ]} | | kqS r= r= rW r¤ r= r> rY D s c s g | ]} | | k qS r= r= rW r¤ r= r> rY E s c s g | ]} | | kqS r= r= rW r¤ r= r> rY F s r rE r= r¤ r> Ú test_unicode> s z TestStringCompare.test_unicodeN)rN rO rP r¨ r« r¬ r= r= r= r> r ) s
|
|
r c @ sD e Zd Zddddddddd d
|
d d d dgZej dejdd eD ¡ej de j
|
e j g¡dd Z ej dddg¡dd Z ej dddg¡dd Zej d d d g¡ej dddg¡d d! Zej d"e jdd#d$fe jdd%d&fg¡d'd( Zej d)de j
|
fde j fg¡d*d+ Zejjd,d-ej d.d/d0g¡d1d2 Zd3S )4ÚTestArgmaxArgminCommonr= 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° )rM ©r z
|
size, axisc s8 g | ]0 fd dt tt t dg D qS )c s g | ] } |fqS r= r= )rX r¨ ©r9 r= r> rY P s ÿz,TestArgmaxArgminCommon.<listcomp>.<listcomp>N)rÎ rq r )rX r= r² r> rY P s þ
|
ÿz!TestArgmaxArgminCommon.<listcomp>r c C s t jj|d}|d kr.dd tt|D }nt|}d||< t|}|||d}| |¡}|||dd}t|| t |j
|
|k t j |j
|
|j d} |||| dd }
|
t |
|
| k t|| t|d
|
kr"t|} |d k râd | |< nd | d
|
< t j | |j d} t t¡ ||j|| dd W 5 Q R X |d krDd d tt|D }nt|d d d
}d||< t|}||j|d}| |¡}||j|dd}t|| t |j
|
|k t j |d d d
|j d} | j} ||j|| dd }
|
t |
|
| k t|| t|d
|
kr&t t¡ ||d
|
|| dd W 5 Q R X t|d
|
krt|} |d k rPd | |< nd | d
|
< t j | |j d} t t¡ ||j|| dd W 5 Q R X d S )Nr² c S s g | ]}d qS rh r= ©rX rà r= r= r> rY Z s zITestArgmaxArgminCommon.test_np_argmin_argmax_keepdims.<locals>.<listcomp>r# rª T©r¨ Úkeepdimsr )r¨ rµ rµ r r" c S s g | ]}d qS rh r= r³ r= r= r> rY x s r% )r' r Únormalrq r rÎ r rÆ r r r7 r2 r r¨ r r) r ) rF r9 r¨ r rS Z new_shapeZ _res_origZres_origrL ZoutarrayZres1Z wrong_shapeZwrong_outarrayr= r= r> Ú test_np_argmin_argmax_keepdimsP s|
|
|
ÿ
|
|
ÿ
|
|
|
|
ÿ
|
|
ÿ
|
|
ÿz5TestArgmaxArgminCommon.test_np_argmin_argmax_keepdimsr» Úminc C s t j ddd¡}t|d| }t||}t|jD ]P}||}||}tt|j}| |¡ tt ||j
|
|j |f| k¡ q2d S )Nr r# )r rH rÄ rì r Úarg) r' r r¶ r~ rq rÏ rÎ Úremover r¬ r» rê ) rF r rD Ú
|
arg_methodZ
|
val_methodr¿ Za_maxminZ aarg_maxminÚaxesr= r= r> Útest_all s
|
|
ÿz TestArgmaxArgminCommon.test_allÚargmaxÚargminc C s t d¡}t||}t jdt jd}tt|d| t jdt jd}tt|d| t jdt jd}tt|d| t jdt jd}|d|d t||d d S ) N)rB rH r r r% r{ ©r# rB rB r· )r' r r~ rØ r r) r )rF r rD r» rµ r= r= r> Útest_output_shape© s
|
|
z(TestArgmaxArgminCommon.test_output_shaperÏ r r# c C sL t dd| ¡}t||}t jd| t jd}|d|d}||ksHtd S )N©r r± r r ©r¨ rµ )r' r r~ r2 r r~ )rF rÏ r rD r» rµ r¾ r= r= r> Útest_ret_is_out½ s
|
|
z&TestArgmaxArgminCommon.test_ret_is_outznp_array, method, idx, vali6 Úasiq Ú0c C s. |ddd}t ||}|||< t| | d S )Ni z<U9r )r~ r )rF Znp_arrayr r rp rÕ r» r= r= r> r¬ Æ s
|
z#TestArgmaxArgminCommon.test_unicodezarr_method, np_methodc C s t jjdd}t||}t jdtd}t jdtd}t|d|||d| t|| t jdtd}t jdtd}t||dd|||dd t|| d S ) Nrä r² r" r r# rÌ r )rµ r¨ )r' r r¶ r~ r* rÝ r )rF r# Z np_methodrD r» Úout1Úout2r= r= r> Útest_np_vs_ndarrayÏ s
|
|
ÿz)TestArgmaxArgminCommon.test_np_vs_ndarrayzreplaces None with NULL.rn z method, vals)r¾ )rB rÍ )r¿ )rÍ rB c C sl t jddd}t||}t |jjd|j¡ t| d |d |d< t| d |d |d< t| d d S )Nr r r r rÌ r# )r' r2 r~ r* r+ r$ rÐ r )rF r r_ rD r» r= r= r> Útest_object_with_NULLså s
|
z-TestArgmaxArgminCommon.test_object_with_NULLsN)rN rO rP Úsizesr¨ r© r« Ú itertoolsÚchainr' r¾ r¿ r· r½ rÁ rÄ r* r r¬ rÉ Úleaks_referencesrÊ r= r= r= r> r H sV üþ H
|
|
ÿÿ
|
ÿÿ
|
ÿÿr c @ s e Zd ZdddddddddddddddgdfddddddddgdfddddddddgdfddddddddgdfgZeddddd d
|
d d gdfddddd d d
|
d gdfg Zd d e eejej ej
|
ej f¡D Z e dd e eej ejejejejejf¡D Z e dd e ddddejgdfdddejdgdfejddddgdfejdejddgdfdgd ejg dfdgd ejg dfdgd ejg dfdgd ejg dfdgd ejg dff ejejf¡D Z e ddddedejgdfddddeejdgdfdddeejddgdfdddedejdgdfedejddddgdfeejejddddgdfeejdeejdeejdgdfeejejeejdeejdgdfeejdeejdeejejgdfeddeddeddgdfeddeddeddgdfeddeddeddgdfe d¡e d¡e d¡e d¡e d¡e d¡gdfe d ¡e d ¡e d ¡e d ¡e d ¡e d!¡gdfe d"¡e d"¡e d ¡e d"¡e d#¡e d$¡gdfe d%¡e d&¡e d"¡e d'¡e d(¡e d)¡gdfe dd*¡e dd*¡e d"d*¡e dd*¡gdfe d"d*¡gd dfedd+d,edd-d,ed d.d,gdfedd/d,ed0dd,edd+d,gdfed0d1d,ed0dd,ed0d/d,gdfd2d2d2d2d3gdfd2d2d2d3d2gdfd3d2d2d2d2gdfd3d2d3d2d2gdfg Zej d4e¡d5d6 Z d7d8 Z d9S ):Ú
|
TestArgmaxr# r rÌ r" r rH rÄ rì rÖ rå rã r% c C s* g | ]"\}}t j|d |d|d fqS ©r r r# r& ©rX rÕ r÷ r= r= r> rY ÿ s zTestArgmax.<listcomp>c C s* g | ]"\}}t j|d |d|d fqS rÐ r& rÑ r= r= r> rY s c C s* g | ]"\}}t j|d |d|d fqS rÐ r& rÑ r= r= r> rY s rí rz é' éO é ú1923-04-14T12:43:12ú1994-06-21T14:43:15ú2001-10-15T04:10:32ú1995-11-25T16:02:16ú2005-01-04T03:14:12ú2041-12-03T14:05:03ú1935-09-14T04:40:11ú1949-10-12T12:32:11ú2010-01-03T05:14:12z2015-11-20T12:20:59z1932-09-23T10:10:13z2014-10-10T03:50:30ÚNaTú2015-09-23T10:10:13ú1932-10-10T03:50:30ú2059-03-14T12:43:12ú1996-09-21T14:43:15ú2022-12-25T16:02:16ú1963-10-04T03:14:12ú2013-05-08T18:15:23r¾ r2 ©ÚdaysÚsecondsé# é rú rB rÄ FTr$ c C s |\}}t }| td¡ t |¡}W 5 Q R X tt |¡|d| d t|t |¡ |d| d t |d¡}|d }tt |¡|d| d t|t |¡ |d| d t t |¡d¡}t ||f¡}|}tt |¡|d| d t|t |¡ |d| d d S ©Nz#invalid value encountered in reducer{ r± é i )
|
r ÚfilterÚRuntimeWarningr' r» r r¾ rÌ r¸ rA ) rF r$ rS ÚposÚsuprp ÚrarrÚrposÚpaddr= r= r> Útest_combinationsU s" ÿ z TestArgmax.test_combinationsc C s t jdddgt jd}tt |¡d | d¡ tt |¡d t jdddgt jd}tt |¡d | d¡ tt |¡d t jddd gt jd}tt |¡d | d¡ tt |¡d t jdd
|
d gt jd}tt |¡d | d¡ tt |¡d d S ) Nr# é éÿÿÿr rì éÿ é ÿÿr é r ìûÿÿÿ ) r' ræ r³ r r¾ rÌ r´ r¶ r¸ r¡ r= r= r> Ú test_maximum_signed_integersm s
|
|
|
|
z'TestArgmax.test_maximum_signed_integersN) rN rO rP Úusg_dataÚsg_datarÌ Úproductr' r3 rµ r· r¹ Údarrr³ r´ r¶ r¸ rÅ rÈ rÚ r¹ Ú
|
datetime64Ú timedelta64r Únan_arrr¨ r© r« rô rû r= r= r= r> rÏ ô sø $üþ ÿÿ ÿÿõ ÿôÿ$&& ûûûûûûûû
|
|
|
|
ýý
|
ÿÿ
|
ÿÿ
|
ÿÿÈ;
|
rÏ c @ s e Zd Zdddddddddddddddgdfddddddddgdfdddddddd gdfd dddddddgd fgZeddddd
|
d d d gdfddddd d d d
|
gd fg Zdd e eejej ej
|
ej f¡D Z e dd e eej ejejejejejf¡D Z e dd e ddddejgdfdddejdgdfejddddgdfejdejddgdfdgd ejg dfdgd ejg dfdgd ejg dfdgd ejg dfdgd ejg dff ejejf¡D Z e ddddedejgdfddddeejdgdfdddeejddgdfdddedejdgdfedejddddgdfeejejddddgdfeejdeejdeejdgdfeejejeejdeejdgdfeejdeejdeejejgdfeddeddeddgdfeddeddeddgdfeddeddeddgdfe d¡e d¡e d¡e d¡e d¡e d ¡gdfe d ¡e d ¡e d ¡e d ¡e d!¡e d"¡gdfe d#¡e d#¡e d ¡e d#¡e d!¡e d"¡gdfe d$¡e d%¡e d#¡e d&¡e d'¡e d(¡gdfe dd)¡e dd)¡e d#d)¡e dd)¡gdfe d#d)¡gd dfedd*d+edd,d+ed d-d+gdfedd.d+ed/dd+edd*d+gdfed/d0d+ed/dd+ed/d.d+gdfd1d1d1d1d2gdfd1d1d1d2d1gdfd2d1d1d1d1gdfd2d1d2d1d1gdfg Zej d3e¡d4d5 Z d6d7 Z d8S )9Ú
|
TestArgminr# r r rÌ r" r rH rÄ rì rÖ rå rã r% c C s* g | ]"\}}t j|d |d|d fqS rÐ r& rÑ r= r= r> rY s zTestArgmin.<listcomp>c C s* g | ]"\}}t j|d |d|d fqS rÐ r& rÑ r= r= r> rY s c C s* g | ]"\}}t j|d |d|d fqS rÐ r& rÑ r= r= r> rY s rí rz rÒ rÓ rÔ rÕ rÖ r× rØ rÙ rÚ rÛ rÜ rÝ z2014-11-20T12:20:59rß rà rÞ rá râ rã rä rå r¾ r2 ræ ré rê rú rB rÄ TFr$ c C s |\}}t }| td¡ t |¡}W 5 Q R X tt |¡|d| d t|t |¡ |d| d t |d¡}|d }tt |¡|d| d t|t |¡ |d| d t t |¡d¡}t ||f¡}|}tt |¡|d| d t|t |¡ |d| d d S rë )
|
r rí rî r' r¸ r r¿ rÌ r» rA ) rF r$ rS rï rð Zmin_valrñ rò ró r= r= r> rô ä s" ÿ z TestArgmin.test_combinationsc C s t jddddgt jd}tt |¡d | d¡ tt |¡d t jdddd gt jd}tt |¡d | d¡ tt |¡d t jdd
|
d d gt jd}tt |¡d | d¡ tt |¡d t jdd ddgt jd}tt |¡d | d¡ tt |¡d d S )Nr# rö iÿÿÿrõ r rì rø iÿÿr÷ rù i r rú lûÿÿÿÿÿÿÿ r ) r' ræ r³ r r¿ rÌ r´ r¶ r¸ r¡ r= r= r> Ú test_minimum_signed_integersû s
|
|
|
|
z'TestArgmin.test_minimum_signed_integersN) rN rO rP rü rý rÌ rþ r' r3 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 sø $üþ ÿÿ ÿÿõ ÿôÿ$&& ûûûûûûûû
|
|
|
|
ýý
|
ÿÿ
|
ÿÿ
|
ÿÿÈ;
|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú
|
TestMinMaxc C sx t tjtjdd t tjtjdd ttjdddd ttjdddd ttjdd dd ttjdd dd d S )Nr# r rª )r r' r ÚamaxÚaminr rE r= r= r> Ú test_scalar s zTestMinMax.test_scalarc C s8 t tjtjdddgd ttjdddggddd d S )Nr# r" rÌ rp rª )r r' r r r rE r= r= r> Ú test_axis s zTestMinMax.test_axisc C sv dD ]l}t d¡ |¡}tt |¡|d tt |¡|d d|d< tt |¡|d tt |¡|d qd S )N)zm8[s]zm8[Y]rB r rí rÞ rÌ )r' rC rE r r r )rF r rD r= r= r> Ú test_datetime s zTestMinMax.test_datetimeN)rN rO rP r r r
|
r= r= r= r> r s r c @ s e Zd Zdd ZdS )Ú TestNewaxisc C s< t dddg¡}d|d d
t jf }t| ¡ d| d S )Nr g¹¿g¹?éú )r' ræ rk r
|
r )rF ÚskrL r= r= r> Ú
|
test_basic, s zTestNewaxis.test_basicN©rN rO rP r r= r= r= r> r + s r c @ s> e Zd Zdd ZdddZdd Zd d
|
Zd d Zd d ZdS )ÚTestClipc C s( t t ||k¡ t t ||k¡ d S rR )r r' r¬ )rF rá ZcminZcmaxr= r= r> Ú _check_range3 s zTestClip._check_rangeFNc C sÌ |d kr |}|d kr|}t j| D ]¤}tjdkr:ddg} nddg} | D ]~}
|
t |¡ |
|
¡} t j d¡| | ¡} |r| j||| dd n| ||¡} d}
|
| jjdkr¨d}
|
t | jj|
|
|
|
| ||¡ qFq"| S ) Nr8 ú=ú>ú<rp r r ú|) r' Úsctypesr; r< r r r rE Úclipr r ) rF Z
|
type_groupZ array_maxZclip_minZclip_maxr Ú expected_minZ expected_maxr Z byte_ordersr< r rá r= r= r> Ú
|
_clip_type7 s(
|
|
zTestClip._clip_typec C s dD ]~}| j dddd|d | j dddd|d | j ddd d
|
|d | j dddd|d | j d ddd|d | j d dd d
|
|dd qd S ) N)FTrÿ é g)ÀgÍÌÌÌÌ Y@r r rÝ iÿÿÿrµ Zuint)r r )r )rF r r= r= r> r Y sX ÿ ÿ ÿ ÿ ÿ ÿzTestClip.test_basicc C s: t jddgdddgd}|d dd ¡}| |dd ¡ d S )
|
N©rÕ r rà ©rD rò rà ©rá ró )rø ró ©Úzró r rá g333333Ó¿rö )r' ræ r r )rF rð rø r= r= r> Útest_record_arrayj s
|
|
ÿzTestClip.test_record_arrayc
|
C sv t ddddddddg¡}| d¡}tt |dk¡ |jdd }tt |dk¡ |jdd
|
}tt |dk¡ d S ) Nr r# r" rÌ r rH rÄ rì )r¸ )r» )r' ræ r r r¬ )rF rp rá r= r= r> Útest_max_or_minp s
|
zTestClip.test_max_or_minc C sN t dt jdddt jg¡}| dd¡}t dt jdddt jg¡}t|| d S ) Nç Àrö rà r÷ r% r# ç ð¿rý )r' ræ rÚ r r )rF Z input_arrr râ r= r= r> Útest_nany s zTestClip.test_nan)FNN) rN rO rP r r r r r! r$ r= r= r= r> r 2 s þ
|
" r c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestCompressc C sx dddddgg}t d¡ dd¡}t jdd g|dd
|
}t|| d d gddgg}t jdd dd dg|d d
|
}t|| d S r± r² r´ r= r= r> r s
|
zTestCompress.test_axisc C s@ dgdgg}t d¡ dd¡}t jddg|dd}t|| d S )Nr# rÄ rB r" rH r rª r² r´ r= r= r> Ú test_truncate s zTestCompress.test_truncatec C s0 t d¡ dd¡}t ddg|¡}t|d d S )NrB r" rH r r# r² ©rF rS rµ r= r= r> r s zTestCompress.test_flattenN)rN rO rP r r& r r= r= r= r> r% s
|
r% c @ sR e Zd Zdd Zdd Zdd Zej dd¡d d
|
Z d d Z
|
d d Z dd Z dS )Ú TestPutmaskc C s( t |||¡ t|| t ||¡ d S rR )r' Úputmaskr ræ )rF rá r Úmaskrp r= r= r> Ú tst_basic s zTestPutmask.tst_basicc C sÆ t ttjg}tj d¡d }|dk }dD ]}tj ¡ D ]X}|D ]N}||kr>|dk rrt |¡jdkrrt |¡j
|
d }| | ¡ |¡|||¡ q>q6t d¡}| | |¡|j|| |¡d d
¡ q(d S )
|
Nrp rµ rc )éÿÿÿr ru r rA éc ÚS3rÌ )rw rÚ r' rr r r rj r r} r r» r+ r rE rY )rF Úunchecked_typesrá r* rp ri r r© r= r= r> Ú test_ip_types s
|
zTestPutmask.test_ip_typesc C s$ t ttjt dddg¡dgd d S )Nr# r" rÌ TrH )r r) r' r) ræ rE r= r= r> Útest_mask_size® s zTestPutmask.test_mask_sizer ©rN rM c C s: t dddg|¡}t |dddgd¡ t|dddg d S )Nr# r" rÌ TFr% ©r' ræ r) r ©rF r rá r= r= r> Útest_byteorder± s zTestPutmask.test_byteorderc C s´ t jddgdddgd}t |d dd gd
|
¡ t|d d
|
d g t|d d dg t|d ddg t |d dd gd¡ t|d d
|
d g t|d ddg t|d ddg d S )Nr r r ©rø rý r r rá TFrB rH rø r" r r rÌ r r3 )rF rð r= r= r> r · s
|
ÿz TestPutmask.test_record_arrayc C s t ddddg¡}t |dd
dddg|d d
¡ t|t ddddg¡ t ddddg¡}t |dd
|d d
dddg¡ t|t ddddg¡ d S )NTFr# r rÌ )r' ræ r) r r r= r= r> Ú test_overlapsÄ s $$zTestPutmask.test_overlapsc C s> t d¡}d|j_t t¡ t ||dkd¡ W 5 Q R X d S )NrH Fr" rÌ )r' rC rJ rK r¨ r r) r) r¡ r= r= r> rL Î s
|
zTestPutmask.test_writeableN) rN rO rP r+ r0 r1 r¨ r© r« r5 r r7 rL r= r= r= r> r( s
|
|
r( c @ sv e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d ¡d d Z dd Z dd Z ej
|
ddddg¡dd ZdS )ÚTestTakec C s* t t|jd }t|j|dd| d S )Nr rª )rÎ rq r7 r Útake)rF rá rÆ r= r= r> r+ × s zTestTake.tst_basicc C sl t ttjg}tj d¡d }d|_tj ¡ D ]:}|D ] }||kr4| | ¡
|
|¡¡ q4| |
|
d¡¡ q,d S )NrÄ rµ r¯ r. ) rw rÚ r' rr r r7 r rj r+ r rE )rF r/ rá ri r r= r= r> r0 Û s zTestTake.test_ip_typesc C sd t j d¡d }d|_tt|jdddgdd tt|jdgdd t|jd gddd |d d S )
|
NrÄ rµ r¯ r r# r" rª rå r% )r' r r7 r r] r9 r r r= r= r> Ú
|
test_raiseè s
|
zTestTake.test_raisec C sZ t j d¡d }d|_t|jdgdddd |d t|jdgdddd |d d S )
|
NrÄ rµ r¯ r% r r ©r¨ rº r" r# ©r' r r7 r r9 r r= r= r> Ú test_clipï s zTestTake.test_clipc C sz t j d¡d }d|_t|jdgdddd |d t|jd gdddd |d t|jd
|
gdddd |d d S ) NrÄ rµ r¯ r% r r¸ r; r# r" rÌ r< r r= r= r> Ú test_wrapõ s
|
zTestTake.test_wrapr r2 c C s2 t dddg|¡}t| dddg¡dddg d S )Nr# r" rÌ r )r' ræ r r9 r4 r= r= r> r5 ü s zTestTake.test_byteorderc C sD t jddgdddgd}| dg¡}t|d d ko<|d
|
d k d S ) Nr r r r6 r r r# rá rD rø rò )r' ræ r9 r )rF rð Zrec1r= r= r> r s
|
|
ÿ zTestTake.test_record_arrayc C sD t d¡}t j|dddg|dd
dd}t|t dddg¡ d S )NrH r# r" rÌ r¸ r¹ )r' rC r9 r ræ rs r= r= r> Útest_out_overlap s
|
zTestTake.test_out_overlapr7 rÚ rh r= c C sJ t d¡}t j|t jd}t j||jd}t j|||d}||ksFtd S )NrH r r· )r' rC r* r r r9 r~ )rF r7 rá Úindsrµ r¾ r= r= r> rÄ s
|
|
zTestTake.test_ret_is_outN)rN rO rP r+ r0 r: r= r> r¨ r© r« r5 r r? rÄ r= r= r= r> r8 Ö s
|
r8 c @ sn e Zd Zej dejejej ej
|
ej ej ej ejejejejg ¡dd Zdd Zdd Zdd Zd
|
d Zd S ) Ú TestLexsortr c C sz t jddddddg|d}t jddddddg|d}t ||f¡}t ddddddg¡}t|| t|| t |¡ d S ) Nr# r" rÌ rH r r r rÄ )r' ræ Úlexsortr r )rF r rD r¦ r Ú expected_idxr= r= r> r s
|
zTestLexsort.test_basicc C sb t ddddddg¡}t jddddddgdd }t ||f¡}t ddddddg¡}t|| d S )
|
Nr# r" rÌ rH r r rÄ r r ©r' ræ rB r ©rF rD r¦ r rC r= r= r> r« & s
|
zTestLexsort.test_mixedc C s¤ t jdddgdd}t jdddgdd}t ||f¡}t dddg¡}t|| t jdddgdd}t jdddgdd}t ||f¡}t dddg¡}t|| d S )Nr r r r" r# r rD rE r= r= r> r
|
. s
|
zTestLexsort.test_datetimec C sò t j dd¡}t j ddddddgd¡}||fD ].}t | d ¡f¡}t j|d
|
d }t|| q0||f||ffD ]|\}}t ||f¡}t|t | d ¡|f¡ t|t || d ¡f¡ t j|d d t j|d d }}t|t ||f¡ qpd S )NrB rp ÚabcZxyZwzZefghiZqwstrá r Z mergesortrÜ r( r )r' r ÚchoicerB rE r r ræ )rF rD r¦ rA rG rJ rx r r= r= r> Ú test_object; s zTestLexsort.test_objectc C s. t ddd¡ dd¡}tt jt j|dd d S )Nrü rý é~ r_ rÌ r" rª )r' rd rÆ r r rB r r= r= r> Útest_invalid_axisK s z TestLexsort.test_invalid_axisN)rN rO rP r¨ r© r« r' r3 rµ r· r¹ r³ r´ r¶ r¸ rð rÅ rÈ r r« r
|
rH rJ r= r= r= r> rA s$ ý
|
rA c @ sÀ e Zd ZdZe ¡ dd Zejddgddd Zd d
|
Zd d Z d d Z
|
dd Z dd Z dd Z dd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Z ej j e d3d4d5d6 Z d7d8 Z!ejd9d:gdd;d< Z"d=d> Z#d?d@ Z$dAdB Z%dCdD Z&ej j'dEdF Z(dGdH Z)dIdJ Z*dKdL Z+dMdN Z,dOdP Z-dQdR Z.dSdT Z/dUdV Z0dWdX Z1dYdZ Z2d[d\ Z3d]d^ Z4d_d` Z5dadb Z6dcdd Z7deS )fÚTestIOz.Test tofile, fromfile, tobytes, and fromstringc C sN d}t jj}|||| t¡d }t jt jt j t jg|dd d
df< |S )N)r" r rÌ rw r r# )r' r rE r¹ rÚ Úinf)rF r7 r rá r= r= r> rá R s
|
$zTestIO.xÚstringZpath_obj)Úparamsc c s$ |d }|j dkrt|}|V d S )NÚfilerM )rå rÚ )rF Ztmp_pathÚrequestÚfilenamer= r= r> Ú tmp_filenameZ s
|
zTestIO.tmp_filenamec s> t ¡ tttj tjd t d¡}tt fdd| d S )Nr¸ rì c s
|
| ¡S rR ©Útofilerà ©r¦ r= r> r4 j r5 z$TestIO.test_nofile.<locals>.<lambda>)r# r$ r ÚOSErrorr' Úfromfiler3 r r! r= rU r> Ú test_nofiled s
|
zTestIO.test_nofilec C s8 t jddddgt jd}t jddt jd}t|| d S )NTFr z 1 0 -2.3 0.0ú ©Úsepr )r' ræ rP rg r )rF rx rø r= r= r> Útest_bool_fromstringl s zTestIO.test_bool_fromstringc C s4 t jdt jdd}t jddgt jd}t|| d S )Nz#9923372036854775807 104783749223640rY ©r r[ l ÿ *·M ì ØM<3} r )r' rg r¹ ræ r r r= r= r> Útest_uint64_fromstringq s ÿz TestIO.test_uint64_fromstringc C s4 t jdt jdd}t jddgt jd}t|| d S )Nz -25041670086757 104783749223640rY r] lýÿÿÿe ³o[r^ r )r' rg r¸ ræ r r r= r= r> Útest_int64_fromstringw s ÿz TestIO.test_int64_fromstringc C s& t jddt jdd}|jdks"td S )Nz1,2ú,r )r[ r ÚcountrÓ )r' rg r¸ r7 r~ r! r= r= r> Útest_fromstring_count0} s z TestIO.test_fromstring_count0c C s4 t |d}W 5 Q R X t |¡}t|jdkd d S )Nr r úArray not empty©Úopenr' rW r r9 ©rF rR r× rø r= r= r> Útest_empty_files_text s
|
|
z TestIO.test_empty_files_textc C s8 t |d}W 5 Q R X tj|dd}t|jdkd d S )NÚwbrY ©r[ r rd re rg r= r= r> Útest_empty_files_binary s
|
z TestIO.test_empty_files_binaryc C sV t |d}| |¡ W 5 Q R X t |d}tj||jd}W 5 Q R X t||j d S )Nri Úrbr )rf rT r' rW r r rb ©rF rá rR r× rø r= r= r> Útest_roundtrip_file s
|
zTestIO.test_roundtrip_filec C s* | |¡ tj||jd}t||j d S r )rT r' rW r r rb )rF rá rR rø r= r= r> r} s
|
zTestIO.test_roundtripc C s0 t |¡}| |¡ tj|dd}t|| d S )NT)Z allow_pickle)ÚpathlibÚPathÚdumpr' r' r )rF rá rR r rø r= r= r> Útest_roundtrip_dump_pathlib s
|
|
z"TestIO.test_roundtrip_dump_pathlibc C sR | ¡ }tj||jd}t||j | d¡}tj||jd}t|| d¡ d S )Nr r )rá r' rc r r rb r ©rF rá r¾ rø r= r= r> Útest_roundtrip_binary_str s
|
z TestIO.test_roundtrip_binary_strc C sd |j ¡ }d tt|¡}tj|dd}t |¡ }t|| || t || || dd d S )Nú@rj rH ©Údecimal)
|
rx r rY r rÚ r' rg Úisfiniter r )rF rá r¾ rø Znan_maskr= r= r> Útest_roundtrip_str© s
|
zTestIO.test_roundtrip_strc C s6 |j ¡ }d tt|¡}tj|dd}t|| d S )Nru rj )rx r rY r r6 r' rg r rs r= r= r> Útest_roundtrip_repr² s
|
zTestIO.test_roundtrip_reprc C sR | |¡ dd }tj|ddd&}||_||_tttj||j d W 5 Q R X d S )Nc _ s t dd S )NzCan not tell or seek)rV r r= r= r> Úfail¼ s z-TestIO.test_unseekable_fromfile.<locals>.failrl r ©Ú bufferingr )
|
rT r# rf r& Útellr rV r' rW r )rF rá rR r{ r× r= r= r> Útest_unseekable_fromfile¸ s
|
z TestIO.test_unseekable_fromfilec C sF | |¡ tj|ddd"}tj||jd}t||j W 5 Q R X d S )Nrl r r| r ©rT r# rf r' rW r r rb rm r= r= r> Ú test_io_open_unbuffered_fromfileÄ s
|
z'TestIO.test_io_open_unbuffered_fromfilec C sö t d¡}| |¡ ttj |¡|j t|t |¡ t
|
|d4}| |j¡ | |¡ ttj |¡|jd W 5 Q R X t
|
|d ¡ t
|
|d}| |¡ W 5 Q R X t|t |¡ t
|
|d}| |¡ W 5 Q R X ttj |¡|jd d S )Ni @ úr+br" r rD ) r' r* rT r ÚosÚpathÚgetsizerÐ r rW rf r& Úclose)rF rR rÕ r× r= r= r> Útest_largish_fileË s
|
|
|
" zTestIO.test_largish_filec C sF | |¡ tj|ddd}tj||jd}W 5 Q R X t||j d S )Nrl r% r| r r rm r= r= r> Ú test_io_open_buffered_fromfileß s
|
z%TestIO.test_io_open_buffered_fromfilec C s² t jd t jt jd g}|D ]}t|d }| |d ¡ | d¡ W 5 Q R X dD ]X}d||f }t||*}| d¡ tj|tjdd | ¡ }W 5 Q R X t
|
|d |d
|
qRq d S ) Nr ri r# ó )rl r z%d %sr" ©r rb rB r± ) r# ÚDEFAULT_BUFFER_SIZErf r& r] Úreadr' rW rÈ r~ r )rF rR rË r9 r× rº r² rï r= r= r> Ú!test_file_position_after_fromfileæ s þ
|
z(TestIO.test_file_position_after_fromfilec
|
C sú t jd t jt jd g}|D ]Ø}d|f }t|dR}| |d ¡ | d¡ | d¡ | d¡ tjdgtjd |¡ | ¡ }W 5 Q R X t
|
|d
|
|d t|d <}| d ¡ | dd¡ tjdgtjd |¡ | ¡ }W 5 Q R X t
|
|d|d q d S )Nr z%dri r# r rB s 12r r r r± r r" ) r# r rf r& r] r' ræ rÈ rT r~ r r )rF rR rË r9 r² r× rï r= r= r> Ú test_file_position_after_tofileú s( þ
|
|
|
|
|
z&TestIO.test_file_position_after_tofilec C sX t |d}W 5 Q R X t |d}ttdtj|td W 5 Q R X ttdtj|td d S )Nr rl z Cannot read into object arrayr )rf r r) r' rW r( )rF rR r× r= r= r> Ú test_load_object_array_fromfile s
|
ÿ ÿz&TestIO.test_load_object_array_fromfilec
|
C sL t |d}| |¡ W 5 Q R X t |d$}tj||jdd}t||j W 5 Q R X t |d}t|jd }t|jd }|jj| }tj||j||d}t||j|||
|jj}tj||j|d}t||j|| d d
W 5 Q R X t |d}|j|d d
|
W 5 Q R X t |d }t t
|
d tj||jd dd W 5 Q R X d S ) Nri rl r )r r< r r )r rb r< r# ra rj z1'offset' argument only permitted for binary files)r r[ r< ) rf rT r' rW r r rb r r1 r rô ) rF rá rR r× rø Z count_itemsZ offset_itemsZ offset_bytesr r= r= r> Útest_fromfile_offset sD ÿ ÿ& üzTestIO.test_fromfile_offsetz bug in PyPy's PyNumber_AsSsize_trn c C sv dd }dd }t j}zRt|d>}| |¡ |tf|tffD ] \}}|t _t|tj| q>W 5 Q R X W 5 |t _X d S )Nc S s dS )NrF r= ©Úfdr= r= r> Údup_strD s z-TestIO.test_fromfile_bad_dup.<locals>.dup_strc S s dd S )Nr" éD r= r r= r= r> Ú
|
dup_bigintG s z0TestIO.test_fromfile_bad_dup.<locals>.dup_bigintri ) r Úduprf rT rô rV r r' rW ) rF rá rR r r Zold_dupr× r Úexcr= r= r> Útest_fromfile_bad_dupB s
|
z TestIO.test_fromfile_bad_dupc K sl d|krt j|f|}nt j|f|}t|| t|d}| |¡ W 5 Q R X t j|f|}t|| d S )Nr[ ri )r' rc rg r rf r] rW )rF r¾ rú rQ r5 rø r× r= r= r> Ú _check_fromT s
|
zTestIO._check_fromÚperiodÚcommac c sH |j dkrdV n2|j dkr6t dV W 5 Q R X ndsDt|j dS )aÏ
|
Including this fixture in a test will automatically
|
execute it with both types of decimal separator.
|
|
So::
|
|
def test_decimal(decimal_sep_localization):
|
pass
|
|
is equivalent to the following two tests::
|
|
def test_decimal_period_separator():
|
pass
|
|
def test_decimal_comma_separator():
|
with CommaDecimalPointLocale():
|
pass
|
r Nr F)rå r r~ )rF rP r= r= r> Údecimal_sep_localization` s
|
|
z TestIO.decimal_sep_localizationc C s2 | j dtjtjtjtjtjtjtjg|dd d S )Ns3 nan +nan -nan NaN nan(foo) +NaN(BAR) -NAN(q_u_u_x_)rY rj )r r' rÚ ©rF rR r r= r= r> r$ | s üzTestIO.test_nanc C s8 | j dtjtjtj tjtj tjtj g|dd d S )Ns. inf +inf -inf infinity -Infinity iNfInItY -inFrY rj )r r' rL r r= r= r> Útest_inf s $üzTestIO.test_infc C s | j ddddddg|dd d S ) Ns' 1.234 -1.234 .3 .3e55 -123133.1231e+133çX9´Èv¾ó?gX9´Èv¾ó¿g333333Ó?g´. LKR?KgY]xÜrY rj ©r r r= r= r> Ú test_numbers s üzTestIO.test_numbersc C s$ | j dt ddddg¡|dd d S )Ns ? @ @@ @r# r" rÌ r r
|
r )r r' ræ ©rF rR r= r= r> Ú test_binary s üzTestIO.test_binaryc C sà t jdksdt jkrdS zªd}tjdtjd}t|}t ¡ }|j }t
|
|||j ¡ |¡ | d¡ tj|tjd}| ¡ tt||k t|d|
|k ¡ t|| d
|k ¡ W n ttfk
|
rÚ Y nX dS )zÃTest workarounds for 32-bit limit for MSVC fwrite, fseek, and ftell
|
|
These normally would hang doing something like this.
|
See : https://github.com/numpy/numpy/issues/2256
|
Úwin32z[GCC Nl r r r )r; Úplatformro r' rC r³ r ÚtempfileÚNamedTemporaryFilerO r rÐ rT r& rW r r r¬ Ú MemoryErrorr) )rF Z
|
fourgbplusZ testbytesro Zfliker× rD r= r= r> Útest_big_binary s"
|
zTestIO.test_big_binaryc C s | j dddddg|dd d S )Nó 1,2,3,4rý r rà rò ra rj r r¢ r= r= r> r¨ ³ s zTestIO.test_stringc C sV | j dddddg|ddd | j ddddg|d dd | j dddddg|d
|
dd d S ) Nrª rý r rà rò r ra )rb r[ rÌ r% r r r= r= r> Útest_counted_string¶ s*
|
ÿ ÿ
|
ÿzTestIO.test_counted_stringc C s | j dddddg|tdd d S )Nó 1 2 3 4 r# r" rÌ r rY r] ©r rÝ r¢ r= r= r> Útest_string_with_ws¾ s
|
ÿzTestIO.test_string_with_wsc C s | j ddddg|dtdd d S )Nr¬ r# r" rÌ rY )rb r r[ r r¢ r= r= r> Útest_counted_string_with_ws s þz"TestIO.test_counted_string_with_wsc C s: | j dddddg|dd | j dddddg|tdd d S )
|
Ns 1 , 2 , 3 , 4rý r rà rò ra rj rª r] )r rÿ r r= r= r> Ú
|
test_asciiÇ s
|
ÿ
|
ÿzTestIO.test_asciic C s. t t | jdddg|dd W 5 Q R X d S )Ns 1.234 1,234r rý rY rj )r r r r r= r= r> Útest_malformedÍ s
|
ÿzTestIO.test_malformedc C s | j dddddg|dd d S )Ns 1_x_3_x_4_x_5r# rÌ r rH Z_x_rj r r¢ r= r= r> Ú test_long_sepÒ s
|
ÿzTestIO.test_long_sepc C s2 t jddddgt jd}| jd||dt jd d S ) Nr# r" rÌ r r rª ra rZ )r' ræ rØ r )rF rR rx r= r= r> Ú
|
test_dtypeÖ s zTestIO.test_dtypec C sj t jddddgt jd}d}t|d}| |¡ W 5 Q R X t j|dt jd}t|jdk t|| d S ) NTFr s
|
1,0,-2.3,0ri ra rZ rW ) r' ræ rP rf r] rW r r r )rF rR rx r¾ r× rø r= r= r> Útest_dtype_boolÚ s zTestIO.test_dtype_boolc C s t jddddgtd}t|d}|j|dd W 5 Q R X t|d }| ¡ }W 5 Q R X t d
|
d | d¡D ¡}t|| d S ) Nç)\Âõ(ø?r" ç®Gáz @r r r ra rj rê c S s g | ] }t |qS r= )rÿ )rX r r= r= r> rY ë s z*TestIO.test_tofile_sep.<locals>.<listcomp>)r' ræ rÿ rf rT r Úsplitr )rF rR r rá r× r¾ rø r= r= r> Útest_tofile_sepä s zTestIO.test_tofile_sepc C sh t jddddgtd}t|d}|j|ddd W 5 Q R X t|d
|
}| ¡ }W 5 Q R X t|d d S ) Nrµ r" r¶ r r r ra z%.2f)r[ rò rê z1.51,2.00,3.51,4.00)r' ræ rÿ rf rT r r )rF rR r rá r× r¾ r= r= r> Útest_tofile_formatî s zTestIO.test_tofile_formatc sd t jdtdtd tt fdd W 5 Q R X t ¡ ttfdd t ¡ d S )NrB r ri c s j ddS )Nr` rj rS r= )r× rá r= r> r4 ù r5 z,TestIO.test_tofile_cleanup.<locals>.<lambda>c s
|
¡S rR rS r= )rR rá r= r> r4 þ r5 )r' r* r( rf r rV r rº r¢ r= )r× rR rá r> Útest_tofile_cleanupö s
|
zTestIO.test_tofile_cleanupc C sr t jddd ddd¡}| |¡ t j|dd}t|| | ¡ }tt t j |dd}t|| W 5 Q R X d S )NrÄ r¤ r r" rÌ r z(3,4)i4)
|
r' rC rÆ rT rW r rá r r rg )rF rR rá rL Zx_strr= r= r> Ú test_fromfile_subarray_binary s
|
|
|
z$TestIO.test_fromfile_subarray_binaryc C st d}t t¡ tj|ddd}W 5 Q R X t|d}| |¡ W 5 Q R X t t¡ tj|ddd W 5 Q R X d S )Na 12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,12,42,13,z(3,)ira r] r )r¨ r r) r' rg rf r] rW )rF rR r$ râ r× r= r= r> Ú!test_parsing_subarray_unsupported s z(TestIO.test_parsing_subarray_unsupportedc
|
C s t jddd dd¡}| ¡ }t t¡, t t¡ t j |ddd W 5 Q R X W 5 Q R X |
|
|¡ t j |ddd}t || d S ) Niö r¿ r r% rB z(10,)ié' r ) r' rC rÆ rá r¨ r r) Úwarnsr rg rT rW r )rF rR râ ÚbinaryrL r= r= r> Ú%test_read_shorter_than_count_subarray s $
|
z,TestIO.test_read_shorter_than_count_subarrayN)8rN rO rP rä r¨ Zfixturerá rR rX r\ r_ r` rc rh rk rn r} rr rt ry rz 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¼ rÀ r= r= r= r> rK O sh
|
|
"
|
|
|
|
|
rK c @ sz e Zd Zej dddg¡ej deeeg¡dd Z ej de
|
d¡d g¡d
|
d Z d d Z ejjedddd ZdS )ÚTestFromBufferr< r r r c C sH t |¡ |¡}t j d¡d |¡}| ¡ }tt j||d|j d S )N)r rì rH r ) r' r r r rE rá r rc rb )rF r< r r© rá r: r= r= r> r + s zTestFromBuffer.test_basicr! rB s 12345678c C s t |¡}|j|kstd S rR )r' rc rv r~ )rF r! r r= r= r> Útest_array_base3 s
|
z TestFromBuffer.test_array_basec C s t t d¡t g ¡ d S )Nr5 )r r' rc ræ rE r= r= r> Ú
|
test_empty; s zTestFromBuffer.test_emptyzjPyPy's memoryview currently does not track exports. See: https://foss.heptapod.net/pypy/pypy/-/issues/3724rn c
|
C sv t jdd`}| d¡ | ¡ t | ¡ d¡}tj|tjd}t
|
t ¡ | ¡ W 5 Q R X ~| ¡ W 5 Q R X d S )Nri )rº s asdfr r ) r¦ Ú TemporaryFiler] ÚflushÚmmapÚfilenor' rc r3 r¨ r Ú BufferErrorr )rF r) ÚmmrS r= r= r> Útest_mmap_close> s
|
z TestFromBuffer.test_mmap_closeN)rN rO rP r¨ r© r« rÿ rÝ r¹ r r' rC r rà rª r rÊ r= r= r= r> rÁ * s
|
ÿrÁ c @ sN e Zd Zdd Zdd Zdd Zdd Zejj e
|
d d
|
d d Z d d Z dS )ÚTestFlatc C sh t d¡}| dd¡}d|_d|j_|| _|d d d
d d d
f | _|| _|d d d
d d d
f | _ d S )Nró r rH rË Fr" )
|
r' rC rÆ r7 rJ rK rD r¦ rQ rR )rF rQ rD r= r= r> rG P s
|
zTestFlat.setup_methodc C sN d}zd| j jd< W n tk
|
r, d}Y nX t| t| j jd dk d S )NFç Y@r+ Tç (@)rD rb r) r ©rF Z
|
testpassedr= r= r> Útest_contiguousZ s
|
zTestFlat.test_contiguousc C sN d}zd| j jd< W n tk
|
r, d}Y nX t| t| j jd dk d S )NFrÌ r TrÍ )r¦ rb r) r rÎ r= r= r> Útest_discontiguousc s
|
zTestFlat.test_discontiguousc C s´ | j j ¡ }| jj ¡ }| jj ¡ }| jj ¡ }t|jjdk t|jjdk t|jjdk t|jjdk t|jj dk t|jj dk t|jj dk t|jj dk d S )NFT)
|
rD rb r r¦ rQ rR r rJ rK r )rF r³ rÕ rë r× r= r= r> Útest___array__l s zTestFlat.test___array__zPython lacks refcountsrn c
|
C sÀ t d¡t dg| jj ¡t dg¡d g}t t j¡}t |¡}|D ]v}t |¡}tdD ]*}z| jj | W qZ t
|
k
|
r Y qZX qZt t t |¡| dk t t t |¡| dk qDd S )Nr Trµ ro ) r' r ræ rD r9 r r; rÀ rq rb r] r r\ )rF r@ ZindtypeZ
|
rc_indtyperÆ Zrc_indrà r= r= r> Ú test_refcount{ s *
|
|
zTestFlat.test_refcountc C sR t d¡ ddd¡j}t t¡ d|_W 5 Q R X |D ]}q6|j|jj ksNt
|
d S )NrB r" r# rH ) r' rC rÆ rb r¨ r r/ r¾ rv r9 r~ )rF Úitrà r= r= r> Útest_index_getset s zTestFlat.test_index_getsetN) rN rO rP rG rÏ rÐ rÑ r¨ r© rª r rÒ rÔ r= r= r= r> rË O s
|
|
rË c @ sx e Zd Zedd Zdd Zedd Zdd Zd d
|
Zd d Z ed d Z
|
edd Z edd Z dd Z dd ZdS )Ú
|
TestResizec C s t dddgdddgdddgg¡}tr6|jddd n
|
| d¡ t|jd d
t dddgdddgdddgg¡j t|dd
jd d S )Nr# r ©rH rH F©Zrefcheckrí )r' ræ r r r rb r r= r= r> r s "
|
"ÿzTestResize.test_basicc C s: t dddgdddgdddgg¡}|}tt|jd ~d S ©Nr# r )rH r# )r' ræ r r) r rs r= r= r> Útest_check_reference¤ s "z TestResize.test_check_referencec C sH t d¡}tr |jddd n
|
| d¡ t|t d¡dd d
f d S )NrÌ Fr× r )r' r½ r r r r r= r= r> Útest_int_shapeª s
|
|
|
zTestResize.test_int_shapec C s@ t d¡}| d ¡ t|t d¡ | ¡ t|t d¡ d S r¼ )r' r½ r r r r= r= r> Útest_none_shape³ s
|
|
|
zTestResize.test_none_shapec C sj t dD ]\}t d¡}| d¡ t|jd t|jd t d¡}| d¡ t|jd t|jd qd S )NrB rh r= r# )rq r' r2 r r r7 r9 )rF r¿ rá r= r= r> Ú test_0d_shapeº s
|
|
|
|
zTestResize.test_0d_shapec C sX t tt d¡jd t tt d¡jd t tt d¡jdd t tt d¡jdd d S )NrÌ Úhir% r# r& r× )r rô r' r½ r r) rE r= r= r> Útest_invalid_argumentsÆ s z!TestResize.test_invalid_argumentsc C sB t d¡}tr"|jddddd n| ddd¡ t|jdk d S )NrÌ r" r# Fr× rö )r' r½ r r r r7 r r= r= r> Útest_freeform_shapeÌ s
|
|
z TestResize.test_freeform_shapec C s\ t d¡}tr"|jddddd n| ddd¡ t|d t d¡ t|d t d¡ d S )NrÌ r" Fr× r r# r° )r' r½ r r r r* r r= r= r> Útest_zeros_appendedÕ s
|
z TestResize.test_zeros_appendedc C sp t jddtdfgd}tr*|jddd n
|
| d¡ t|jd t|d d d
d
|
t|d d d
d d S ) NrB r r" r ru Fr× )ru rÕ r r# )r' r r( r r r r7 r r¡ r= r= r> Ú test_obj_objß s
|
zTestResize.test_obj_objc C s, t dt¡}|d }| d¡ | d¡ d S )N)rB r .)r rB )r rµ )r' r* rÝ r )rF rá Zx_viewr= r= r> Útest_empty_viewë s
|
zTestResize.test_empty_viewc C s@ t dddgdddgdddgg¡}t |¡}tt|jd ~d S rØ )r' ræ rx ry r r) r )rF rá Zxrefr= r= r> Útest_check_weakrefó s "
|
z TestResize.test_check_weakrefN)rN rO rP r r 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 dd Z
|
dd Z dd Z dd Z dS )Ú
|
TestRecordc C s4 t dtfdtfg¡}ddg|_t|jddg d S )Nr× r¿ r rü )r' r rÿ rÝ r1 r ©rF r© r= r= r> Útest_field_renameû s
|
z TestRecord.test_field_renamec C s dd }t t| d S )Nc S s t dddg¡ d S )N)rÑ r ©r[ r r¨ r= r= r= r> Útest_dtype_init s zGTestRecord.test_multiple_field_name_occurrence.<locals>.test_dtype_init©r r) )rF rè r= r= r> Ú#test_multiple_field_name_occurrence s z.TestRecord.test_multiple_field_name_occurrencec C s t ttjdtfg t ttjdtfg t dtfg¡}t t|jd tjdddg|d}t t|jd |d }t t|jd d S ) Nrj )r¦ rj )rj r¦ rh rr r r r )r rô r' r rÝ r< ræ r] )rF r© rá rø r= r= r> Útest_bytes_fields s z TestRecord.test_bytes_fieldsc C s dd }t t| d S )Nc S s t dddg¡ d S )N)u â¹r rç r¨ r= r= r= r> Útest_dtype_unicode s zGTestRecord.test_multiple_field_name_unicode.<locals>.test_dtype_unicoderé )rF rì r= r= r> Ú test_multiple_field_name_unicode s z+TestRecord.test_multiple_field_name_unicodec C sD t jjjdgdggddd}t|d d d t|d d d d S )Nr r# za,búi4,i4©r1 r2 rD r¦ )r' re rf r: r r r= r= r> Útest_fromarrays_unicode s
|
ÿz"TestRecord.test_fromarrays_unicodec C sH d}t jdddg|tfgd}|j|d t|d t dddg¡ d S )Nr¦ r# rÌ r" r r& )r' ræ rÝ r r )rF r rá r= r= r> Útest_unicode_order% s z TestRecord.test_unicode_orderc C s¾ t jdddddgfgd}tt|jdd tt|jd tt|d jd
|
d tt|d jd
|
| ¡ }td }d||< t|| d td }tt |j|d tt |j| d |d |< t|| d tt |d j|d tt |d j| td}td}d|| |< t|| | d tt || j|d tt || j| td}d||< t|d dg d
|
¡ d t|dd g d
|
¡ d t|d dg d
|
¡ d tt |jdd tt |jd d S )Nrh )rÿ r¤ )Úf2r¤ Zf3)Úsf1r¤ r s f1r# rÿ s sf1z
|
not at allr" r ró rò rÌ rä r® )r" rh u Ï ) r' r* r r] Ú __setitem__r< r rÚ r r) rg )rF rD r¦ Zfn1ZfnnZfn3Zsfn1Zfn2r= r= r> Útest_field_names- s@ þ zTestRecord.test_field_namesc C sÖ t jddgdd}d|j_t jddgddgd}d|j_t jddgdd}d|j_tt|d t|d k tt|d t|d k tt|d t|d k tt|d t|d koÎ|d |d k d S )
|
NrÚ úi1,i2r FrÐ )Znum1rî )Znum2rÜ r r# )r' ræ rJ rK r rº r£ r= r= r> Útest_record_hashV s zTestRecord.test_record_hashc C s& t jddgdd}ttt|d d S )NrÚ rö r r )r' ræ r rô rº r¡ r= r= r> Útest_record_no_hashb s z TestRecord.test_record_no_hashc C s@ t jdgg g g ddd t jdddddgg g g ddd d S )Nr= r+ ©r1 r2 r3 r1 r r& rE r= r= r> Ú test_empty_structure_creationf s ÿ
|
ÿz(TestRecord.test_empty_structure_creationc C sx t jddddgd}|ddg }t|j|k t|jt ddgdd gd
|
d gd ¡k d |d d
< t|d
|
¡ d d S )NrÌ r£ )r¦ rì )r³ rè r rD r³ r¤ rè r r r0 rË )r r# rH )r' r r rv r r rb )rF rD rx r= r= r> Ú test_multifield_indexing_viewm s þ z(TestRecord.test_multifield_indexing_viewN)rN rO rP ræ rê rë rí rð rñ rõ r÷ rø rú rû r= r= r= r> rä ú s ) rä c @ s e Zd Zdd ZdS )ÚTestViewc C sd t jddgdt jfdt jfdt jfdt jfgd}|jdd}| d¡}t|| t|d d
|
g d S ) Nr° )rH rÄ rì r rê rê r¦ rD r rM ii)r' ræ r³ rW r )rF rá rø r r= r= r> r x s
|
ÿÿ
|
|
zTestView.test_basicNr r= r= r= r> rü w s rü c K s | j f |S rR )Úmean©rD r r= r= r> Ú_mean s rÿ c K s | j f |S rR )Úvarrþ r= r= r> Ú_var s r c K s | j f |S rR )Ústdrþ r= r= r> Ú_std s r c @ sÔ e Zd ZeeegZdd Zdd Zdd Z dd Z
|
d d
|
Z d d Z d d Z dd Zdd Zdd Zdd Zdd Zdd Zdd Zej d d ¡d d Zd!d" Zd#d$ Zd%d& Zd'd( Z d)d* Z d+d, Z d-d. Z d/S )0Ú TestStatsc C s^ t j td¡ t j d¡| _| jd| j | _t dd | jjD ¡| _| j dd¡| _d S )NrÌ rË rw c S s g | ]}t t|qS r= )r r6 )rX rê r= r= r> rY s z*TestStats.setup_method.<locals>.<listcomp>r rH )
|
r' r rª rq ÚrmatÚcmatræ rb ÚomatrÆ rE r= r= r> rG s
|
zTestStats.setup_methodc C sR t d¡dddfD ]:}tt |g¡d tt |g¡d tt |g¡d qd S )Nrý r# y ð? rü )r' rð r rý r r r r= r= r> Útest_python_type s zTestStats.test_python_typec C sz t d¡}| jD ]d}dD ]4}|||dd}t|j|jk t|j| dk qdD ] }|||dd}t|jdk qRqd S )NrÌ r Tr´ r# rR ri )r' r½ Úfuncsr rÏ r7 )rF Úmatr× r¨ rL r= r= r> Ú test_keepdims¡ s
|
|
zTestStats.test_keepdimsc C s t d¡}| jD ]<}t d¡}||dd}||d|d}t|| t|| qt d¡}tt||d|d t d¡}tt||d|d d S )NrÌ r# rª rà r" r )r' r½ r r* r
|
r2 r r) )rF r
|
r× rµ r rL r= r= r> Útest_out¬ s
|
|
|
|
|
|
zTestStats.test_outc C s° t jd }t jd }| jD ]^}t tdgd gd ¡}|jj}||ddjj}t||k t||d d}t|tk q| jD ]Z}|D ]P}t jd|d}t j }||ddjj}t||k ||d djj}t||k qqt
|
fD ]\}|D ]R}t jd|d}|jj}||ddjj}t||k ||d djj}t||k qêqât t fD ]b}|D ]V}t jd|d}|j jj}||ddjj}t||k ||d djj}t||k qPqHd S )Nr
r¢ r# rÌ rª r )r' r r ræ r r rY r r½ rÈ rÿ r r rx )rF rÊ rË r× r
|
r rL r³ r= r= r> Útest_dtype_from_input¹ sB
|
|
|
|
|
|
z TestStats.test_dtype_from_inputc C sr t d¡}| jD ]\}t jd D ]L}t |¡j}||d|djj}t||k ||d |djj}t||k q qd S )NrÌ r¢ r# )r¨ r )r' r½ r r r rY r )rF r
|
r× r³ r rL r= r= r> Útest_dtype_from_dtypeê s
|
|
z TestStats.test_dtype_from_dtypec C sÄ t fD ]H}tdD ]:}| jjd }|| jdd| }|| jd|d|| }qqtfD ]h}tdD ]Z}| jjd }|| jddt |¡ }|| jd|dt || ¡ }t|| t|| qbqVd S )NrÌ r# rª ©r¨ Úddof)r rq r r7 r r' Úsqrtr
|
)rF r× r rØ r rL r= r= r> Ú test_ddof s
|
|
|
zTestStats.test_ddofc C s | j jd }ttfD ]}t||d D ]l}tjddV}t d¡ || j d|d}t|dk ¡ tt
|
|dk tt |d j t W 5 Q R X q&qd S )Nr# r" Tr r r r )r r7 r r rq r r r r r§ r r Úcategoryrî )rF rØ r× r r rL r= r= r> Útest_ddof_too_big s
|
zTestStats.test_ddof_too_bigc C sÈ t d¡}| jD ]²}dD ]d}tjddN}t d¡ tt |||d¡ ¡ tt |dk tt
|
|d j t W 5 Q R X qdD ]>}tjdd(}t d¡ t |||dt g ¡ W 5 Q R X qqd S ) Nr r¦ Tr r rª r rh )r' r* r r r r r r~ r¬ r r r rî r )rF rÑ r× r¨ r r= r= r> rà s
|
|
|
|
zTestStats.test_emptyc C s | j | j| jfD ]r}dD ]0}|j|d}t||d|j| }t|| qdD ]2}|j|d}t||dt |j¡ }t|| qNqd S )Nr rª rR ) r r r r¦ rÿ r7 r
|
r' rÉ ©rF r
|
r¨ r rL r= r= r> Útest_mean_values, s zTestStats.test_mean_valuesc C s t ttjddddk d S )NrV rð r r# )r rÿ r' r rE r= r= r> Útest_mean_float167 s zTestStats.test_mean_float16c C s0 t tjjj t d¡jdd W 5 Q R X d S ©NrB r" rª )r r' re r r rC rý rE r= r= r> Útest_mean_axis_error< s z TestStats.test_mean_axis_errorc C sä t d¡ d¡}t ddddgddddgddddgddddgg¡}t dgdgdgdgg¡}dddddd gfd
|
|d d d dgfd|dd ddgfd
|
|d dddgfg}|D ]@\}}}t|j||dt |¡ tt j|||dt |¡ q¤t d¡ d¡}t ddddg¡} ddgdd gg}t|jd| dt |¡ tt j|d| dt |¡ t t¡,}
|
t|jd|dt t j ddt j g¡ W 5 Q R X t t¡}
|
t
|
|jddt j W 5 Q R X t t¡ }
|
t
|
t j|ddt j W 5 Q R X d S )Nr® )r r FTr# rÕ g @g #@g +@r ç @rD ç $@rm r g !@g -@ç @r r§ ©r" r" r r" r« ) r' rC rÆ ræ r rý r¨ r¾ rî rÚ r © rF rD r Z
|
wh_partialÚ_casesr® Ú_whÚ_resÚa3dÚ _wh_partialr r= r= r> Útest_mean_whereB sP
|
|
|
ýýýÿÿÿÿ ÿ zTestStats.test_mean_wherec C sj | j | j| jfD ]T}dD ]J}t|| ¡ |d}t||d}||| ¡ }t||d}t|| qqd S ©N©r r# Nrª )r r r rÿ rã râ r r
|
©rF r
|
r¨ Úmsqrrý r rL r= r= r> Útest_var_valuesf s zTestStats.test_var_values)Ú complex_dtypeÚndec))rÆ rÄ )rÇ rì )rê rì c C sh dD ]^}| j ¡ |¡}t|| ¡ |d}t||d}||| ¡ }t||d}t|||d qd S )Nr& rª rv )r r rE rÿ rã râ r r
|
) rF r* r+ r¨ r
|
r( rý r rL r= r= r> Útest_var_complex_valueso s z!TestStats.test_var_complex_valuesc C sf t | jgd ¡}dD ]J}t|| ¡ |d}t||d}||| ¡ }t||d}t|| qd S )NrÌ )r r# r" r% Nrª )r' Ústackr rÿ rã râ r r
|
r' r= r= r> Útest_var_dimensions~ s z TestStats.test_var_dimensionsc C s6 | j ¡ d¡}| |j ¡ ¡}t| ¡ | ¡ d S )NrÇ )r r rE r r r
|
r )rF r Z cmat_swappedr= r= r> Útest_var_complex_byteorder s z$TestStats.test_var_complex_byteorderc C s0 t tjjj t d¡jdd W 5 Q R X d S r )r r' re r r rC r rE r= r= r> Útest_var_axis_error s z TestStats.test_var_axis_errorc C sþ t d¡ d¡}t dddddgdddddgdddddgdddddgdddddgg¡}t dgdgdgdgdgg¡}dddddddgfdddddddgfg}|D ]@\}}}t|j||d t |¡ tt j|||d t |¡ q t d
|
¡ d ¡}t ddddg¡} d d gd d gg}t|jd | d t |¡ tt j|d | d t |¡ tt j|d|d t j|| d¡dd tt j|d|d t j||d d
df dd t t¡}
|
t |jddt j
|
W 5 Q R X t t¡ }
|
t t j|ddt j
|
W 5 Q R X d S )Nr9 rÖ FTr g I@r# r r§ r® r r÷ r" ©rH rÌ rª r« ) r' rC rÆ ræ r r r¨ r¾ rî r rÚ r r= r= r> Útest_var_where sT üüÿÿÿÿÿÿ ÿ zTestStats.test_var_wherec C sL | j | j| jfD ]6}dD ],}t t||d¡}t||d}t|| qqd S r% )r r r r' r r r r
|
r r= r= r> Útest_std_values» s
|
zTestStats.test_std_valuesc C sx t d¡ d¡d d d
}t dddddgdddddgdddddgdddddgdddddgg¡}t dgdgdgdgdgg¡}dddt d¡ fd dd
|
t d¡ fd|t d d d ddg¡fd|dt d¡ fg}|D ]4\}}}t|j||d| tt j|||d| qÔt d¡ d¡}t ddddg¡} ddgddgg}t|jd| dt |¡ tt j|d| dt |¡ t|jd |dt j|| d¡d d tt j|d |d|| d¡jd d t|jd|dt j||d d
df dd tt j|d|d||d d
df jdd t t ¡}
|
t
|
|jddt j W 5 Q R X t t ¡ }
|
t
|
t j|ddt j W 5 Q R X d S )Nr9 rÖ r% FTr gMþþÿÅH @rH r# gß'Üe ö?g½ê·evT@gùÐ fvT @rD g¢ @g,I\Ï)ÿ @rE r§ r® r rö r" r1 rª r« ) r' rC rÆ ræ r r r r¨ r¾ rî r rÚ ) rF rD ZwhfZwhpr r® r r! r" r# r r= r= r> Útest_std_where s` üüÿûÿÿÿÿ ÿÿ zTestStats.test_std_wherec C s G dd dt j}|ddddgddd d
|
ggd }| d¡}t|j|jk | d¡}t|j|jk | d¡}t|j|jk d S ) Nc @ s e Zd Zdd Zdd ZdS )z*TestStats.test_subclass.<locals>.TestArrayc S s t |¡}| | ¡}||_|S rR )r' ræ rW Úinfo)Úclsr$ r5 r r= r= r> Ú__new__ð s
|
|
z2TestStats.test_subclass.<locals>.TestArray.__new__c S s t |dd| _d S )Nr5 r` )r~ r5 r r= r= r> Ú__array_finalize__ö s z=TestStats.test_subclass.<locals>.TestArray.__array_finalize__N)rN rO rP r7 r8 r= r= r= r> Ú TestArrayï s r9 r# r" rÌ r rH rÄ rì r Zjubba)r' r6 rý r r5 r r )rF r9 ZdatrL r= r= r> Ú test_subclassî s
|
|
|
|
zTestStats.test_subclassN) rN rO rP rÿ r r r rG r r r r r r r rà r r r r$ r) r¨ r© r« r, r. r/ r0 r2 r3 r4 r: r= r= r= r> r s0
|
1 $
|
|
%,r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestVdotc C sü t jd t jd }t jd }t d¡}|d D ]:}| |¡}t ||¡}tt |¡ tt ||¡d q0t d¡d }|d D ]:}| |¡}t ||¡}tt |¡ tt ||¡d qt jdtd}t ||¡}tt |¡ tt ||¡d d S ) Nr¢ r
rî rÌ r rw r T) r' r r½ rE Úvdotr Zisscalarr r )rF Z
|
dt_numericZ
|
dt_complexrD r© r¦ rL r= r= r> r s$
|
|
|
|
zTestVdot.test_basicc C sz t jddgddggdd}t jddgddggdd}t ||¡}tt ||¡| tt ||¡| tt ||¡| d S ©Nr# r" rÌ r r r& r )r' ræ r< r ©rF rD r¦ rL r= r= r> Útest_vdot_array_order s z TestVdot.test_vdot_array_orderc C s2 dD ]&}t |ddf¡}t |ddf¡}t |¡|d d
ddf< t |¡d |d d
ddf< |d }|d }tt ||¡t | ¡ | ¡ ¡ tt || ¡ ¡t | ¡ | ¡ ¡ tt | ¡ |¡t | ¡ | ¡ ¡ tt | d¡|¡t | ¡ | ¡ ¡ tt || d¡¡t | ¡ | ¡ ¡ qd S )N)r" rp r" r r# ©.r r )r' r* rC r r< r r )rF r9 rD r¦ r= r= r> Útest_vdot_uncontiguous' s,
|
ÿÿÿÿÿz TestVdot.test_vdot_uncontiguousN)rN rO rP r r? rA 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 dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zejjej d)ej ej g¡e d*d+d,d- Z d.d/ Z d0S )1ÚTestDotc C sb t j d¡ t j dd¡| _t j dd¡| _t j d¡| _t j dd¡| _t j d¡| _d| _ d S )NrM r r" r# rì )
|
r' r rª r rÑ rS Úb2Úb3Úb4r? rE r= r= r> rG ? s zTestDot.setup_methodc C s@ | j }t | ¡ |¡}t ddgddgg¡}t||| jd d S )NgÑh®5÷?gÑïVñ¡ë?gÂ1×-ë?rv )rÑ r' r¢ rê ræ r
|
r? )rF rÑ rL r r= r= r> Útest_dotmatmatH s
|
ÿzTestDot.test_dotmatmatc C sH | j | j }}t ||¡}t dgdgdgdgg¡}t||| jd d S )NgcLU5Ô?g ¡ ©?gè )XFÄ?gÄÑÑÕ?rv )rÑ rS r' r¢ ræ r
|
r? )rF rÑ rS rL r r= r= r> Útest_dotmatvecO s ÿzTestDot.test_dotmatvecc C s@ | j | j }}t ||¡}t ddddg¡}t||| jd d S )NgHoþÒ?gį"§?g-P½ûÂ?gÐNzÝÓ?rv )rÑ rC r' r¢ ræ r
|
r? )rF rÑ rC rL r r= r= r> Útest_dotmatvec2V s zTestDot.test_dotmatvec2c C s< | j | j }}t ||¡}t ddg¡}t||| jd d S ©Ng×Ø©â[Âó?gâäÀ£ôñ?rv )rÑ rE r' r¢ ræ r
|
r? ©rF rÑ rE rL r r= r= r> Útest_dotvecmat\ s zTestDot.test_dotvecmatc C sF | j | j }}t || ¡ ¡}t ddddgg¡}t||| jd d S )Ngú»ocÐâ?gTbi\î¶?gï¾2rÓ?gú°bá¹ä?rv )rD rÑ r' r¢ rê ræ r
|
r? )rF rD rÑ rL r r= r= r> Útest_dotvecmat2b s zTestDot.test_dotvecmat2c C s@ | j | j }}t | ¡ |¡}t ddg¡}t||| jd d S rI )rÑ rE r' r¢ rê ræ r
|
r? rJ r= r= r> Útest_dotvecmat3h s zTestDot.test_dotvecmat3c C sD | j | j }}t ||¡}t ddgddgg¡}t||| jd d S )Ngc×Q0¾ÃÉ?g¢X¸öOµ?g5쵨h²?g 5î»?rv ©rS rD r' r¢ ræ r
|
r? ©rF rS rD rL r r= r= r> Útest_dotvecvecoutern s zTestDot.test_dotvecvecouterc C s< | j | j }}t ||¡}t dgg¡}t||| jd d S )Ngþ".!Í?rv rN rO r= r= r> Útest_dotvecvecinnert s zTestDot.test_dotvecvecinnerc C s@ t d¡}dg}t ||¡}t dddg¡}t||| jd d S )N©rÌ r# g333333@rv )r' r r¢ ræ r
|
r? ©rF rS rC rL r r= r= r> Útest_dotcolumnvect1z s
|
|
zTestDot.test_dotcolumnvect1c C sD t d¡ ¡ }dg}t ||¡}t dddg¡}t||| jd d S )NrR r4 rv )r' r rê r¢ ræ r
|
r? rS r= r= r> Útest_dotcolumnvect2 s
|
zTestDot.test_dotcolumnvect2c C s\ t j d¡ t j dd¡}t j dd¡}t ||¡}t ddddgg¡}t||| jd d S ) Nrµ r# r gè#:º\Ã?gN
p³Í?göwr*aÝ?çÉp<e?rv ©r' r rª r r¢ ræ r
|
r? rS r= r= r> Útest_dotvecscalar s zTestDot.test_dotvecscalarc C sb t j d¡ t j dd¡}t j dd¡}t ||¡}t dgdgdgdgg¡}t||| jd d S ) Nrµ r r# rV gpVÈ
U?gX;'Ii`?g3½«»Tp?rv rW rS r= r= r> Útest_dotvecscalar2 s zTestDot.test_dotvecscalar2c
|
C s dddg}dddddddddg }t |t ||¡D ]V\}\}}t |¡}t |¡}t ||¡}t |¡} t|j| jk t|| | j d q2d S )Nr= rh ri rv )
|
ÚziprÌ rþ r' r* r¢ r r7 r
|
r? )
|
rF ÚdimsZdoutrØ Zdim1Zdim2rS rC rL r r= r= r> r½ s
|
|
|
|
zTestDot.test_allc s¶ G fddd t ddgddgg¡}t |¡}t ddg ddgg¡}t ddg ddgg¡}t ||¡t ||¡ }t|d j|d j t|d j|d j d S )Nc sB e Zd Zd ddZ fddZ fddZ fdd Zd
|
d ZdS ) z#TestDot.test_vecobject.<locals>.VecNc S s |d kr g }t |¡| _d S rR r& )rF Úsequencer= r= r> rU ¥ s z,TestDot.test_vecobject.<locals>.Vec.__init__c s }| j |j |_ |S rR r ©rF r rµ ©ÚVecr= r> Ú__add__ª s z+TestDot.test_vecobject.<locals>.Vec.__add__c s }| j |j |_ |S rR r r] r^ r= r> Ú__sub__¯ s z+TestDot.test_vecobject.<locals>.Vec.__sub__c s | j ¡ }| j |9 _ |S rR )ræ r r] r^ r= r> rI ´ s z+TestDot.test_vecobject.<locals>.Vec.__mul__c S s | | S rR r= r r= r= r> Ú__rmul__¹ s z,TestDot.test_vecobject.<locals>.Vec.__rmul__)N)rN rO rP rU r` ra rI rb r= r^ r= r> r_ ¤ s
|
|
r_ rý r rü r r# )r' rê r ræ r¢ r )rF Z
|
U_non_contZU_contrá r* Z
|
zeros_testr= r^ r> Útest_vecobject£ s
|
zTestDot.test_vecobjectc C sr ddl m} tjddgddggtd}tjddgddggtd}tjddgddggtd}|||}t|| d S ) Nr ©r¢ r# r" rÌ r r rì )Únumpy.core.multiarrayr¢ r' ræ rÿ r )rF r¢ rD r¦ r³ rÕ r= r= r> Útest_dot_2argsÄ s
|
zTestDot.test_dot_2argsc C sê ddl m} tj d¡ tj d¡}tj d¡}t d¡}tdD ]}|||| qBtrht t
|
|¡d |||d d }t || t |||||d k |d d
df ¡ }|d d
df ¡ }|||}t |||||k t || d S )
|
Nr rd é ©r r® ©r® r¯ ©r r¯ r+ r" r· )re r¢ r' r rª Ú random_sampler2 rq r r r; rÀ r r r )rF r¢ r× rx rê r¿ Úr2r= r= r> Útest_dot_3argsÎ s"
|
|
|
zTestDot.test_dot_3argsc
|
C s8 ddl m} tj d¡ tj d¡}tj d¡}t d¡}tt|||| t d¡}tt|||| t d¡}tt|||| t d ¡}tt|||| tt||||j t d
|
¡}tt||||d d
d d d
f tt||||d d
d d
f tjd tj
|
d}tt|||| tjd t d}tt|||| d S )Nr rd rg rh ri )r ry )r )r¯ )r¯ r )r r° r" r¯ rj r ) re r¢ r' r rª rk r2 r r) r rÅ rÝ )rF r¢ r× rx rê r= r= r> Útest_dot_3args_errorsä s(
|
|
|
|
|
" z TestDot.test_dot_3args_errorsc C sz t jddgddggdd}t jddgddggdd}t ||¡}tt ||¡| tt ||¡| tt ||¡| d S r= )r' ræ r¢ r r> r= r= r> Útest_dot_array_order s z TestDot.test_dot_array_orderc s ddd d fdd }dd } dd t j} d
|
d t j}t ||¡ t d d d d¡}|D ]\}}}}t j ||¡} t j |¡}
|
t | |
|
¡} || |t j|d} ||
|
|t j} || | | | d d d
}t ||
|
¡} | d d d
}||| | |d d
d d d
f }|
|
d d d
}t ||¡} |d d
d d d
f }| d d d
}|||| |dkrpt|jdd| f nt|jd| df t|jd |||t j}|||| | d d
d d d
f }t ||¡} | d d
d d d
f }|||| |||| q\d S )Nr c S s |d}t | ¡}t j||j | t jd}|jd d }t|D ]}|| | dkrB q\qB|||||j
j|d}|j| |dS )Nr r r$ r& ) r' rÉ r* rÐ r3 r4 rq rW rÆ ) r7 r8 r r rÕ r? r) Úaddressr< r= r= r> Ú aligned_array s
|
zBTestDot.test_accelerate_framework_sgemv_fix.<locals>.aligned_arrayc s( | j |||}| d d
|d d
< |S rR )r7 )rS r8 r r r ©rq r= r> Ú
|
as_aligned s z?TestDot.test_accelerate_framework_sgemv_fix.<locals>.as_alignedc S s t t | |¡|ddd d S )Ngñhãµøä>gH¯¼ò×z>)ZrtolZatol)r r' r¢ )rÑ r Údesiredr= r= r> Úassert_dot_close s zETestDot.test_accelerate_framework_sgemv_fix.<locals>.assert_dot_closerµ ru r¿ )ru r¯ )r½ )éÈ éY )r r r& r" r r rÎ )r )r ) r' rÅ r¢ rÌ rþ r r r rÎ )rF rs ru rÍ r¾ Ztestdatar8 ro Za_orderZA_dZX_drt ZA_fZX_fZA_d_2ZA_f_2ZA_d_22ZX_d_2ZA_f_22ZX_f_2ZX_f_2cZA_d_12ZA_f_12r= rr r> Ú#test_accelerate_framework_sgemv_fix sF
|
|
z+TestDot.test_accelerate_framework_sgemv_fixr g ÐÃB)Z
|
free_bytesc C s* t jd|d}t ||¡}|dks&td S )Nid @r )r' r r¢ r~ )rF r r$ rL r= r= r> Útest_huge_vectordotL s zTestDot.test_huge_vectordotc C s^ G dd dt }t t¡ t | | ¡ W 5 Q R X t t¡ t d| ¡ W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z5TestDot.test_dtype_discovery_fails.<locals>.BadObjectc S s t dd S )Nzjust this tiny mint leafr rE r= r= r> r Y s z?TestDot.test_dtype_discovery_fails.<locals>.BadObject.__array__Nr r= r= r= r> Ú BadObjectX s rz rà )r( r¨ r rô r' r¢ )rF rz r= r= r> Útest_dtype_discovery_failsV s
|
z"TestDot.test_dtype_discovery_failsN)!rN rO rP rG rF rG rH rK rL rM rP rQ rT rU rX rY r½ rc rf rm rn ro rx r¨ r© rç r« r' rÈ rÇ r ry r{ r= r= r= r> rB > s2 !
|
|
@rB c @ sT e Zd ZdZdZdd Zdd Zdd Zd d
|
Zd d Z d d Z
|
dd Z dd Z dS )Ú MatmulCommonz5Common tests for '@' operator and numpy.matmul.
|
|
z?bhilqBHILQefdgFDGOc C sb ddddddddd g }t | j|¡D ]8\}\}}tj||d
|
}tj||d
|
}tt| j|| q$d S ) N)rh rr )rï rr )rr rÚ )rÚ rR )rh r= )r= r# )ri r= )r= ri ))r" r" r# )rÌ r# r" r )rÌ rþ ri r' r r r) r® )rF r[ r© Údm1Údm2rD r¦ r= r= r> Útest_exceptionsk s ÷ z MatmulCommon.test_exceptionsc C s¨ dddg}t | j|¡D ]B\}\}}tj||d}tj||d}| ||¡}t|jdk q| jD ]@}tjd|d}tjd|d}| ||¡}tt |¡jdk qbd S )N)ri ©r" r# r# )r ri )r r r r rr r= ) rÌ rþ ri r' r r® r r7 ræ ) rF r[ r© r} r~ rD r¦ rL r³ r= r= r> Ú test_shapes} s ý
|
zMatmulCommon.test_shapesc C s t d¡}t d¡}| jD ]x}| |¡}| |¡}||f||f||ffD ] }| j| }t|j|k qH|dkr| ||¡}tt|t |¡jk qd S )Nri rh r )r' r ri rE r® r r rY )rF r
|
Úvecr© rÍ rx r¹ rL r= r= r> Útest_result_types s
|
|
|
|
|
|
z MatmulCommon.test_result_typesc C sÆ t dg¡}t ddg¡ dd¡}t ddg¡}| jdd
D ]F}| |¡}| |¡}| ||¡}t|| | |j|¡}t|| q>t jddgd d
|
dd¡}| |d d
d f |¡}t|d d S ) Nr" rÌ r r# r% rÄ r TrW r r )r' ræ rÆ ri rE r® r r ) rF Úvec1Úvec2r r© r6 r7 rL r r= r= r> Útest_scalar_output s
|
|
|
z MatmulCommon.test_scalar_outputc
|
C sÎ t ddg¡}t ddg¡ dd¡}t dg¡}t ddgddgg¡}| jdd
D ]L}| |¡}| |¡}| ||¡}t|| | || dd¡¡}t|| qTt jd d gd
|
d } | | | ¡}t|d d S ) Nr# r" rÌ r r% r rÄ r TrW r )r' ræ rÆ ri rE r® r )
|
rF r r
Útgt1Útgt2r© r6 r7 rL r r= r= r> Útest_vector_vector_values± s
|
|
|
z&MatmulCommon.test_vector_vector_valuesc C sB t ddg¡}t ddgddgg¡}t j|gd dd}t ddg¡}t j|gd dd}| jdd
D ]N}| |¡}| |¡}| |¡} | ||¡}
|
t|
|
| | || ¡}
|
t|
|
| qht d d
|
g¡}t d d
|
gd
|
d gg¡}t j|gd dd}t d d
|
g¡}t j|gd dd}| ||¡}
|
t|
|
| | ||¡}
|
t|
|
| d S ) Nr# r" rÌ r r rª rì rB TF©r' ræ r- ri rE r® r © rF r Úmat1Úmat2r r r© rx Úm1r rL r= r= r> Útest_vector_matrix_valuesÄ s,
|
|
|
|
|
z&MatmulCommon.test_vector_matrix_valuesc C sB t ddg¡}t ddgddgg¡}t j|gd dd}t ddg¡}t j|gd dd}| jdd
D ]N}| |¡}| |¡}| |¡} | ||¡}
|
t|
|
| | | |¡}
|
t|
|
| qht d d
|
g¡}t d d
|
gd
|
d gg¡}t j|gd dd}t d d
|
g¡}t j|gd dd}| ||¡}
|
t|
|
| | ||¡}
|
t|
|
| d S ) Nr# r" rÌ r r rª rH r TFr r r= r= r> Útest_matrix_vector_valuesß s,
|
|
|
|
|
z&MatmulCommon.test_matrix_vector_valuesc C s t ddgddgg¡}t ddgddgg¡}t j||gdd}t j||gdd}t ddgd d
|
gg¡}t ddgddgg¡}t ddgdd gg¡}t j||gdd}t j||fdd} t j||fdd}
|
| jdd
D ]} | | ¡} | | ¡} | | ¡}| | ¡}| | | ¡}t|| | | | ¡}t|| | || ¡}t||
|
| | |¡}t|| | ||¡}t|| qÖt jddgddggt jd } t jddgddggt jd } t j| | gdd}t j| | gdd}| }| }t jddgddggt jd }t j||gdd}t j||fdd} t j||fdd}
|
| | | ¡}t|| | | | ¡}t|| | || ¡}t||
|
| | |¡}t|| | ||¡}t|| d S ) Nr# r" rÌ r r rª rì rB ru rg rÄ r )r' ræ r- ri rE r® r rP )rF r r Zmat12Zmat21Ztgt11Ztgt12Ztgt21Ztgt12_21Ztgt11_12Ztgt11_21r© r r Zm12Zm21rL r= r= r> Útest_matrix_matrix_valuesú sZ
|
|
|
|
|
|
|
|
|
|
|
|
z&MatmulCommon.test_matrix_matrix_valuesN) rN rO rP rä ri r r r r r r r r r= r= r= r> r| c s r| c % @ sp e Zd ZejZdd Zdd Zdd Ze d¡ dd ¡Z
|
e d
|
¡ d d ¡Z e d ¡ dd ¡dd
ddd
f Z e d¡Z e d¡Ze d¡Zej de
|
e fe je
|
jfe j ¡ e
|
jfe je
|
j ¡ fe
|
e
|
jfe
|
je
|
fe
|
e jfe e
|
jfe e jfe je fe e fe je jfe j ¡ e jfe
|
edd
fe dd
e
|
fe
|
je dd
fedd
e
|
jfe
|
eddd
fe ddd
e
|
fe
|
je ddd
feddd
e
|
jfe edd
fe dd
e fe je dd
fedd
e jfe eddd
fe ddd
e fe je ddd
feddd
e jfeejfejefe
|
efeje
|
jff!¡dd Zdd Zdd Zdd Zd d Zd d Zd d! Z dS )"Ú
|
TestMatmulc C s" t jdtd}t jddgddggtd}t ||¡}d}t jdtd}| |||¡ t|||d d }t jdtd}| j|||d
|
t|||d d }t jdt jd}t t
|
|| j|||d
|
t jdt j d}| j|||d
|
}t ||k t }| t jd ¡ | |j¡}W 5 Q R X t|| d S ) N©rH r" r r# rÌ rH rì zout positional argumentr± zout keyword argumentr· zCannot cast ufunc .* outputr` )r' r rÿ ræ r¢ r* r® r r¶ r rô rÇ r r rí ÚComplexWarningrE r )rF rD r¦ r r rµ r³ rð r= r= r> Ú test_out_arg? s* zTestMatmul.test_out_argc C sX t d¡}t d¡}| ||¡jdks*ttjtdd | j|||d W 5 Q R X d S )N)r r# r# r znon-broadcastabler r· )r' r r® r7 r~ r¨ r r) r' r= r= r> Útest_empty_out^ s
|
|
|
zTestMatmul.test_empty_outc C sJ t jdtd}t jddgddggtd}t jddgtd}t ||¡}t ||¡}t jdtd}| j|||d d }|j|kstt|| | j|||d d
d
|
d
|
f d }t|| | j||j |d d
d
|
d
|
f d }t|| t jd td}| j|||d d d
d d
f d }t|| t jdtd}| j|j |j |j d }t|| d S ) Nr r r# rÌ rH rì )rH r" r" r@ r· r )rB r" r" )
|
r' r rÿ ræ r¢ r® rv r~ r r )rF rD r¦ rx r Ztgt_mvrµ r³ r= r= r> Útest_out_contiguoush s&
|
|
|
"
|
z TestMatmul.test_out_contiguousg .@rH rÌ g 5@rì g >@rÄ Nr" r r ©rÌ r r c C sH t j| }t j| }t|| t |d ¡ |d ¡ ¡}t|| d S r~ )r' r® r¢ r r )rF r Zr1rl Zr3r= r= r> Útest_dot_equivalent s
|
|
|
|
z TestMatmul.test_dot_equivalentc
|
C st dd l }t |j¡}dd }|| | }|| | }| ||¡}dd |||fD \}}} t| | ||¡ d S )Nr c S s t jjddddS )Nr# rp )rB rÌ rÌ r² )r' r r r= r= r= r> Ú random_ints¨ s z2TestMatmul.test_matmul_object.<locals>.random_intsc S s g | ]}| t¡qS r= )rE rÿ rz r= r= r> rY ¯ s z1TestMatmul.test_matmul_object.<locals>.<listcomp>)Ú fractionsr' Z vectorizeÚFractionr® r )
|
rF r r× r ZM1ZM2ZM3ZN1ZN2ZN3r= r= r> Útest_matmul_object¤ s z TestMatmul.test_matmul_objectc C sF ddl m} t |dd|ddg¡}| ||¡}tt||k d S )Nr )r r" rÌ rH rì )r r r' ræ r® r rY )rF r rx rL r= r= r> Ú test_matmul_object_type_scalar³ s z)TestMatmul.test_matmul_object_type_scalarc C s< t jdtd}t jdtd}t d¡}tt ||¡| d S )Nr r r r° )r' r2 r( r* r r® r£ r= r= r> Útest_matmul_empty¹ s
|
z TestMatmul.test_matmul_emptyc C s@ G dd d}t d| ¡}tt t ||¡}W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )zCTestMatmul.test_matmul_exception_multiply.<locals>.add_not_multiplyc S s | S rR r= r r= r= r> r` Â s zKTestMatmul.test_matmul_exception_multiply.<locals>.add_not_multiply.__add__N)rN rO rP r` r= r= r= r> Úadd_not_multiplyÁ s r r° ©r' Úfullr rô r® )rF r rD r¦ r= r= r> Ú test_matmul_exception_multiply¿ s
|
z)TestMatmul.test_matmul_exception_multiplyc C s@ G dd d}t d| ¡}tt t ||¡}W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z>TestMatmul.test_matmul_exception_add.<locals>.multiply_not_addc S s | S rR r= r r= r= r> rI Ë s zFTestMatmul.test_matmul_exception_add.<locals>.multiply_not_add.__mul__N)rN rO rP rI r= r= r= r> Úmultiply_not_addÊ s r¤ r° r¡ )rF r¤ rD r¦ r= r= r> Útest_matmul_exception_addÈ s
|
z$TestMatmul.test_matmul_exception_addc C sô t jddgddggtd}t | t j¡¡dks4tt ||¡}t | t j¡¡dksZtt j t j
|
d¡¡}|j ddt j d}| dd ¡dk}t || d d¡¡}t || d d¡¡}t|| t t jd
|
tdt jdtd¡}t |¡rðtd S ) Nr# r r rú r" r r r rH )r" r )r' ræ r r» rW r3 r~ r® r Ú default_rngZPCG64Úintegersr³ rÆ r¢ r r* r§ )rF rD r¦ ZrgrÕ rÇ rÈ r³ r= r= r> Útest_matmul_boolÑ s
|
zTestMatmul.test_matmul_bool) rN rO rP r' r® r r r rC rÆ r r r ZvcZvrr* Zm0r¨ r© r« r r r r r r r£ r¥ r¨ r= r= r= r> r < sh
|
$
|
|
|
|
|
|
ï
|
r c @ s* e Zd ZddlZejZdd Zdd ZdS )ÚTestMatmulOperatorr Nc C sF G dd d}| }t d¡}t| ||¡d t| ||¡d d S )Nc @ s e Zd ZdZdd Zdd ZdS )z:TestMatmulOperator.test_array_priority_override.<locals>.Arp c S s dS ©NrÑ r= r r= r= r> Ú
|
__matmul__í s zETestMatmulOperator.test_array_priority_override.<locals>.A.__matmul__c S s dS rª r= r r= r= r> Ú __rmatmul__ð s zFTestMatmulOperator.test_array_priority_override.<locals>.A.__rmatmul__N)rN rO rP r r« r¬ r= r= r= r> rÑ ê s rÑ r" )r' r r r® )rF rÑ rD r¦ r= r= r> Ú test_array_priority_overrideè s
|
|
z/TestMatmulOperator.test_array_priority_overridec C sX t t| jt d¡t d¡ t t| jt d¡t d¡ t t| jt d¡t d¡ d S )NrH s abcrB )r rô r® r' r³ rr rC rE r= r= r> Útest_matmul_raisesø s z%TestMatmulOperator.test_matmul_raises)rN rO rP r/ r® r r® r= r= r= r> r© ä s r© c C sR t d¡} t d¡}tt| j| dd l}tt|j| | tttdt t d S )NrÌ r za @= b)
|
r' r½ r rô Ú __imatmul__r/ ÚimatmulÚexecÚglobalsrI )rD r¦ r/ r= r= r> Útest_matmul_inplaceý s
|
|
r³ c C s¢ t d¡ ddd¡} t j| | dddgd}|jd ks8tt j| | ddd
|
gd}|jd ks\tt |d d ¡}t|| t j| t d¡dd d gd}|jdkstd S )Nrà rÌ r rH )rã r% )r% rã rÚ )r¼ )rÌ r r r )r r rÌ r r" )r# r rË )r' rC rÆ r® r7 r~ rÐ r )rD r³ rÕ rë r× r= r= r> Útest_matmul_axes s
|
r´ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestInnerc C s6 d}t jddd}ttt j|| ttt j|| d S r¥ )r' ræ r rô r² r¦ r= r= r> Útest_inner_type_mismatch s z"TestInner.test_inner_type_mismatchc C s| t jd t jd d D ]^}t jd|dd }t jddg|d}t jdd g|d}tt ||¡| tt ||¡| qd S )
|
Nr
r¢ rW rÌ r r= r# r" rÄ )r' r ræ r r² )rF r© Zscar rt r= r= r> Ú test_inner_scalar_and_vector s z&TestInner.test_inner_scalar_and_vectorc C s. t jdt jd}t ||¡}t|ddd d S )N)r# r¸ rà r r2 rv )r' r* rÈ r² r
|
)rF rD r r= r= r> Ú test_vecself% s zTestInner.test_vecselfc C sx t jd t jd d D ]X}t jddgddgg|d}t jddgddgg|d}t jddg|d}t jdd g|d}tt |j|¡| tt ||j¡| tt ||¡| tt ||¡| t jd
|
d gd d gg|d}tt ||¡| t jddgddgg|d}tt ||¡| tt || ¡ ¡| t d¡ |¡}|d d d
}t jd |d ¡ }tt ||¡| qd S )Nr
r¢ rW r# r" rÌ r r rÄ rì rB ru rg rH r r9 r% )
|
r' r ræ r r² r r rC rE rb )rF r© rÑ r[ r rt rD r¦ r= r= r> Ú,test_inner_product_with_various_contiguities- s$ z6TestInner.test_inner_product_with_various_contiguitiesc
|
C s t jd t jd d D ]ê}t d¡ ddd¡ |¡}t dd¡ ddd¡ |¡}t d d
|
d gd d dggdddgdddggdddgdddgggdd d gd d d ggd!d"d#gd$d%d&ggd'd(d)gd*d+d,gggg¡ |¡}tt ||¡| tt ||¡ ddd-d.¡| qd S )/Nr
r¢ rW rÄ r" rÌ r é0 é é¶ éÎ éæ éþ i i6 i iæ i> i iî iÎ if iþ i i. iÆ if i> i iî iÆ i iþ i i. iF
|
i^ iv i iî iF i iö iN r r# ) r' r rC rÆ rE ræ r r² rê )rF r© rD r¦ rt r= r= r> Útest_3d_tensorF s< ÿÿÿú ÿÿÿú÷ÿîzTestInner.test_3d_tensorN)rN rO rP r¶ r· r¸ r¹ rÀ r= r= r= r> rµ s
|
rµ c @ s e Zd Zdd Zdd Zdd Zdd Zej d d
|
e
|
j d ge
|
j d fd e
|
j d ge
|
j d fde
|
d¡fde
|
j dge
|
j d fg¡dd ZdS )Ú
|
TestChoosec C s` dt jdtd | _dt jdtd | _dt jdtd | _dt jdtd | _dddg| _d S )Nr" r r rÌ rä r r# )r' r rÝ rá rø r¼ r½ rÆ rE r= r= r> rG b s
|
zTestChoose.setup_methodc C s* t | j| j| jf¡}t|dddg d S ©Nr" rÌ )r' r» rÆ rá rø r r³ r= r= r> r i s zTestChoose.test_basicc C s4 t | j| j| jf¡}t|dddgdddgg d S r )r' r» rÆ r¼ r½ r r³ r= r= r> Útest_broadcast1m s zTestChoose.test_broadcast1c C s4 t | j| j| jf¡}t|dddgdddgg d S r )r' r» rÆ rá r½ r r³ r= r= r> Útest_broadcast2q s zTestChoose.test_broadcast2r rp r# r r% rý rÌ c C s& t j| }t dg|¡j|ks"td S rÜ )r' rB r» r r~ )rF r Z expected_dtr= r= r> Útest_output_dtypeu s
|
z TestChoose.test_output_dtypeN)rN rO rP rG r rà rÄ r¨ r© r« r' ræ r3 rÅ rÅ r= r= r= r> rÁ a s ýÿrÁ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestRepeatc C s* t ddddddg¡| _| j d¡| _d S ©Nr# r" rÌ r rH rÄ rä )r' ræ rÍ rÆ rÎ rE r= r= r> rG s zTestRepeat.setup_methodc C s< t | jddddddg¡}t|ddddddddddg
|
d S )Nr# rÌ r" r rH rÄ ©r' rÌ rÍ r r³ r= r= r> r s ÿzTestRepeat.test_basicc C s4 t | jd¡}t|ddddddddddddg d S )Nr" r# rÌ r rH rÄ rÈ r³ r= r= r> rà s ÿzTestRepeat.test_broadcast1c C sz t j| jddgdd}t|dddgdddgdddgg t j| jdddgdd}t|ddddddgddddddgg d S ) Nr" r# r rª rÌ r rH rÄ ©r' rÌ rÎ r r³ r= r= r> Útest_axis_spec s þÿzTestRepeat.test_axis_specc C sx t j| jddd}t|dddgdddgdddgdddgg t j| jddd}t|ddddddgddddddgg d S ) Nr" r rª r# rÌ r rH rÄ rÉ r³ r= r= r> rÄ s ýÿzTestRepeat.test_broadcast2N)rN rO rP rG r rà rÊ rÄ r= r= r= r> rÆ s
|
|
rÆ r# r" rÌ r )ÚzerorÅ ÚconstantÚcircularÚmirrorr© rÿ r( ©Úidsc @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestNeighborhoodIterc C sX t jddgddgg|d}t jdddgdddgg|dt jdddgdddgg|dt jdddgdddgg|dt jdddgdddgg|dg}t |ddddg|d td ¡}t|| t jdddgdddgg|dt jdddgdddgg|dt jdddgdddgg|dt jdddgdddgg|dg}t |ddddg|d td ¡}t|| t jd d d gd ddgg|dt jd d d gddd gg|dt jd ddgd ddgg|dt jddd gddd gg|dg}t |ddddgd td
|
¡}t|| t jd ddgd ddgg|dt jddd gddd gg|dg}t |ddddgd td
|
d¡}t|| d S ) Nr r# r" rÌ r r% rË rÅ r rÌ ©r' ræ rm Útest_neighborhood_iteratorÚ
|
NEIGH_MODEr ©rF r© rá rê rÔ r= r= r> Ú test_simple2d« s` ý
|
ÿ
|
ý
|
ÿ
|
ý
|
ÿ
|
ÿ
|
ÿz"TestNeighborhoodIter.test_simple2dc C s¼ t jddgddgg|d}t jdddgdddgg|dt jdddgdddgg|dt jdddgdddgg|dt jdddgdddgg|dg}t |ddddg|d td ¡}t|| d S )Nr r# r" rÌ r r% rÎ rÒ rÕ r= r= r> Ú test_mirror2dÍ s ý
|
ÿz"TestNeighborhoodIter.test_mirror2dc C s t ddd¡ |¡}dddgdddgdddgdddgdddgg}t |ddg|d td ¡}t|| dddgdddgdddgdddgdddgg}t |ddg|d td ¡}t|| |d ddgdddgdddgdddgdd|d gg}t |ddg|d td
|
¡}t|| d S ) Nr# rH r r" rÌ r r% rË rÅ rÌ )r' rd rE rm rÓ rÔ r rÕ r= r= r> Ú test_simpleØ s2 , ÿ
|
, ÿ
|
4 ÿz TestNeighborhoodIter.test_simplec
|
s t ddd¡ ¡}t jdddddgdddddgdddddgdddddgdddddgg d}t |ddg|d td ¡}t fd d
|
|D t|| d S ) Nr# rH r" rÌ r r rã rÎ c s g | ]}|j kqS r= r rW ©r© r= r> rY ñ s z4TestNeighborhoodIter.test_mirror.<locals>.<listcomp>) r' rd rE ræ rm rÓ rÔ r r rÕ r= rÙ r> Ú test_mirrorë s ( ÿÿ ÿz TestNeighborhoodIter.test_mirrorc
|
C s t ddd¡ |¡}t jdddddgdddddgdddddgdddddgdddddgg|d}t |ddg|d td ¡}t|| d S )
|
Nr# rH r r" rÌ r rã r rÍ )r' rd rE ræ rm rÓ rÔ r rÕ r= r= r> Ú test_circularõ s ( ÿÿ ÿz"TestNeighborhoodIter.test_circularN)rN rO rP rÖ r× rØ rÚ rÛ r= r= r= r> rÑ ¨ s
|
"
|
rÑ c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestStackedNeighborhoodIterc
|
C s: t j}t jdddg|d}t jdg|dt jdg|dt jdg|dt jdg|dt jdg|dt jdg|dt jdg|dg}t |ddgtd ddgtd ¡}t|| t jdddg|dt jdddg|dt jdddg|dt jdddg|dt jdddg|dg}t |d dgtd d dgtd
|
¡}t|| d S ) Nr# r" rÌ r r rã r rË r% rÅ ©r' rÈ ræ rm Z test_neighborhood_iterator_oobrÔ r rÕ r= r= r> Útest_simple_const s@ ú ÿ
|
ü ÿz-TestStackedNeighborhoodIter.test_simple_constc
|
C s t j}t jdddg|d}t jdddg|dt jdddg|dt jdddg|dt jdddg|dt jdddg|dg}t |ddgtd ddgtd ¡}t|| t jdddg|d}t jdddg|dt jdddg|dt jdddg|dt jdddg|dt jdddg|dg}t |ddgtd d dgtd ¡}t|| t jdddg|d}t jdddg|dt jdddg|dt jdddg|dt jdddg|dt jdddg|dg}t |ddgtd ddgtd ¡}t|| t jdddg|d}t jdddddg|dt jdddddg|dt jdddddg|dt jdddddg|dt jdddddg|dg}t |ddgtd d dgtd ¡}t|| d S )
|
Nr# r" rÌ r r r% rÎ rË rã rÝ rÕ r= r= r> Útest_simple_mirror sz ü ÿ
|
ü ÿ
|
ü ÿ
|
ü ÿz.TestStackedNeighborhoodIter.test_simple_mirrorc
|
C s t j}t jdddg|d}t jdddg|dt jdddg|dt jdddg|dt jdddg|dt jdddg|dg}t |ddgtd ddgtd ¡}t|| t jdddg|d}t jdddg|dt jdddg|dt jdddg|dt jdddg|dt jdddg|dg}t |ddgtd d dgtd ¡}t|| t jdddg|d}t jdddg|dt jdddg|dt jdddg|dt jdddg|dt jdddg|dg}t |ddgtd ddgtd ¡}t|| t jdddg|d}t jdddddg|dt jdddddg|dt jdddddg|dt jdddddg|dt jdddddg|dg}t |ddgtd d dgtd ¡}t|| d S )
|
Nr# r" rÌ r r r% rÍ rË rã rÝ rÕ r= r= r> Útest_simple_circularK sz ü ÿ
|
ü ÿ
|
ü ÿ
|
ü ÿz0TestStackedNeighborhoodIter.test_simple_circularc C s t j}t jdddg|d}t jddddg|dg}t |ddgtd ddgtd ¡}t|| t jdddg|d}t jddddg|dg}t |ddgtd ddgtd ¡}t|| t jdddg|d}t jddddg|dg}t |ddgtd ddgtd ¡}t|| d S )
|
Nr# r" rÌ r r rË r% rÎ rÍ rÝ rÕ r= r= r> Útest_simple_strict_within{ s> ÿ
|
ÿ
|
ÿz5TestStackedNeighborhoodIter.test_simple_strict_withinN)rN rO rP rÞ rß rà rá r= r= r= r> rÜ ÿ s 00rÜ c @ s e Zd Zdd ZdS )Ú TestWarningsc C sf t ddg¡}t ddg¡}t ¡ 8 t dt j¡ tt j|jtd | t |ddg W 5 Q R X d S )Nr# r" y ð? Ày ð? @r{ )
|
r' ræ r r r r r rô Úslicer rs r= r= r> Útest_complex_warning s
|
z!TestWarnings.test_complex_warningN)rN rO rP rä r= r= r= r> râ s râ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestMinScalarTypec C s" t d¡}t d¡}t|| d S )Nr\ r3 ©r' Zmin_scalar_typer r ©rF r© Úwantedr= r= r> Útest_usigned_shortshort£ s
|
|
z)TestMinScalarType.test_usigned_shortshortc C s" t d¡}t d¡}t|| d S )Niÿÿ rµ ræ rç r= r= r> Útest_usigned_short¨ s
|
|
z$TestMinScalarType.test_usigned_shortc C s" t d¡}t d¡}t|| d S )Nl ÿÿ r· ræ rç r= r= r> Útest_usigned_int s
|
|
z"TestMinScalarType.test_usigned_intc C s" t d¡}t d¡}t|| d S )Nr r¹ ræ rç r= r= r> Útest_usigned_longlong² s
|
|
z'TestMinScalarType.test_usigned_longlongc C s" t d¡}t d¡}t|| d S )Nrk r ræ rç r= r= r> rH · s
|
|
z TestMinScalarType.test_objectN)rN rO rP ré rê rë rì rH r= r= r= r> rå ¡ s
|
rå )Ú_dtype_from_pep3118c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )ÚTestPEP3118Dtypec C s, t |¡}t|}t||d||f d d S )Nzspec %r != dtype %rr± )r' r rí r )rF Úspecrè r© Úactualr= r= r> Ú_checkÁ s
|
|
|
ÿzTestPEP3118Dtype._checkc
|
C sx t d¡j}tdD ]^}|dkr&d}nd| }| d| dd|d|| fd ¡ | d
|
| ddd| fd ¡ qd S ) Nr¿ r r rì zb%dxiru )rî r r# ©r rÿ r )r' r r+ rq rñ )rF r8 rñ r¾ r= r= r> Útest_native_paddingÇ s ÿ
|
ÿz$TestPEP3118Dtype.test_native_paddingc C s< | ddddidfdfi¡ | ddddidfdfi¡ d S ) Nzx3T{xi}r )r¿ r r r z^x3T{xi})r¿ r# r# ©rñ rE r= r= r> Útest_native_padding_2Ó s z&TestPEP3118Dtype.test_native_padding_2c sx t d¡j t d¡j} fdd}tdgdgd}| dtf d||d i|¡ | d tf d||d
|
i|¡ | d tf d||d i|¡ | d tf d||d i|¡ | dtf d||d i|¡ | dtf d|d i|¡ | dtf d|d
|
i|¡ | dtf d|d i|¡ | dtf d|d i|¡ | dtf d|d i|¡ d S )Nr¿ c s d| d S r» r= ©ro rJ r= r> r Þ s z7TestPEP3118Dtype.test_trailing_padding.<locals>.alignedr )r2 r1 Úixr1 r# Zixxr" ZixxxrÌ Zixxxxr Zi7xrì z^ixz^ixxz^ixxxz^ixxxxz^i7x)r' r r+ r1 r" rñ )rF r9 r rv r= rJ r> Útest_trailing_paddingØ s """"" z&TestPEP3118Dtype.test_trailing_paddingc C s` t jdddt d¡fdgdd}| d|¡ t ddd d
|
d dt jdddfg¡}| d |¡ d S ) Nr) ©r¦ r¿ rù úb,ir< TrJ z(T{b:a:xxxi:b:T{b:f0:=i:f1:}:sub:xxxi:c:}©r³ r¦ ©rÕ r¦ ©rë r¦ z-T{b:a:=i:b:b:c:b:d:b:e:T{b:f0:xxxi:f1:}:sub:}©r' r rñ rå r= r= r> Útest_native_padding_3ï s ÿý ÿÿz&TestPEP3118Dtype.test_native_padding_3c C s& t jddddgdd}| d|¡ d S )Nr) rù ©r³ r¦ r r= TrJ zT{b:a:xxxi:b:3b:c:xi:d:}rþ rå r= r= r> Ú%test_padding_with_array_inside_structû s ÿýz6TestPEP3118Dtype.test_padding_with_array_inside_structc C s | dddidfdd¡ d S )Nz@T{^i}xir )r¿ r r )r¿ rH rò rô rE r= r= r> Ú test_byteorder_inside_struct s ÿz-TestPEP3118Dtype.test_byteorder_inside_structc
|
sR t d¡j t d¡j} fdd}| dtdgdgdg||d dd f¡ d S )
|
Nr¿ c s d| d S r» r= rö rJ r= r> r s z4TestPEP3118Dtype.test_intra_padding.<locals>.alignedz(3)T{ix}r r r# rù r )r' r r+ r1 rñ r" )rF r9 r r= rJ r> Útest_intra_padding s
|
üûz#TestPEP3118Dtype.test_intra_paddingc C s4 t d¡}| d|¡ t ddg¡}| d|¡ d S )Nr³ )r ZS1r )rÿ rb Z4c4srþ rå r= r= r> Útest_char_vs_string s
|
z$TestPEP3118Dtype.test_char_vs_stringc C s$ | dddg¡ | dddg¡ d S )Nz (0)I:a:f:b:)rD r rÓ r z (0)I:b:f:a:)r¦ r rÓ r rô rE r= r= r> Útest_field_order s z!TestPEP3118Dtype.test_field_orderc C s> | dddg¡ | dddg¡ | dd¡ | ddg¡ d S )NÚii)r r¿ )rÿ r¿ zii:f0:r¿ zi:f0:rô rE r= r= r> Útest_unnamed_fields# s z$TestPEP3118Dtype.test_unnamed_fieldsN)rN rO rP rñ ró rõ rø rÿ r r r r r r r= r= r= r> rî À s rî c @ s´ e Zd ZdZdd Zdd Zdd Zdd Zd
|
d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zej d d gejejd d ged ed!d"ejed d ed#d"g¡d$d% Zd&d' Zd(d) Z d*d+ Z ejj d,d-ej d.d/d fd0d1Z ejj d,d-ejj!ej d2d3d4j"j# d5d-e d6¡fd7d8Z$d9d: Z%d;d< Z&ejj'd=d> Z(d?d@ Z)dAdB Z*dCdD Z+dEdF Z,ej d e dG¡ej d dHd dI g¡dJdK Z-dLdM Z.dNS )OÚTestNewBufferProtocolz Test PEP3118 buffers c C s t |¡}t|}t |¡}t |¡}t|jj t|jj t|j|j t|j |j t
|
|| t|j|j t|j |j t
|
|| d S rR ) r' rX r ræ r rJ r} r r r7 r )rF r! rá rø r½ r= r= r> Ú_check_roundtrip. s
|
|
|
|
z&TestNewBufferProtocol._check_roundtripc C s> t jdddddgdd}| |¡ t jddgddggt jd}| |¡ t jdt jdd d
d d d
f }| |¡ d
|
d d d dddddddt jfdt jfdt jfdt j fdt j
|
fdt j fddd d d t j fg}t jd g|d}| |¡ t jddgddggfd t d!ffgd}| |¡ t jdddgd"d}| |¡ t jdddgd#d}| |¡ t jdddgd$d}| |¡ t jdddgd%d}| |¡ t jdddgd&d}| |¡ tjd'krút jdddgd(d}tt| j| t jdddgd)d}| |¡ n@t jdddgd(d}| |¡ t jdddgd)d}tt| j| d S )*Nr# r" rÌ r rH r¤ r ©rÌ rÌ rÌ r r) ©r¦ rý r< ©rÕ rÔ ©Zdxrü ©rë r[ ©r× ÚH©rê r ©rý ÚL©ÚhxÚQr¿ rñ r r÷ ÚjxÚkx©rÔ rb ©rÍ ZU4©ro r@ ©rh rW r )r# r# r# r# r# r# r# r# r# r# r# r# r# r# r# r# ó aaaaÚbbbbs xxxTrý rD r r rR rN rM z>qr8 z>gz<g)r' ræ r rÈ r* rÅ ræ rè rD rç ré rê ÚhalfrÝ r; r< r r) )rF rá r© r= r= r> r} > sl
|
|
"
|
ìý
|
&
|
|
|
|
|
|
|
z$TestNewBufferProtocol.test_roundtripc
|
C sT dddddddt dt d d
|
g
|
}tj|d d }| |¡ tj|d d }| |¡ d S )Nrý r" g üï@g ?g p>rü g z+infz-infg TÕ?z>er z<e)rÿ r' ræ r )rF Z half_listrá r= r= r> Útest_roundtrip_half s ö
|
z)TestNewBufferProtocol.test_roundtrip_halfc C s t j ¡ D ]}t |¡}|jdkr$q
|
|jdkr0q
|
t jd|d}| |¡ |jdkr
|
| d¡}t jd|d}| |¡ | d¡}t jd|d}| |¡ q
|
d S )NZMmr r r ZqQgGr r )r' Z
|
sctypeDictrj r rÙ r* r r )rF Útypr rá r© r= r= r> Útest_roundtrip_single_types s
|
|
|
|
|
|
|
|
z1TestNewBufferProtocol.test_roundtrip_single_typesc C s | d¡ d S rÜ )r rE r= r= r> Útest_roundtrip_scalar® s z+TestNewBufferProtocol.test_roundtrip_scalarc C sF t ddg¡}t d|¡}tttft| tttftt dd¡ d S )N)rD rµ )r¦ zM8[s]rÌ rV )r' r r2 r r) rÈ r ræ )rF r© rD r= r= r> Útest_invalid_buffer_format² s z0TestNewBufferProtocol.test_invalid_buffer_formatc C sl t jdddddgdd}t|}t|jd t|jd t|jd t|jd t|jd
|
t|j d d S ) Nr# r" rÌ r rH r¿ r )rH r r= ©
|
r' ræ r r rò r7 rÏ rÎ Ú
|
suboffsetsr1 rs r= r= r> Útest_export_simple_1dº s z+TestNewBufferProtocol.test_export_simple_1dc C sp t jddgddggt jd}t|}t|jd t|jd t|jd t|jd t|j d t|j
|
d
|
d S ) Nr# r" rÌ r r rÕ r )r® r r= r ) r' ræ rÈ r r rò r7 rÏ rÎ r& r1 rs r= r= r> Útest_export_simple_ndÄ s z+TestNewBufferProtocol.test_export_simple_ndc C sv t jdt jdd d
dd d
f }t|}t|jd t|jd t|jd t|jd t|j d t|j
|
d d S )
|
Nr
|
r r r× r° r" )rÁ r r= r ) r' r* rÅ r r rò r7 rÏ rÎ r& r1 rs r= r= r> Útest_export_discontiguousÎ s " z/TestNewBufferProtocol.test_export_discontiguousc C s ddddddddd d
|
d t jfd t jfd t jfdt jfdt jfdt jfdddddt jfg}t jdg|d}t |}t
|
|j d t
|
|j d t
|
|j d tdd |D }t d ¡jd krÄt
|
|jd n t
|
|jd t d¡jd! t t j¡jksüt
|
|j|f t
|
|j| d S )"Nr) 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# r# r# r r s Trý r rh r# r= c S s g | ]\}}t |¡jqS r= )r' r r1 )rX rD r¦ r= r= r> rY ø s z<TestNewBufferProtocol.test_export_record.<locals>.<listcomp>rÔ r zjT{b:a:=h:b:i:c:l:d:q:dx:B:e:@H:f:=I:g:L:h:Q:hx:f:i:d:j:^g:k:=Zf:ix:Zd:jx:^Zg:kx:4s:l:=4w:m:3x:n:?:o:@e:p:}zjT{b:a:=h:b:i:c:q:d:q:dx:B:e:@H:f:=I:g:Q:h:Q:hx:f:i:d:j:^g:k:=Zf:ix:Zd:jx:^Zg:kx:4s:l:=4w:m:3x:n:?:o:@e:p:}r )r' ræ rè rD rç ré rê r ræ r r r7 rÏ r& r¦ r r1 rò r rÎ r r r» )rF r© rá rø Úszr= r= r> Útest_export_recordØ sJ ìý z(TestNewBufferProtocol.test_export_recordc C sr t jddgddggfdgd}t|}t|jd t|jd t|jd t|jd t|jd t|j d
|
d S ) Nr# r" rÌ r )rD )r¿ r r z T{(2,2)i:a:}r= r r® r% rs r= r= r> Útest_export_subarray s z*TestNewBufferProtocol.test_export_subarrayc C s t jdddgdd}t|}tjdkr4t|jd n t|jd t jdddgdd}t|}tjdkrtt|jd n t|jd d S ) Nr# r" rÌ z>ir r8 r¿ z<i)r' ræ r r; r< r rò rs r= r= r> Útest_export_endian s
|
|
z(TestNewBufferProtocol.test_export_endianc C s$ t ttjt d¡d d d
d d S )NrH r" )ZSIMPLE)r r) rm Úget_buffer_infor' rC rE r= r= r> Útest_export_flags s
|
þz'TestNewBufferProtocol.test_export_flagsr! r{ r# r" r ræ r r c C sP t |¡ t |d¡ W 5 Q R X t |d¡ t |¡}t |¡}t|| d S )N)Ú
|
STRIDED_ROZFORMAT)r0 )r¨ r rm r. r rt rs r )rF r! r{ Z
|
pickle_objrL r= r= r> Ú!test_export_and_pickle_user_dtype! s
|
|
z7TestNewBufferProtocol.test_export_and_pickle_user_dtypec C s6 t dD ](}tjddgdt|fid}| |¡ qd S )Nr rh rr r r )rq r' ræ rÝ r )rF rñ rá r= r= r> Ú test_padding0 s z"TestNewBufferProtocol.test_paddingc C sP t rt tjj¡}t d¡}t|}t |¡}t rJt tjj¡}t || ~d S )Nr )
|
r r; rÀ r' re Ú _internalr* r rX r )rF Zcount_1rD r¦ r³ Zcount_2r= r= r> Útest_reference_leak5 s
|
|
|
z)TestNewBufferProtocol.test_reference_leakc C sÄ t jdddt d¡fdgdd}t j|jt jd |¡}| |¡ t jddd d
|
gdd}t j|jt jd |¡}| |¡ t ddd d d dt jdddfg¡}t j|jt jd |¡}| |¡ d S )Nr) rù rù rú r< TrJ r r r= rû rü rý )r' r rC r1 r³ rW r )rF rF r= rG r¼ Zdt3r> r= r= r> Útest_padded_struct_array@ s( þ
|
|
þ
|
ÿÿz.TestNewBufferProtocol.test_padded_struct_arrayz$leaks buffer info cache temporarily.rn )r# rB rB r- c C s¬ d|_ tt|j dk t ¡ }| |j¡ |j}tt|j dk t d¡}|j
|
j r¨t |dg¡\}}t|d dk tj dd d
|
}t |d g¡\}}t|d dk d S ) N)r% r¸ r )é r¸ r )r r¸ r6 rÀ Ú F_CONTIGUOUSr r rö r r& Ú C_CONTIGUOUSr% )rÎ r r r# r$ r] r$ r r' r rJ r rm r. )rF r³ r r% rS r7 rÎ r= r= r> Útest_relaxed_stridesS s&
|
ÿ ÿz*TestNewBufferProtocol.test_relaxed_stridesrö r r& z8Test is unnecessary (but fails) without relaxed strides.rÀ c C sN t dD ]@}t |dg¡\}}|dks*tt |dg¡\}}|dkstqdS )aE Test that alternating export of C- and F-order buffers from
|
an array which is both C- and F-order when relaxed strides is
|
active works.
|
This test defines array in the signature to ensure leaking more
|
references every time the test is run (catching the leak with
|
pytest-leaks).
|
rB r7 )r r r8 )r¸ r N)rq rm r. r~ )rF rS r¿ rà r¾ r= r= r> Ú%test_relaxed_strides_buffer_info_leakl s
|
z;TestNewBufferProtocol.test_relaxed_strides_buffer_info_leakc C sN t tddgddgddgdd¡}t d|¡}tt t| W 5 Q R X d S ) NrM rÅ rÇ r r r ©r2 r1 r3 r1 r# )r' r r" r2 r r) r )rF r© rS r= r= r> Útest_out_of_order_fields} s ü
|
z.TestNewBufferProtocol.test_out_of_order_fieldsc C s t d¡}| |¡ d S )N) 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 r r¡ r= r= r> Ú test_max_dims s
|
z#TestNewBufferProtocol.test_max_dimsc C s` dd }|dt j}t| }t|jd ttdtj| ~~t dD ]}t
|
¡ dkrF q\qFd S )Nc S s$ |}| d d d
D ] }|| }q|S )Nr% r= )r7 Z scalar_typer÷ rØ r= r= r> Ú
|
make_ctype s
|
zBTestNewBufferProtocol.test_error_too_many_dims.<locals>.make_ctype)!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Ï r ) r* Úc_uint8r r rÏ r ré r' ræ rq ÚgcZcollect)rF r> Zc_u8_33drÍ r¿ r= r= r> Útest_error_too_many_dims s
|
þ z.TestNewBufferProtocol.test_error_too_many_dimsc C s4 t t t ¡ ¡}td|jk ttdtj | d S )Nú&z format string)
|
r r* Úpointerr@ r rò r r) r' ræ )rF rÍ r= r= r> Útest_error_pointer_type¦ s þz-TestNewBufferProtocol.test_error_pointer_typec C sP t jd }tt}t | ¡ W 5 Q R X |j}ttd |j W 5 Q R X d S )Nr z(Unrepresentable .* 'u' \(UCS-2 strings\))
|
r* Úc_wcharr r) r' ræ Ú exceptionr r± Ú __cause__)rF r÷ Úcmr r= r= r> Ú test_error_message_unsupported¯ s
|
|
þz4TestNewBufferProtocol.test_error_message_unsupportedc
|
C sR t jt jt jhD ]<}|d}tjdd t ddt¡ t |¡ W 5 Q R X qd S )Nr_ Tr r ú .*\bctypes\b)
|
r* Úc_intÚc_longÚ
|
c_longlongr r r rî r' rX )rF Z c_integerrú r= r= r> Ú"test_ctypes_integer_via_memoryview½ s
|
z8TestNewBufferProtocol.test_ctypes_integer_via_memoryviewc C s G dd dt j}|ddd}tjdd t dd t¡ t |¡}W 5 Q R X t|d
|
d t|d d d |_ t|d
|
d d S ) Nc @ s e Zd ZdejfdejfgZdS )zDTestNewBufferProtocol.test_ctypes_struct_via_memoryview.<locals>.foorD r¦ N)rN rO rP r* r@ Úc_uint32Ú_fields_r= r= r= r> r, Ç s r, r# r" r) Tr r rK rD r¦ rÌ )
|
r* Ú Structurer r r rî r' rX r rD )rF r, r× rS r= r= r> Ú!test_ctypes_struct_via_memoryviewÅ s z7TestNewBufferProtocol.test_ctypes_struct_via_memoryviewrÌ r r= c C sJ t |¡ t|}tjtd| dd t| W 5 Q R X t |¡ dS )a
|
If a user extends a NumPy array before 1.20 and then runs it
|
on NumPy 1.20+. A C-subclassed array might in theory modify
|
the new buffer-info field. This checks that an error is raised
|
if this happens (for buffer export), an error is written on delete.
|
This is a sanity check to help users transition to safe code, it
|
may be deleted at any point.
|
z.*z appears to be C subclassedr N)rm Zcorrupt_or_fix_bufferinforY r¨ r ré r )rF r! r r= r= r> Ú-test_error_if_stored_buffer_info_is_corruptedÔ s
|
|
ÿzCTestNewBufferProtocol.test_error_if_stored_buffer_info_is_corruptedc
|
C sÎ z dd l }W n tk
|
r* t d¡Y nX dD ]}ttt |¡}|j||d|j d}d}tj
|
t |d t |¡ W 5 Q R X tj
|
t |d t |g¡ W 5 Q R X t
|
t ¡ t |¡ W 5 Q R X q0d S )Nr z _testbuffer is not available)rä r¯ r¿ )rò rJ z,NumPy currently does not support.*suboffsetsr )Ú _testbufferÚ ImportErrorr¨ ÚskiprÎ rq r' rÉ r6 ZND_PILr rÈ rX rc )rF rU r7 r$ rß r r= r= r> Útest_no_suboffsetsç s
|
ÿ z(TestNewBufferProtocol.test_no_suboffsetsN)/rN rO rP rä r r} r r" r# r$ r' r( r) r+ r, r- r/ r¨ r© r« rå r' ræ r r) rô r1 r2 r4 r5 Zvalgrind_errorr r9 rª rJ r r: r< r= rç rB rE rJ rO rS rT rX r= r= r= r> r + sP F
|
|
|
*
|
þ
|
ÿ
|
$
|
r c @ sÀ e Zd ZG dd dZdejjejgZdejj ej
|
gZ dd Z dd Z d d
|
Zd d Zd d Zej de d¡e d¡ d¡g¡ej ddddg¡ej dddddg¡dd Zd d ZdS ) Ú TestArrayCreationCopyArgumentc @ s e Zd Zdd ZdS )z)TestArrayCreationCopyArgument.RaiseOnBoolc C s t d S rR r¼ rE r= r= r> Ú__bool__ s z2TestArrayCreationCopyArgument.RaiseOnBool.__bool__N©rN rO rP rZ r= r= r= r> Ú RaiseOnBoolþ s r\ TFc
|
C s² t jd D ]¢}t jd|d}|d }| d¡}ttt j|t jjd ttt j|t jjd ttt j|| ¡ d ttt
|
j dg t t¡ t j|t jt jjd W 5 Q R X q
|
d S )Nr¯ r= r r r r# )r r )r' r r* rb r r) ræ Ú _CopyModeÚNEVERr\ rm Znpy_ensurenocopyr¨ r r¸ )rF r rS r Zpyscalarr= r= r> Ú test_scalars s$
|
|
ÿ
|
ÿ
|
ÿÿ z*TestArrayCreationCopyArgument.test_scalarsc C sD ddd}| D ],}|dD ] }t jd|d}| jD ]2}t j|||d}||k r\|jjs`tt|| q8||krÒ| jD ]*}t j|||d}||ksz|j |ksztqzt j|t j
|
j |d}||ksÐ|j |ksÐtq | jD ]6}t j|||d}||k r |jjstt|| qØt t t j|t j
|
j |d t t t j|d |d q qd S ) NFc s sB t jd t jd }|D ]$}t |¡V | rt |¡ ¡ V qd S )NZIntegerZUnsignedInteger)r' r r r )r Ú int_typesZint_typer= r= r> r` s ÿ zETestArrayCreationCopyArgument.test_compatible_cast.<locals>.int_typesTrB r )r r )F)r' rC Ú true_valsræ rJ r} r~ r Ú
|
false_valsrv r] r^ r r) )rF r` Zint1Zint2rS r rL r= r= r> Útest_compatible_cast s>
|
|
|
þ
|
þ þz2TestArrayCreationCopyArgument.test_compatible_castc C s t d¡}t|}| jD ]"}t j||d}t ||¡rtq| jD ]"}t j||d}t ||¡sBtqBt j|t jj d}t ||¡std S )NrB r )
|
r' rC r ra ræ rÑ r~ rb r] r^ )rF rS rW r rL r= r= r> Útest_buffer_interfaceH s
|
|
|
z3TestArrayCreationCopyArgument.test_buffer_interfacec sv t d¡ G fddd}| }dt jjd fd|ft jj|ft jj|ffD ]$\}}t j||d}|j|ksLtqLd S )NrB c s e Zd Z jZdS )zFTestArrayCreationCopyArgument.test_array_interfaces.<locals>.ArrayLikeN©rN rO rP r4 r= ©Zbase_arrr= r> Ú ArrayLike] s rg )TNFr ) r' rC r] ÚALWAYSÚ IF_NEEDEDr^ ræ rv r~ )rF rg rS r rp rL r= rf r> Útest_array_interfacesY s
|
|
|
þ z3TestArrayCreationCopyArgument.test_array_interfacesc s® t d¡ G fddd}| }| jD ](}t j||d}t| | k s(tq(| jD ](}t j|dd}t| | ksXtqXt t ¡ t j|t j
|
j d W 5 Q R X d S )NrB c s e Zd Z fddZdS )z?TestArrayCreationCopyArgument.test___array__.<locals>.ArrayLikec s S rR r= rE rf r= r> r l s zITestArrayCreationCopyArgument.test___array__.<locals>.ArrayLike.__array__Nr r= rf r= r> rg k s rg r F) r' rC ra ræ r r~ rb r¨ r r) r] r^ )rF rg rS r rL r= rf r> rÑ h s
|
|
|
|
|
z,TestArrayCreationCopyArgument.test___array__rS r= éQ )rí rí Úorder1r r NÚorder2rÑ rÍ c C s¨ | |¡}|dkr |jjsdtnD|dkr6|jjsdtn.|jdkrd|d d d
d d d
f }|jjrdt|dkrv|jj}n|dkr|jj}nd}|t|fD ]}| jD ]2}t j
|
|||d}||k rÈ|jj sÌtt || q¤|rN| j D ]0}t j
|
|||d}tsä||ksä|jj|ksätqät j
|
|t jj|d}ts¢||ks¢|jj|ks¢tq| j D ] }t j
|
|||d}t || qTttt j
|
|t jj|d ttt j
|
|d |d qd S )Nr r r r" Tr )r rJ r r~ r rÏ r r ra r' ræ r} r rb r rv r! r] r^ r r) )rF rS rl rm Zno_copy_necessaryrW r rL r= r= r> Útest_order_mismatch sR
|
|
|
|
|
|
ÿ
|
þ þz1TestArrayCreationCopyArgument.test_order_mismatchc C s t dddgdddgg¡}ttt j|jt jjdd ttt j|jt jjdt jd ttt j|t jjd d ttt j|t jjd t jd d S )
|
Nr# r" r rÌ rH r r )r r r r )r' ræ r r) r r] r^ r¸ rT r= r= r> Útest_striding_not_ok¾ s. þ þ þ þz2TestArrayCreationCopyArgument.test_striding_not_ok)rN rO rP r\ r' r] rh rn ra ri rm rb r_ rc rd rj rÑ r¨ r© r« r rC rÆ rn ro r= r= r= r> rY ü s . ÿ7rY c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestArrayAttributeDeletionc C sV t d¡}dddddddg}t *}| td ¡ |D ]}ttt|| q4W 5 Q R X d S )
|
Nr" r7 rÎ r$ r rx ry rb z Assigning the 'data' attribute)r' r r rí r r r/ Údelattr)rF rD Úattrrð r¾ r= r= r> Ú,test_multiarray_writable_attributes_deletionÐ s
|
zGTestArrayAttributeDeletion.test_multiarray_writable_attributes_deletionc C sB t d¡}dddddddd d
|
d d d g }|D ]}ttt|| q*d S )Nr" rÏ rJ r1 r9 rÐ rv r* r r4 r r r8 )r' r r r/ rq ©rF rD rr r¾ r= r= r> Ú0test_multiarray_not_writable_attributes_deletionÙ s
|
þzKTestArrayAttributeDeletion.test_multiarray_not_writable_attributes_deletionc C s4 t d¡j}ddddg}|D ]}ttt|| q d S )Nr" r Z updateifcopyr rK ©r' r rJ r r/ rq rt r= r= r> Ú1test_multiarray_flags_writable_attribute_deletioná s zLTestArrayAttributeDeletion.test_multiarray_flags_writable_attribute_deletionc C sB t d¡j}dddddddd d
|
d d g }|D ]}ttt|| q*d S ) Nr" rz r r r% r} r r r r r rÒ rv rt r= r= r> Ú5test_multiarray_flags_not_writable_attribute_deletionç s þzPTestArrayAttributeDeletion.test_multiarray_flags_not_writable_attribute_deletionN)rN rO rP rs ru rw rx r= r= r= r> rp Î s rp c @ s´ e Zd ZG dd dZedZej dei dfegi dgfeegi ddgfeddidfeddiefeddidggfedd ie fed
|
didfed
|
d ie fed
|
d iefg
|
¡d d Z
|
dS )ÚTestArrayInterfacec @ s( e Zd Zdd Zdd Zedd ZdS )zTestArrayInterface.Fooc C s || _ ddi| _d S ©NÚtypestrr )rú Úiface)rF rú r= r= r> rU ò s z TestArrayInterface.Foo.__init__c C s
|
t | jS rR )rÿ rú rE r= r= r> Ú __float__ö s z TestArrayInterface.Foo.__float__c C s | j S rR )r| rE r= r= r> r4 ù s z*TestArrayInterface.Foo.__array_interface__N)rN rO rP rU r} r. r4 r= r= r= r> ÚFooñ s r~ rö zval, iface, expectedr7 r= Nri rr rÎ r® c C s ddi| j _| j j |¡ tr.t t d¡¡}t|t rHt
|
|tj | n*t |¡}t t |¡| |jdkspt ~trt t d¡¡}t || d S rz )r× r| Úupdater r; rÀ r' r ru rY r ræ r r~ )rF rp r| râ Zpre_cntr Zpost_cntr= r= r> Útest_scalar_interface ! s
|
|
z(TestArrayInterface.test_scalar_interface) rN rO rP r~ r× r¨ r© r« rô r) r r= r= r= r> ry ð s ö ry c C s$ G dd d} t t | ¡d d S )Nc @ s e Zd Ze d¡ZejZdS )z*test_interface_no_shape.<locals>.ArrayLiker# N)rN rO rP r' ræ r4 r= r= r= r> rg ! s
|
rg r# )r r' ræ )rg r= r= r> Útest_interface_no_shape ! s r c C s` t ddgddgddgdd¡} t jd| d }t |jd
|
¡}t |jd ¡}t|j|j d S ) NrÑ r[ rì r r r® rù rB r Údescrr{ )r' r r r4 r r1 )Zmy_dtyperD Zdescr_tZ typestr_tr= r= r> Ú test_array_interface_itemsize$! s ÿr c s t dddg¡} t| j d d< G fddd}t | d ¡ d < G fd
|
d d }t | ¡}t | ¡}| d d
d¡}t|| t|| d S ) Nr# r" rÌ r= r7 c s e Zd Z ZdS )z5test_array_interface_empty_shape.<locals>.DummyArray1Nre r= )Ú
|
interface1r= r> Ú DummyArray14! s r
r r$ c s e Zd Z ZdS )z5test_array_interface_empty_shape.<locals>.DummyArray2Nre r= )Ú
|
interface2r= r> Ú DummyArray2>! s r )r' ræ r" r4 rá rX rÆ r )rS r
r Úarr1Zarr2Zarr3r= )r r r> Ú test_array_interface_empty_shape.! s
|
|
r c sn t jdddgdd} t| j t| d< d d< d d
|
< G fd d d }t | ¡}t|| dd
d S ) Nr# r" rÌ r¶ r r$ rr r7 r r< c s e Zd Z ZdS )z/test_array_interface_offset.<locals>.DummyArrayNre r= ©Z interfacer= r> Ú
|
DummyArrayO! s r )r' ræ r" r4 r rX r )rS r r r= r r> Útest_array_interface_offsetG! s
|
r c s^ t jdddgdd} t| j d d< G fdd d }t t¡ t | ¡ W 5 Q R X d S )
|
Nr# r" rÌ r¶ r u âr{ c s e Zd Z ZdS )z8test_array_interface_unicode_typestr.<locals>.DummyArrayNre r= r r= r> r Z! s r )r' ræ r" r4 r¨ r rô rX )rS r r= r r> Ú$test_array_interface_unicode_typestrU! s
|
r c C sP t d¡j} z| d= | dd
= W n* tk
|
r4 Y n tk
|
rJ tY nX d S )NrÌ r# r" )r' r rb rô rè r~ )rÓ r= r= r> Útest_flat_element_deletiona! s r c C s( t jdddgd} tt| d jd d S )Nr" )rá rÝ )rø rÝ r r rá )r' r* r r) Ú __delitem__r r= r= r> Ú test_scalar_element_deletionl! s r c @ s e Zd Zdd ZdS )Ú TestMapIterc C sÄ t d¡ d¡ t¡}ddddgddddgf}dddd g}t |||¡ t|d
|
d d d gddddgddddgg t d¡ t¡}t dddg¡f}dddg}t |||¡ t|ddddddg d S ) Nr+ rÐ r# r" r rÌ ro rÍ r® rü rý r g 3@g Z@rD r r r rm g D@g &@rÄ r gfffffY@g I@rà rò ) r' rC rÆ rE rÿ rm Ztest_inplace_incrementr ræ )rF rD r¾ r_ r¦ r= r= r> Ú test_mapiterr! s
|
|
ÿ
|
|
þ
|
zTestMapIter.test_mapiterN)rN rO rP r r= r= r= r> r q! s r c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestAsCArrayc C s. t jdt jd}t |d¡}t|d | d S )NrÄ r rÌ )r' rC rè rm Útest_as_c_arrayr ©rF ræ Zfrom_cr= r= r> Ú test_1darray! s zTestAsCArray.test_1darrayc C s8 t jdt jd dd¡}t |dd¡}t|d | d S )NrÄ r rÌ r r" r )r" r ©r' rC rè rÆ rm r r r r= r= r> Ú test_2darray! s zTestAsCArray.test_2darrayc C s< t jdt jd ddd¡}t |ddd¡}t|d | d S )NrÄ r r" rÌ r r# r5 r r r= r= r> Ú test_3darray! s zTestAsCArray.test_3darrayN)rN rO rP r r r r= r= r= r> r ! s r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestConversionc C s² t jd D ]¨}tdt jd|dkd|f tdt jd|dk d|f t jd D ]Z}tt jd|dt jd|dkd||f tt jd|dt jd|dk d||f qVq
|
dD ]ì}tdt jd|dk d|f tdt jd|dk d|f tdt jd|dkd|f d D ]}tt jd|dt jd|dkd||f tt jd|dt jd|dk d||f tt jd|dt jd|dkd||f q q¸d
|
t jd D ]ø}tdt jd|dkd|f tdt jd|dk d|f tdt jd|dkd|f d
|
t jd D ]}tt jd|dt jd|dkd||f tt jd|dt jd|dk d||f tt jd|dt jd|dkd||f q"q´d S ) Nr
r# r r ztype %s failedztype %s and %s failedZBHILQPr% ZbhilqpZbhlqpr )r' r r ræ )rF rF rG r= r= r> Ú&test_array_scalar_relational_operation! sL
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
|
ÿ
|
ÿ
|
ÿz5TestConversion.test_array_scalar_relational_operationc C sÐ t tt dg¡d t tt dg¡d t tt dgg¡d tttt ddg¡ G dd d}tttt | ¡ tttt | g¡ tr t d¡ t d g¡}||d < t
|
}t|t| d |d < d S )
|
NFTr_ r# r" c @ s e Zd Zdd ZdS )z:TestConversion.test_to_bool_scalar.<locals>.NotConvertiblec S s t d S rR r° rE r= r= r> rZ É! s zCTestConversion.test_to_bool_scalar.<locals>.NotConvertible.__bool__Nr[ r= r= r= r> ÚNotConvertibleÈ! s r z"Pyston disables recursion checkingr ) r r r' ræ r r) r± r r¨ rW ÚRecursionError)rF r Zself_containingÚErrorr= r= r> Útest_to_bool_scalarÂ! s
|
z"TestConversion.test_to_bool_scalarc C s0 t dd f}|D ]}t|t d¡d t|t dg¡d t|t dgg¡d tt|t ddg¡ td|t d¡ td |t d
|
¡ td |t d ¡ tj d k ròG dd d}td|t | ¡ td|t | g¡ n G dd d}tt
|
|t | ¡ tt
|
|t | g¡ qd S )Nc S s | ¡ S rR )Ú__int__rà r= r= r> r4 Û! r5 z3TestConversion.test_to_int_scalar.<locals>.<lambda>r r# r_ r" r Ú4rH ó 5rÄ Ú6)rÌ r c @ s e Zd Zdd ZdS )z3TestConversion.test_to_int_scalar.<locals>.HasTruncc S s dS r¼ r= rE r= r= r> Ú __trunc__ë! s z=TestConversion.test_to_int_scalar.<locals>.HasTrunc.__trunc__N)rN rO rP r¤ r= r= r= r> ÚHasTruncê! s r¥ rÌ c @ s e Zd Zdd ZdS )z9TestConversion.test_to_int_scalar.<locals>.NotConvertiblec S s t d S rR r° rE r= r= r> r ó! s zATestConversion.test_to_int_scalar.<locals>.NotConvertible.__int__N)rN rO rP r r= r= r= r> r ò! s r ) rÝ r r' ræ r rô rv rt r; Ú version_infor± )rF Z int_funcsZint_funcr¥ r r= r= r> Útest_to_int_scalarÙ! s,
|
|
|
ÿ ÿz!TestConversion.test_to_int_scalarN)rN rO rP r r r§ r= r= r= r> r ! s )r c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dS )Ú TestWherec
|
C sö t tjtjtjtjtjtjtjg}|D ]È}tj dt d}t
|
t ||d|d¡|d t
|
t | |d|d¡|d t
|
t d|d|d¡|d t
|
t d|d|d¡|d t |¡ |¡}t |¡}| |¡}d|d< |d |d< t
|
t |||¡| t
|
t |||¡| t
|
t |||d ¡| t
|
t ||d |¡| t
|
t |d d d
|d d d
|d d d
¡|d d d
t
|
t |dd d
|dd d
|dd d
¡|dd d
t
|
t |d d d
|d d d
|d d d
¡|d d d
t
|
t |dd d
|dd d
|dd d
¡|dd d
t
|
t |d d d
|
|d d d
|
|d d d
|
¡|d d d
|
t
|
t |d d d
|d d d
|d d d
¡|d d d
t
|
t |dd d
|dd d
|dd d
¡|dd d
q&d S ) Né5 r r r# TFrì r" rÌ rã rå )r r' r´ r¶ r¸ rè rÇ rD rê r r r© Z ones_likerE r² )rF Zdtsr© r³ rÕ rë rê r= r= r> r ü! s4 ÿ
|
"
|
|
<<<<<<zTestWhere.test_basicc C s0 t t dd d ¡t d ¡ tjg td dd¡}tjg tjd dd¡}t t |d|¡t g ¡ dd¡ t ddddd d
|
d d d dddddddddddg¡}td}tjdd|d ||d d ||d ||d ||d!|d"d#gtd}tjddd$dd$d$ddd$d$dd$d$dd$d$dd$ddgtd}|d d
}|t |¡ |t |¡< t t |||¡| |d d
}|t | ¡ |t | ¡< t t |||¡| t t |||¡| tjd%d>j d}td}t
|
t d||¡j tj td'}t
|
t d||¡j tj td(}t
|
t d||¡j tj td)}t
|
t d||¡j tj d S )*NTr r rÌ gq=
|
×£põ¿g{®GázÄ¿gHáz®Gá¿g×£p=
|
×Ó¿g{®Gáz´¿gffffffî¿rü gÕxé&1Ô?g/Ý$á?g
|
×£p=
|
Ç¿gÕxé&1ì?ghí|?5Î?g/Ý$ÿ?g×£p=
|
×Ó?gí|?5^æ?ghí|?5ð?gÀÊ¡Eô?g Zd;ßOÍ?g=
|
×£p=ö¿g^ºI +ß?ÚNaNZ5zZ0lZWzZXqÚcsZQNZFdZkpZ36rî r# rý r ÚInfinityz -Infinityg¯P.5_) r r' r© ræ r rÆ rÈ rÿ r( rÅ r r )rF rÍ r¦ rÕ rÚ rë rê r= r= r> Ú test_exotic" s~ " þ ÿþ ÿÿ zTestWhere.test_exoticc C sÄ ddg}t d¡}t d¡}t t |¡d d
t jf ||¡}t|d |d t|d |d |j}|j}t |||¡}t|d d
df |d d
df t|d d
df |d d
df d S )NTF)r" r9 r r# )r' r* r r© ræ rk r r )rF r³ rD r¦ rê r= r= r> Ú test_ndim?" s
|
|
""zTestWhere.test_ndimc C s t ddddddddddddg ¡}t d¡}t jdddddd dd
|
d d ddg t jd }t jdd dddd d d
|
d d d dg t jd }tt |||¡| | t j¡}| t j¡}tt |||¡| | t ¡}d||dk< tt |||¡| |dk}d||dk< d||< tt |||¡| d S )NFTr# rD rü rà r r# ç Àç $Àr rý r i
|
r iìau)
|
r' ræ r· rÈ r r© rE rÅ r¸ rÝ )rF r³ rD r¦ rê Ztmpmaskr= r= r> Útest_dtype_mixM" s0 ÿ
|
ÿ ÿ
|
zTestWhere.test_dtype_mixc C sò t ddddddddddddg ¡}t jdddddddd d
|
dddg t jd }t jd d d }t jdddddddd d
|
dddg t jd }tt |||¡| | d¡}tt |||¡| | d¡}tt |||¡| | d ¡}tt |||¡| d S )NFTrD rý rà r r# r¯ r° r r r# rN rü rý rM )r' ræ rÈ r r r© rE )rF r³ rê rD r¦ r= r= r> Ú test_foreigne" s( ÿ ÿ ÿ
|
|
|
zTestWhere.test_foreignc C sH ddg}t d¡}t d¡}ttt j||| ttt j|d || d S )NTrË rÖ r )r' r r r) r© )rF r³ rD r¦ r= r= r> Ú
|
test_errorx" s
|
|
|
zTestWhere.test_errorc C s| t d¡}t d¡}tt d||¡d tt d||¡d t d¡}t d¡}tt d||¡d tt d||¡d d S )NrF Añ xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxTFZabcdZxxxxxxxx)r' ræ r r© rW r= r= r> r¨ " s
|
|
|
|
zTestWhere.test_stringc C sB t d¡}t t |dk¡¡}t|t t jg g gt jd¡ d S )Nri g ÀX@r )r' r* Zvstackr© r Z
|
atleast_2dræ r )rF rá Zibadr= r= r> Útest_empty_result" s
|
|
ÿzTestWhere.test_empty_resultc C sT ddddddg}t j d¡ t jj| }tdD ] }| ¡ }| ¡ }t|| q0d S )NrB r" rÌ r rH rÄ )r' r rª r rq Znonzeror )rF r7 ræ r¿ Z benchmarkr r= r= r> Ú test_largedim" s zTestWhere.test_largedimN) rN rO rP r r r® r± r² r³ r¨ r´ rµ r= r= r= r> r¨ û! s ) r¨ c @ s` e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
e dd Z dd Z dS )Ú
|
TestSizeOfc C s t g ¡}tt |¡dk d S rÜ )r' ræ r r; Ú getsizeofr r= r= r> Útest_empty_array¤" s
|
zTestSizeOf.test_empty_arrayc C s< |dj }dD ](}tj||d}tt |¡|| k qd S )Nr )rB ro rµ iô r )r1 r' rC r r; r· )rF r Z elem_sizer_ rá r= r= r> Ú check_array¨" s
|
zTestSizeOf.check_arrayc C s | tj¡ d S rR )r¹ r' r¶ rE r= r= r> Útest_array_int32¯" s zTestSizeOf.test_array_int32c C s | tj¡ d S rR )r¹ r' r¸ rE r= r= r> Útest_array_int64²" s zTestSizeOf.test_array_int64c C s | tj¡ d S rR )r¹ r' rÅ rE r= r= r> Útest_array_float32µ" s z TestSizeOf.test_array_float32c C s | tj¡ d S rR )r¹ r' rÈ rE r= r= r> Útest_array_float64¸" s z TestSizeOf.test_array_float64c C s* t d¡}tt |d ¡t |¡k d S )Nrµ .)r' r r r; r· r! r= r= r> r »" s
|
zTestSizeOf.test_viewc C s4 t d¡}tt |¡t | ddd¡ ¡ ¡k d S )Nrµ r# )r' r r r; r· rÆ r r! r= r= r> rÑ ¿" s
|
zTestSizeOf.test_reshapec C sP t d¡}t |¡}| d¡ t|t |¡k | d¡ t|t |¡k d S )Nrµ ro é )r' r r; r· r r )rF rÕ Úoldr= r= r> r Ã" s
|
|
|
|
zTestSizeOf.test_resizec C s t d¡}tt|jd d S )Nrµ rD )r' r r rô Ú
|
__sizeof__r! r= r= r> r³ Ì" s
|
zTestSizeOf.test_errorN)rN rO rP r¸ r¹ rº r» r¼ r½ r rÑ r r r³ r= r= r= r> r¶ ¢" s
|
r¶ c @ s e Zd Zdd Zdd ZdS )Ú TestHashingc C s t d¡}ttt| d S r¼ )r' r r rô rº r r= r= r> Útest_arrays_not_hashableÓ" s
|
z$TestHashing.test_arrays_not_hashablec C s" t g ¡}tt|tjj d S rR )r' ræ r ru Ú collectionsrF ÚHashabler r= r= r> Útest_collections_hashable×" s
|
z%TestHashing.test_collections_hashableN)rN rO rP r rÅ r= r= r= r> rÁ Ñ" s rÁ c @ s² e Zd ZeZejejejejej ej
|
ej ej ej ejejejej ejejejejejejgZG dd dejZG dd dejZG dd dZ dd Z d d
|
Z d d Z d d Z dS )ÚTestArrayPriorityc @ s e Zd ZdZdd ZdS )zTestArrayPriority.FoorÌ c O s t j|| | ¡S rR ©r' ræ rW ©r6 r r r= r= r> r7 é" s z TestArrayPriority.Foo.__new__N©rN rO rP r r7 r= r= r= r> r~ æ" s r~ c @ s e Zd ZdZdd ZdS )zTestArrayPriority.Barg @Y@c O s t j|| | ¡S rR rÇ rÈ r= r= r> r7 ï" s z TestArrayPriority.Bar.__new__NrÉ r= r= r= r> ÚBarì" s rÊ c @ s e Zd ZdZdd Ze ZZe ZZe Z Z
|
e Z Z e Z Ze ZZe ZZe ZZe ZZe ZZe ZZe ZZ e Z Z eZ eZ eZ!eZ"eZ#eZ$dS )zTestArrayPriority.Otherg @@c C s | ¡ S rR ©r r r= r= r> Ú_allõ" s z TestArrayPriority.Other._allN)%rN rO rP r rÌ r` r- ra Ú__rsub__rI rb rM rN rJ Z__rdiv__Ú__mod__Ú__rmod__Ú __truediv__Ú __rtruediv__Ú __floordiv__Ú __rfloordiv__Ú__and__Ú__rand__Ú__xor__Ú__rxor__Ú__or__Ú__ror__Ú
|
__lshift__Ú __rlshift__Ú
|
__rshift__Ú __rrshift__r r r r r
|
r r= r= r= r> ÚOtherò" s* rÞ c C sd t ddg¡}| ddg¡}| jD ]<}t|}tt|||| j| tt|||| j| q"d S ©Nr# r" )r' ræ rÊ Ú
|
binary_opsr6 r ru ©rF rD r¦ r× r r= r= r> Útest_ndarray_subclass # s
|
z'TestArrayPriority.test_ndarray_subclassc C s^ t ddg¡}| ¡ }| jD ]<}t|}tt|||| j| tt|||| j| q d S rß )r' ræ rÞ rà r6 r ru rá r= r= r> Útest_ndarray_other# s
|
z$TestArrayPriority.test_ndarray_otherc C sd | ddg¡}| ddg¡}| jD ]<}t|}tt|||| j| tt|||| j| q"d S rß )r~ rÊ rà r6 r ru rá r= r= r> Útest_subclass_subclass # s
|
z(TestArrayPriority.test_subclass_subclassc C s^ | ddg¡}| ¡ }| jD ]<}t|}tt|||| j| tt|||| j| q d S rß )r~ rÞ rà r6 r ru rá r= r= r> Útest_subclass_other$# s
|
z%TestArrayPriority.test_subclass_otherN)!rN rO rP r/ r rþ ru rù r0 rû rú rü Úand_Úor_r rÿ r ÚgtÚgeÚltÚleÚneÚeqrà r' r6 r~ rÊ rÞ râ rã rä rå r= r= r= r> rÆ Ü" s8 ýrÆ c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestBytestringArrayNonzeroc C s t tjdgtd d S ©Nr` r )r r' ræ rÚ rE r= r= r> Ú"test_empty_bstring_array_is_falsey/# s z=TestBytestringArrayNonzero.test_empty_bstring_array_is_falseyc C s& t jdgtd}d|d< t| d S )Nr© r ú r ©r' ræ rÚ r r¡ r= r= r> Ú'test_whitespace_bstring_array_is_falsey2# s zBTestBytestringArrayNonzero.test_whitespace_bstring_array_is_falseyc C s& t jdgtd}d|d< t| d S )Nr© r ú r rò r¡ r= r= r> Ú%test_all_null_bstring_array_is_falsey7# s z@TestBytestringArrayNonzero.test_all_null_bstring_array_is_falseyc C s$ t jdgtd}d|d< t| d S )Nr© r ú r rò r¡ r= r= r> Ú(test_null_inside_bstring_array_is_truthy<# s zCTestBytestringArrayNonzero.test_null_inside_bstring_array_is_truthyN)rN rO rP rð ró rõ r÷ r= r= r= r> rî -# s rî c @ s( e Zd ZdZdd Zdd Zdd ZdS ) ÚTestUnicodeEncodingz_
|
Tests for encoding related bugs, such as UCS2 vs UCS4, round-tripping
|
issues, etc
|
c C sX t jddd}tdtjd D ]4}t|}||d< |d |ksBt| ¡ |ks tq dS )z; Tests that GETITEM, SETITEM, and PyArray_Scalar roundtrip r= ZU1rà r# N)r' r* rq r; Ú
|
maxunicoder÷ r~ rb )rF rS r¿ râ r= r= r> Útest_round_tripG# s z#TestUnicodeEncoding.test_round_tripc C s2 t ddg¡}t d¡|d d
< t|ddg d S ©NZaaZbbÚcc)r' ræ rt r ©rF rÔ r= r= r> Útest_assign_scalarQ# s z&TestUnicodeEncoding.test_assign_scalarc C s0 t ddg¡}| t d¡¡ t|ddg d S rû )r' ræ r5 rt r rý r= r= r> Útest_fill_scalarW# s z$TestUnicodeEncoding.test_fill_scalarN)rN rO rP rä rú rþ rÿ r= r= r= r> rø B# s
|
rø c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestUnicodeArrayNonzeroc C s t tjdgtjd d S rï )r r' ræ rt rE r= r= r> Ú"test_empty_ustring_array_is_falsey`# s z:TestUnicodeArrayNonzero.test_empty_ustring_array_is_falseyc C s( t jdgt jd}d|d< t| d S )NÚeggsr rñ r ©r' ræ rt r r¡ r= r= r> Ú'test_whitespace_ustring_array_is_falseyc# s z?TestUnicodeArrayNonzero.test_whitespace_ustring_array_is_falseyc C s( t jdgt jd}d|d< t| d S )Nr r rô r r r¡ r= r= r> Ú%test_all_null_ustring_array_is_falseyh# s z=TestUnicodeArrayNonzero.test_all_null_ustring_array_is_falseyc C s& t jdgt jd}d|d< t| d S )Nr r rö r r r¡ r= r= r> Ú(test_null_inside_ustring_array_is_truthym# s z@TestUnicodeArrayNonzero.test_null_inside_ustring_array_is_truthyN)rN rO rP r r r r r= r= r= r> r ^# s r c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú
|
TestFormatc C s4 t t j¡}td |¡d td |d ¡d d S )Nz{:0.3g}z3.14r= )r' ræ Úpir rò r¡ r= r= r> Útest_0du# s zTestFormat.test_0dc C s& t t jg¡}td |¡t| d S )Nz{})r' ræ r r rò rÚ r¡ r= r= r> Útest_1d_no_formatz# s z TestFormat.test_1d_no_formatc C s t t jg¡}ttdj| d S )Nz{:30})r' ræ r r rô rò r¡ r= r= r> Útest_1d_format~# s zTestFormat.test_1d_formatN)rN rO rP r r
|
r r= r= r= r> r s# s r )r c @ sÞ e Zd Zdd Zdd Zdd Zejjde
|
dd d
|
g¡e
|
d d gd dgg¡e j
|
ddde j de e ddgddgdd gdd¡de j
|
dgede
|
g ¡e d¡ee
|
dd d
|
g¡gdd d d d d d!d"gd#d$d% Zd&d' ZdS )(Ú
|
TestCTypesc C s> t dddgdddgg¡}tt|jj tt|jjd d S rÇ )r' ræ r r* Ú_ctypesr r7 )rF Útest_arrr= r= r> Útest_ctypes_is_available# s z#TestCTypes.test_ctypes_is_availablec C sd ddl m} d |_zDt dddgdddgg¡}tt|jj|j t t
|
|jj d W 5 t|_X d S )
|
Nr )r3 r# r" rÌ r rH rÄ rä ) Z
|
numpy.corer3 r* r' ræ r ru r Z_missing_ctypesr r r7 )rF r3 r r= r= r> Ú test_ctypes_is_not_available# s
|
ÿz'TestCTypes.test_ctypes_is_not_availablec C s d| j _| S )NF)rJ rK rà r= r= r> Ú_make_readonly# s zTestCTypes._make_readonlyrS r# r" rÌ rÅ rÇ rÉ ZfourrÚ rî r rr rM rD r¦ r rÄ r; Nrï Z1dZ2dZ
|
structuredZ overlappingr( r2 zempty-2dr rÏ c C sX | ¡ }t |¡}|j tj¡}~t t| d k d ~trDt t| d kd d S ©Nz,ctypes pointer did not hold onto a referencez3unknowable whether ctypes pointer holds a reference) r rx ry r* Zdata_asÚc_void_pr r r ©rF rS Zarr_refZ
|
ctypes_ptrr= r= r> Ú#test_ctypes_data_as_holds_reference# s
|
z.TestCTypes.test_ctypes_data_as_holds_referencec C sZ t d g¡ ¡ }t |¡}|jj}~t t| d k d ~t rFt t| d kd d S r )
|
r' ræ r rx ry r* Z_as_parameter_r r r r r= r= r> Ú(test_ctypes_as_parameter_holds_referenceÎ# s
|
z3TestCTypes.test_ctypes_as_parameter_holds_reference)rN rO rP r r r r¨ r© r« r' ræ r* r r" r( r2 r r r= r= r= r> r
# s@ üÿñøð
|
r c @ sz e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z e jjdddd Zdd ZdS ) ÚTestWritebackIfCopyc C s: t d¡}t jddd}t j|d|d}t|td d S ©NrH rÜ r r r· )r' r½ r2 r¾ r rq ©rF r
|
rµ rL r= r= r> Útest_argmax_with_outã# s
|
z(TestWritebackIfCopy.test_argmax_with_outc C s< t d¡ }t jddd}t j|d|d}t|td d S r )r' r½ r2 r¿ r rq r r= r= r> Útest_argmin_with_outé# s z(TestWritebackIfCopy.test_argmin_with_outc C sf t d¡ dd¡j}t ||dkddg¡ t|t ddgddgddgg¡ ttt j||dkg d S ) NrÄ r" rÌ é, r r r# r ) r' rC rÆ r Zplacer ræ r r) r¡ r= r= r> Útest_insert_noncontiguousï# s "z-TestWritebackIfCopy.test_insert_noncontiguousc C sP t d¡ dd¡j}t |ddgddg¡ t|t ddgddgddgg¡ d S ) NrÄ r" rÌ r r r r rH )r' rC rÆ r rÉ r ræ r¡ r= r= r> Útest_put_noncontiguous÷# s z*TestWritebackIfCopy.test_put_noncontiguousc C sP t d¡ dd¡j}t ||dk|d ¡ t|t ddgddgddgg¡ d S ) NrÄ r" rÌ r rí r# r® r9 )r' rC rÆ r r) r ræ r¡ r= r= r> Útest_putmask_noncontiguousü# s z.TestWritebackIfCopy.test_putmask_noncontiguousc C sJ t jddd}t jddd}t j|ddg|dd t|t ddg¡ d S )NrÄ rÝ r r" r Úraiser¹ )r' rC r2 r9 r ræ )rF rD rµ r= r= r> Útest_take_mode_raise$ s z(TestWritebackIfCopy.test_take_mode_raisec C sv t dddgdddgdddgg¡}t jddd}ddg}t j|||dd t|t dddgdddgdddgg¡ d S )
|
Nr# r r° rÝ r iöÿÿÿrB r r¹ )r' ræ r2 r» r )rF rD rµ Úchoicesr= r= r> Útest_choose_mod_raise$ s "þz)TestWritebackIfCopy.test_choose_mod_raisec C s( t d¡ dd¡}|jj}| ¡ }~d S )Nrí rÌ )r' rC rÆ r rb r r£ r= r= r> Útest_flatiter__array__$ s z*TestWritebackIfCopy.test_flatiter__array__c C sR t jdtd dd¡}t j|||d}t|t dddgdd d
|
gd d d gg¡ d S )Nrí r rÌ r· ru rf r1 r_ é6 éB éE éZ éo )r' rC rÿ rÆ r¢ r ræ rW r= r= r> Ú test_dot_out$ s z TestWritebackIfCopy.test_dot_outc C s ddl m}m} t d¡ dd¡j}||}t|jj t|j
|
|k d|d< || t |d t|j j dk t |j
|
d d|d< t |d d S )Nr )Únpy_create_writebackifcopyÚ npy_resolverí rÌ r, .rµ )r| r+ r, r' rC rÆ r r rJ r rv r r* r$ )rF r+ r, rS Úarr_wbr= r= r> Útest_view_assign $ s
|
z$TestWritebackIfCopy.test_view_assignz9increments self in dealloc; ignore since deprecated path.rn c C sT t D}| t¡ t d¡ dd¡}|j}t |¡ t |j
|
dksFt W 5 Q R X d S )Nrí rÌ r# ) r r
rî r' rC rÆ r rm Znpy_abuse_writebackifcopyr Úlogr~ )rF rð rS rx r= r= r> Útest_dealloc_warning0$ s
|
|
z(TestWritebackIfCopy.test_dealloc_warningc C s¼ ddl m}m} t d¡ dd¡j}| ¡ }tr:t
|
|¡}||}t |j j t |j|k d|d< || t|| t |jjdk t|jd tr¦t|t
|
|¡ d|d< t|| d S )Nr )r+ Ú npy_discardrí rÌ r, .rµ )r| r+ r1 r' rC rÆ r r r r; rÀ r rJ r rv r r* r$ )rF r+ r1 rS rW Zarr_cntr- r= r= r> Útest_view_discard_refcount:$ s"
|
|
z.TestWritebackIfCopy.test_view_discard_refcountN)rN rO rP r r r r r r! r# r$ r* r. r¨ r© rÎ r0 r2 r= r= r= r> r á# s ÿ
|
r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d dddg¡dd Z dd Z dd Ze j
|
ddddg¡dd Zd S ) Ú
|
TestArangec C s t tdtjdtj d S )Nz size exceededr )r r) r' rC rL rE r= r= r> Ú test_infiniteS$ s þzTestArange.test_infinitec C s t tdtjddtj d S )Nzcannot compute lengthr r# )r r) r' rC rÚ rE r= r= r> Ú test_nan_stepY$ s þzTestArange.test_nan_stepc C sL t ttjddd t ttjddd t ttjddd t ttjddd d S )Nr rB rü r )r ÚZeroDivisionErrorr' rC rE r= r= r> Útest_zero_step_$ s zTestArange.test_zero_stepc C s@ t ttj t ttjdd t ttjdd t ttjdd d S )NrÌ )Ústepr¸ r r )Ústart)r rô r' rC rE r= r= r> Útest_require_rangeg$ s z TestArange.test_require_rangec C sf t jdd}t jddd}t jddd}t|dks8tt|dksHtt|dksXtt|| d S )NrÌ )Ústopr )r9 r; rí rÄ )r' rC r r~ r )rF Z keyword_stopZkeyword_zerotostopZkeyword_start_stopr= r= r> Útest_start_stop_kwargm$ s z TestArange.test_start_stop_kwargc C s t jdtd}t|t jg dd t jddd}t|dg t jddd}t|ddg t jdddd}t|ddg t t¡ t jddd W 5 Q R X d S ) NFr r Tr" rÄ r rÌ )r' rC r r ræ r¨ r rô )rF rL r= r= r> Útest_arange_booleansw$ s z TestArange.test_arange_booleansr r. r¶ Z5ic C sN t |¡}t tt|¡}tjtd| d t j d|d W 5 Q R X d S )Nú'arange\(\) not supported for inputs .* r r" r )
|
r' r ÚreÚescaperÚ rY r¨ r rô rC )rF r Ú
|
DType_namer= r= r> Útest_rejects_bad_dtypes$ s
|
ÿz"TestArange.test_rejects_bad_dtypesc C sJ t ttt d¡j¡}tjt d| d t
|
dd¡ W 5 Q R X d S )NrD r> r r¦ ) r? r@ rÚ rY r' ræ r r¨ r rô rC )rF rA r= r= r> Útest_rejects_strings$ s
|
ÿz TestArange.test_rejects_stringsc C sJ t jdddd}t jdddd}|jdks.t|jdks<tt|| d S )Nr# rp rN r rM )r' rC r r~ r )rF Zres_beZres_ler= r= r> Útest_byteswapped$ s
|
zTestArange.test_byteswappedÚwhichr r# r" c C s^ dddg}t d¡||< t j| jt jks.td g g||< t t¡ t j| W 5 Q R X d S )Nr r# r" r )r' rÈ rC r r~ r¨ r r) )rF rE r r= r= r> Ú test_error_paths_and_promotion¡$ s
|
z)TestArange.test_error_paths_and_promotionN)rN rO rP r4 r5 r7 r: r< r= r¨ r© r« rB rC rD rF r= r= r= r> r3 R$ s
|
|
r3 c @ s@ e Zd ZdZdd Zdd Zdd Zdd Zd
|
d Zd d Z dS )ÚTestArrayFinalizez Tests __array_finalize__ c C s6 G dd dt j}t d¡ |¡}t|j|jk d S )Nc @ s e Zd Zdd ZdS )z7TestArrayFinalize.test_receives_base.<locals>.SavesBasec S s | j | _d S rR )rv Ú
|
saved_baser r= r= r> r8 µ$ s zJTestArrayFinalize.test_receives_base.<locals>.SavesBase.__array_finalize__N©rN rO rP r8 r= r= r= r> Ú SavesBase´$ s rJ r# )r' r6 ræ rW r rH rv )rF rJ rD r= r= r> Útest_receives_base²$ s z$TestArrayFinalize.test_receives_basec C s@ G dd dt j}tjtdd t d¡ |¡ W 5 Q R X d S )Nc @ s e Zd Zedd ZdS )z?TestArrayFinalize.test_bad_finalize1.<locals>.BadAttributeArrayc S s t dd S ©Núboohoo!r= rE r= r= r> r8 ½$ s zRTestArrayFinalize.test_bad_finalize1.<locals>.BadAttributeArray.__array_finalize__N)rN rO rP r. r8 r= r= r= r> ÚBadAttributeArray¼$ s rN z not callabler rB ©r' r6 r¨ r rô rC rW ©rF rN r= r= r> Útest_bad_finalize1»$ s z$TestArrayFinalize.test_bad_finalize1c C s@ G dd dt j}tjtdd t d¡ |¡ W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z?TestArrayFinalize.test_bad_finalize2.<locals>.BadAttributeArrayc S s t dd S rL r= rE r= r= r> r8 Æ$ s zRTestArrayFinalize.test_bad_finalize2.<locals>.BadAttributeArray.__array_finalize__NrI r= r= r= r> rN Å$ s rN ztakes 1 positionalr rB rO rP r= r= r> Útest_bad_finalize2Ä$ s z$TestArrayFinalize.test_bad_finalize2c C s@ G dd dt j}tjtdd t d¡ |¡ W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z?TestArrayFinalize.test_bad_finalize3.<locals>.BadAttributeArrayc S s t dd S rL r= r r= r= r> r8 Î$ s zRTestArrayFinalize.test_bad_finalize3.<locals>.BadAttributeArray.__array_finalize__NrI r= r= r= r> rN Í$ s rN rM r rB )r' r6 r¨ r ré rC rW rP r= r= r> Útest_bad_finalize3Ì$ s z$TestArrayFinalize.test_bad_finalize3c C s¾ G dd dt j}G dd d}t | ¡}t |d ¡}tt}| |¡ W 5 Q R X |jj d }~t
|
t || t t
|
| d k d ~t t
|
| d k d ~t t
|
| d kd d S )
|
Nc @ s e Zd Zdd ZdS )zBTestArrayFinalize.test_lifetime_on_error.<locals>.RaisesInFinalizec S s t | d S rR rË r r= r= r> r8 ×$ s zUTestArrayFinalize.test_lifetime_on_error.<locals>.RaisesInFinalize.__array_finalize__NrI r= r= r= r> ÚRaisesInFinalizeÖ$ s rT c @ s e Zd ZdS )z7TestArrayFinalize.test_lifetime_on_error.<locals>.DummyNrM r= r= r= r> ÚDummyÜ$ s rU r= r z!object should not already be deadz*obj_arr should not hold the last referencezno references should remain) r' r6 ræ rx ry r rè rW rG r r ru r )rF rT rU rQ Zobj_refrë Z obj_subarrayr= r= r> Útest_lifetime_on_errorÔ$ s"
|
z(TestArrayFinalize.test_lifetime_on_errorc C s4 G dd dt j}t d¡ |¡}t|jd k d S )Nc s e Zd Z fddZ ZS )z;TestArrayFinalize.test_can_use_super.<locals>.SuperFinalizec s t |¡| _d S rR )Úsuperr8 Ú saved_resultr rË r= r> r8 ø$ s zNTestArrayFinalize.test_can_use_super.<locals>.SuperFinalize.__array_finalize__)rN rO rP r8 Ú __classcell__r= r= rË r> Ú SuperFinalize÷$ s rZ r# )r' r6 ræ rW r rX )rF rZ rD r= r= r> Útest_can_use_superö$ s z$TestArrayFinalize.test_can_use_superN)
|
rN rO rP rä rK rQ rR rS rV r[ r= r= r= r> rG ¯$ s "rG c C s t d¡} tt| jdd d S )NrH õ âr& )r' rC r r) r r r= r= r> Ú2test_orderconverter_with_nonASCII_unicode_orderingÿ$ s
|
r] c C s G dd d} G dd d| }G dd d| }t jddgd d
|
}||fD ]B}| }t||kd t||kd t||kd t||kd qHd S ) Nc @ s e Zd Zdd Zdd ZdS )z*test_equal_override.<locals>.MyAlwaysEqualc S s dS )Nrí r= r r= r= r> r
|
% s z1test_equal_override.<locals>.MyAlwaysEqual.__eq__c S s dS )Nrì r= r r= r= r> r % s z1test_equal_override.<locals>.MyAlwaysEqual.__ne__N)rN rO rP r r r= r= r= r> Ú MyAlwaysEqual % s r^ c @ s e Zd ZdZdS )z-test_equal_override.<locals>.MyAlwaysEqualOldr½ Nr0 r= r= r= r> ÚMyAlwaysEqualOld% s r_ c @ s e Zd ZdZdS )z-test_equal_override.<locals>.MyAlwaysEqualNewNr6 r= r= r= r> ÚMyAlwaysEqualNew% s r` r rä rî r rí rì rÿ )r^ r_ r` ræ Zmy_always_equal_clsZmy_always_equalr= r= r> Útest_equal_override% s ra ÚfunÚnpfunrá rø r³ c C s, |t ||}| |}||}t|| d S rR )r¹ r )rb rc rá rø r³ r Úgotrâ r= r= r> Útest_npymath_complex % s re c
|
C sÈ ddl m} m}m}m}m} | tj|tj|tj |tj
|
|tj i}dtj tj tj f}tjtjtjf}tjddP | ¡ D ]@\}} t ||¡D ]*\}
|
} | |
|
} || } | | }t| | qqxW 5 Q R X d S )Nr )Ú npy_log10Únpy_coshÚnpy_sinhÚnpy_tanÚnpy_tanhr# Úignore)r¬ )r| rf rg rh ri rj r' Úlog10ÚcoshÚsinhÚtanÚtanhrL rÚ rÅ rÈ rD Zerrstater rÌ rþ r )rf rg rh ri rj r r_ ri rb rc rá r÷ r rd râ r= r= r> Útest_npymath_real1% s( ürq c C s t jddd} t jddd}t jddgd| gddd}tt jd | d
|
d jd d tt jd |d
|
d jd d tt jd dd
|
d jd d t ¡ }t| ||g|D ]$\}\}}t|j| t|j | qªt jd| d
|
d }t
|
|¡ t jddd
|
}|d d
d f |d d
d f< d S )Nzu1,c8TrJ zu4,c8rD r¦ r` rï r# r rÿ ZALIGNEDFr Zc8) r' r r r* rJ rm Zget_struct_alignmentsrZ r+ r1 rK ) rJ rK Zd3r¾ rÕ r+ r9 rÛ Údstr= r= r> Ú test_uintalignment_and_alignmentF% s
|
rs c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestAlignmentc C sÞ t ||||f}t||||d}|d kr4t |¡j}t|jd d | d t|drft|j|| nt|j|f| t|j| |dkrt |j
|
j | n@|dkr¼|j dkrÚt |j
|
j | n |d krÔt |j
|
j | nt d S )NrJ r$ r r! r r )r6 r? r' r r+ r r4 r, r7 r rJ r r9 r r) )rF r7 r r r8 r² rá r= r= r> r$ p% s"
|
|
zTestAlignment.checkc C sr dD ]h}dD ]^}dD ]T}t tjd dg D ]:}|dkr:q,|ddd |ffD ]}| |t |¡||¡ qJq,qq qd S )
|
N)
|
r# r" rÌ r r r+ r® r¯ r° N)r r# rÌ r )r r Nr¯ zi4,i4,i4r r# r" rÌ )rÎ r' r r$ r )rF r8 ro r r r7 r= r= r> Útest_various_alignments
% s z%TestAlignment.test_various_alignmentsc
|
C s$ dD ]}t dtj}t dtj|d}t dtj|d}t }| tjd¡ | d¡ W 5 Q R X | tj¡ || }| d¡ | tj¡ || }|| }| ¡ |d d
< | ¡ |d d
< | ¡ |d d
< |d d d
¡ |d d d
< |d d d
¡ |d d d
< |d d d
¡ |d d d
< qd S )N)r# r" r r r+ r® NrÌ rJ zCasting complex valuesr r" )
|
r? r' rÈ rÆ rD r rí r rE r )rF r8 Zxf64Zxc64Zxf128rð r8 r= r= r> Ú test_strided_loop_alignments% s&
|
|
z*TestAlignment.test_strided_loop_alignmentsN)rN rO rP r$ ru rv r= r= r= r> rt j% s rt c C s t jddd} tjdkr"d}d}nd}d}| d|¡}t|| | d|¡}t|d t t| jdd ¡ t t| jdd
|
¡ t t| jd d¡ d S ) Nr¯ rµ r r8 r r# r³ r3 r% r® r¹ ) r' rC r; r< Zgetfieldr r¨ r r) )rD r¿ rñ r¦ r= r= r> Ú test_getfield±% s
|
|
|
rw c @ s@ e Zd ZdZdd Zdd Zdd Zdd Zd
|
d Zd d Z dS )Ú TestViewDtypezP
|
Verify that making a view of a non-contiguous array works as expected.
|
c C s t jdddd d d
}tjtdd | d¡ W 5 Q R X ddgddgd dgd
|
dgd dgg}t|d d
t jf d¡| d S ) NrB rM r r" ú the last axis must be contiguousr rR r r rÄ r )r' rC r¨ r r) rW r rk ©rF rá râ r= r= r> Útest_smaller_dtype_multipleÆ% s ÿ"z)TestViewDtype.test_smaller_dtype_multiplec C s t jdddd d d
}tjtdd | d¡ W 5 Q R X tjtdd |d d
t jf d¡ W 5 Q R X d gd
|
gd gg}t|d d
t jf d ¡| d S ) NrH rM r r" ry r r. z When changing to a smaller dtyper5 ó ó rb )r' rC r¨ r r) rW rk r rz r= r= r> Ú test_smaller_dtype_not_multipleÏ% s ÿÿ"z-TestViewDtype.test_smaller_dtype_not_multiplec C s^ t jddd dd¡d d d
d d
f }t jdgdgdgd gd
|
ggd d}t| d ¡| d S ) Nr rR r rB r" i i i i i rM )r' rC rÆ ræ r rW rz r= r= r> Útest_larger_dtype_multipleÞ% s ( ÿÿz(TestViewDtype.test_larger_dtype_multiplec C sx t jddd dd¡d d d
d d
f }tjtdd | d¡ W 5 Q R X d gd
|
gd gd gd gg}t| d¡| d S )Nr rR r rB r" z When changing to a larger dtyper r. s s s s s rb )r' rC rÆ r¨ r r) rW r rz r= r= r> Ú test_larger_dtype_not_multipleå% s (ÿ ÿz,TestViewDtype.test_larger_dtype_not_multiplec C s@ t jddd dd¡j}tjtdd | d¡ W 5 Q R X d S ) Nr+ rM r r rÌ ry r rR )r' rC rÆ r r¨ r r) rW r r= r= r> Útest_f_contiguousð% s
|
ÿz TestViewDtype.test_f_contiguousc C sd t jddd ddd¡ ddd¡}d d
|
gd d ggd dgddggddgddggg}t| d¡| d S )NrÄ rî r r" rÌ r r# r r i i i i i i i i i
|
i i rR )r' rC rÆ rê r rW rz r= r= r> Útest_non_c_contiguous÷% s ÿ ÿþz#TestViewDtype.test_non_c_contiguousN)
|
rN rO rP rä r{ r~ r r r r r= r= r= r> rx Â% s rx r? r r® rÄ r¯ rº r° é` rM é é¿ r i i iÿ éÿ c C sJ t j d¡ dt j | ¡ d¡ }t j|t j |jd d¡< tt j |ddt j |dd t j
|
t j | dd }d
|
|t j |jd d ¡< tt j |ddt j |dd t j
|
t j | dd }d |t j |jd d ¡< tt j |ddt j |dd t j
|
t j | dd }t j
|
|t j |jd t | d ¡< tt j |ddt j |dd d S )Nr_ g à¿r× r rÌ ÚquickrÜ Úheapr r# rH rý r" ) r' r rª ÚsamplerE rÚ rG r7 r r rL r rÝ )r? rS ZinfarrZ neginfarrr= r= r> Útest_sort_float& s ÿ$r c C s t j d¡} d}t t j¡j}t t j¡j}| j|||d d¡}||t j |j
|
d d¡< ||t j |j
|
d d¡< t t j |ddt j |d d d S )
|
Nr_ r ©ÚlowÚhighr9 r¶ r rB r rÜ r ) r' r r¦ r r¶ r¸ r» r§ rE rG r7 r r )Úrngr? ZminvÚmaxvrS r= r= r> Ú test_sort_int & s r c C sn t j d¡} d}t t j¡j}| jd||d d¡}||t j |j d d¡< t
|
t j |ddt j |d d d S )
|
Nr_ r r r r· rB r rÜ r ) r' r r¦ r r· r» r§ rE rG r7 r r )r r? r rS r= r= r> Útest_sort_uint(& s r )§Úcollections.abcrà r¦ r; r r/ r# rÌ r- r* r rA r? rx r¨ Ú
|
contextlibr Z numpy.compatr ro r] rw r rÆ Únumpyr' r| re rm Znumpy.core._rational_testsr Z numpy.testingr r r r r
|
r r r r r r r r r r r r Z numpy.testing._private.utilsr r Znumpy.core.tests._localesr Znumpy.lib.recfunctionsr r r rÿ r? r@ r¬ rÁ r
|
r( rI rV r} r rè rO rx r¤ rñ rõ rT re rp rr r r r rÏ r r r r r% r( r8 rA rK rÁ rË rÕ rä rü rÿ r r r r; rB r| r r© r³ r´ rµ rÁ rÆ rÔ r© r« rÑ rÜ râ rå Znumpy.core._internalrí rî r r( rY rp ry r r r r r r r r r r r¨ r¶ rÁ rÆ rî rø r r r r r3 rG r] ra Znpy_cabsÚabsoluteZnpy_cargZanglerL rÚ ZcomplexfloatingÚ__subclasses__re rq rs rt rw rx r r r r= r= r= r> Ú<module> sr L $ 4 muh` P &Z` @ f- 6 - N?B7 ^%Hc} t< ' Z ) M &V k T S"-
|
c &/ Q \q]P
|
|
þþ $G@ ÿ
|
|
