\Kc @ s d d k Z d d k Z d d k i Z d d k Td d k l Z d d k Te i i
i Z e i i
i Z e i i
i
Z
d e f d YZ d e f d YZ d e f d YZ d
e f d YZ d e f d
YZ d e f d YZ e d j o e n d S( iN( t *( t
MaskedRecordst TestRecFunctionsc B s_ e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z RS(
s
Misc tests
c C s t i d d g } t i d d d g } t i d d g d
d d t f g } t i d d f d
d f g d
d t f d d t f d t f g f g } | | | | f | _ d S( Ni i i
i i t Ag ?t Bg @t dtypes |S3g @i i g @t at bt bat bb( R g ?( R g @( R s |S3( i g @( i g @( t npt arrayt floatt intt data( t selft xt yt zt w( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt setUp s -c
C s | i \ } } } } t | | f d t } t | t i d t f d t f g t | | f d t } t | t i d t f d t f g t | | f d t } t | t i d t f d d t f g t | | f d t } t | t i d t f d d d t f g f g t | | f d t } t | t i d t f d t f d t f d t f g t | | f d t } t | t i d t f d d t f d d t f d t f g f g f g d
S(
s Test zip_descrt flattent R s |S3R R R R R N( R s |S3( R s |S3( R t zip_descrt Truet assert_equalR
R R
t FalseR ( R R R R R t test( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_zip_descr s0 ""% c C s t i d d
f d d f g d d t f d d
t f d t f g f g } t | d } t i d d g d d d
t f d t f g f g } t | | t | d } t i d d g d d t f g } t | | t | d
g } t i d d f d d f g d d t f d d t f g f g } t | | t | d
d g } t i d d g d d t f g } t | | t | d d g } d S( s Test drop_fieldsi i g @i i g @R R R R R N( i g @( i g @( i g @( ( i g @( i g @( ( i g @( i ( i ( g @( g @( i ( i ( R
R R
R t drop_fieldsR ( R R R t control( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_drop_fields< s$ -$
$
$
$
c C s t i d d d d g f f d d d d g f f g d d
t f d d t f d
t d f f g f g } t | h d d
6d d
6 } d t f d d t f d t d f f g f g } | i | } t | i | t | | d S( s Tests rename fieldsi i g @g >@i i g @g N@R R R R R R t BBN( R
R R
R t
rename_fieldst viewR R ( R R R t newdtypeR ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_rename_fieldsW s 6 * 0c C s t i d d t f g } t | } t | d
t i d t f d d t f d t f g f g } t | } t | d d d f f d S( s Tests get_namesR s |S3R R R R R N( R s |S3( R R ( s bas bb( R
R R t get_namesR R
( R t ndtypeR ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_get_namesc s
3 c C s t i d d t f g } t | } t | d
t i d t f d d t f d t f g f g } t | } t | d d S( s Test get_names_flatR s |S3R R R R R N( R s |S3( R R ( R R s bas bb( R
R R t get_names_flatR R
( R R& R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_get_names_flatn s
3 c C sN t i d
d t f g } t | } t | h g d 6g d 6 t i d t f d d t f d g f g } t | } t | h g d 6g d 6d g d 6d g d 6 t i d t f d d t f d d t f d t f g f g f g } t | } h g d 6g d 6d g d 6d g d 6d d g d 6d d g d 6} t | | d S( s Test get_fieldstructureR s |S3R t BAR s |S1t BBAt BBBN( R s |S3( s BBs |S1( R
R R t get_fieldstructureR R
( R R& R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_get_fieldstructurey s - / * % c C sk t i d d f d d f d d f d d f d d f d d f g d d d f d d f d d f d d f d d f d d! f g d d
t f d d t f d" g f g } t | d t d t } d d g } t t | d | t | d | | d t | d d
d t } d d d d d g } t t | d | t | d | | d t | d d d t } d d d d g } t t | d | t | d | | d t | d d d t } d d d d g } t t | d | t | d | | d t | d d d t } d d d d d g } t t | d | t | d | | d d S(# s Test find_duplicatesi g @R i g ?t Ct maski R R R* R s |S1t
ignoremaskt return_indexit keyi i i N( g @R ( g @R ( g @R ( g ?R ( g @R ( g @R/ ( i i ( i i ( i i ( i i ( i i ( i i ( s BBs |S1( t maR R
R t find_duplicatesR R R t sorted( R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_find_duplicates s2 !!!' c
C s d t f g } t i d d d d d d d g d d d d d d d d g i | } t | d t d t } d d d d g } t t | d
| t | d | | d
t | d t d t } d d d d d d g } t t | d
| t | d | | d
d S(
s- Test the ignoremask option of find_duplicatesR i i i R0 i R1 R2 i ii N( R
R4 R R" R5 R R R6 R ( R R& R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_find_duplicates_ignoremask s !'( t __name__t
__module__t __doc__R R R R$ R' R) R. R7 R8 ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyR s
!t TestRecursiveFillFieldsc B s e Z d Z d Z d Z RS( s%
Test recursive_fill_fields.
c C s t i d d
g d d t f d t f g } t i d d | i } t | | } t i d d d g d d t f d t f g } t | | d S( s, Test recursive_fill_fields on flexible-arrayi g $@i g 4@R R R i i g N( i g $@( i g 4@( i ( i g $@( i g 4@( i g ( R
R R
R t zerosR t recursive_fill_fieldsR ( R R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_simple_flexible s -c C s t i d
d g d d d g d d t f d t f g } t i d d | i } t | | } t i d d d g d d d d g d d t f d t f g } t | | d S( s3 Test recursive_fill_fields on masked flexible-arrayi g $@i g 4@R0 i R R R i g N( i g $@( i g 4@( i i ( i i ( i ( i g $@( i g 4@( i g ( i i ( i i ( i i ( R4 R R
R R= R R> R ( R R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_masked_flexible s ( R9 R: R; R? R@ ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyR< s t TestMergeArraysc B s_ e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z RS(
s
Test merge_arrays
c C s t i d d g } t i d d d g } t i d d g d
d d t f g } t i d d f d
d f g d
d t f d d t f d t f g f g } | | | | f | _ d S( Ni i i
i i R g ?R g @R s |S3g @i i g @R R R R ( R g ?( R g @( R s |S3( i g @( i g @( R
R R R
R ( R R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyR s ' -c C s | i \ } } } } t | } t i d d g d d t f g } t | | t | f } t | | t | d t } t | | t | d t } t | | d S( s$ Test merge_arrays on a single array.i i R t f0R N( i ( i ( R t merge_arraysR
R R
R R R ( R t _R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_solo s $
c C s | i d } t | d t } t | | t | d t } t i d d g d d
t f d t f d t f g } t | | d
S( s6 Test merge_arrays on a single array w & w/o flatteningi R i i g @i i g @R R R R N( i i g @( i i g @( R RC R R R R
R R
R ( R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_solo_w_flatten s
$c C s | i \ } } } } t | | f d t } t i d d d g d d t f d
t f g } t | | t | | f d t } t i d d d g d d d d g d d t f d
t f g } t | | t | i | i d
S( s Test standard & standardt usemaski i
i i ii R RB t f1R0 i N( i i
( i i ( ii ( i i
( i i ( ii ( i i ( i i ( i i (
R RC R R
R R
R R R4 R0 ( R RD R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt
test_standard s
c C s | i \ } } } } t | | f d t } t i d
d g d d t f d d t f g } t | | t | | f d t } t i d d f d d f g d d t f d d d t f g f g } t | | d S( s Test standard & flexibleR i R g ?i R g @R RB s |S3RH N( i R g ?( i R g @( R s |S3( R g ?( R g @( R s |S3( R RC R R
R R
R R R ( R RD R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_flatten s
c C s | i \ } } } } t | | f d t } t i d d g d d t f d
t f d t f d t f g } t | | t | | f d t } d t f d
d
t f d d t f d t f g f g f g } } t i d d d f f d d d f f g d | } d S( s Test flatten standard & nestedR i i g @i i g @R RB R R R RH R g ?N( i i i g @( i i i g @( i g @( i g @( R RC R R
R R
R R R ( R R R RD R R t controldtypeR ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_flatten_wflexible s $
+*c C s | i \ } } } } t i d d d g d d d d g } t | | f d t } t i d
d d g d d d d g d d t f d
t f g } t | | t | | f d t d t } t | | d S( s Test merge_arrays masked arraysi i i R0 i RG iR RB RH t
asrecarrayN( i i ( i i ( ii ( i i ( i i ( i i ( R R4 R RC R R
R ( R RD R t mxR R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_wmasked_arrays, s '
c C s t t i d d g i d t f g t i d d d g f } t i d
d d g d d d d g d
d t f d t f g } t | | d S( s Test single fieldi i R g $@g 4@g >@iR0 i R RH N( i g $@( i g 4@( ig >@( i i ( i i ( i i ( RC R
R R" R
R4 R R ( R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_w_singlefield9 s ' c C sz | i d } t | t i d d d g i d t f g f } t i d d d g d d d t f d t f g } d
S( s+ Test merge_arrays w/ a shorter flexndarray.ii
i i R/ R g ?R g @s -1R s |S3N( R g ?i
( R g @i ( s -1ii ( R s |S3( R RC R
R R" R
R ( R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_w_shorter_flexB s
6( R9 R: R; R RE RF RI RJ RL RO RP RQ ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRA s
t TestAppendFieldsc B s; e Z d Z d Z d Z d Z d Z d Z RS( s
Test append_fields
c C s t i d d g } t i d d d g } t i d d g d
d d t f g } t i d d f d
d f g d
d t f d d t f d t f g f g } | | | | f | _ d S( Ni i i
i i R g ?R g @R s |S3g @i i g @R R R R ( R g ?( R g @( R s |S3( i g @( i g @( R
R R R
R ( R R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyR O s ' -c C s | i \ } } } } t | d d d d d g } t i d d d g d d d d g d d t f d t f g } t | | d
S( s Test simple caseR R i
i i i i iR0 i R RB N( i i
( i i ( ii ( i i ( i i ( i i ( R t
append_fieldsR4 R R
R ( R RD R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_append_singleW s c C s | i \ } } } } t | d d d d d g d d g g } t i d d d g d d d d g d d t f d t f d t f g } t | | d S( s Test simple caseR R R i
i i id i i i iR0 i R RB N( R R ( i i
id ( i i i ( ii i( i i i ( i i i ( i i i ( R RS R4 R R
R ( R RD R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_append_double` s *$c C s~ | i d } t | d d d d d g } t i d d d g d d d d g d d d t f d t f g } t | | d S( s* Test append_fields on flexible type arraysiR/ R i
i i R g ?R g @g R0 i i R s |S3N( R g ?i
( R g @i ( ig i ( i i i ( i i i ( i i i ( R s |S3( R RS R4 R R R
R ( R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_append_on_flexi s
c
C s | i d } t | d d d d d g } t i d d d f d
d d f d
d d f g d d d d f d d d f d d d f g d d t f d d t f d t f g f d t f g } t | | d S( s# Test append_fields on nested fieldsi R/ R i
i i i i g @i i g @ig R0 R R R R R N( i g @( i g @( ig ( i i ( i i ( i i ( R RS R4 R R
R R ( R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_append_on_nestedr s
* ( R9 R: R; R RT RU RV RW ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRR K s t TestStackArraysc B sV e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
RS( s
Test stack_arrays
c C s t i d d g } t i d d d g } t i d d g d
d d t f g } t i d d f d
d f g d
d t f d d t f d t f g f g } | | | | f | _ d S( Ni i i
i i R g ?R g @R s |S3g @i i g @R R R R ( R g ?( R g @( R s |S3( i g @( i g @( R
R R R
R ( R R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyR s ' -c C st | i \ } } } } t | f } t | | | i | | j t | } t | | | i | | j d S( s" Test stack_arrays on single arraysN( R t stack_arraysR t
failUnless( R RD R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRE s
c C s | i \ } } } } t | | f d t } t i d d d d g } t | | t | | f d t } t i d d d d d g } t | | t | | f d t } t i d d d d d g } t | | d S( s- Tests combinations of arrays w/o named fieldsRG i i i
i i N( R RY R R
R R ( R RD R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_unnamed_fields s
c C s | i \ } } } } t | | f } t i d
d d d g d d d d d g d d t f d d t f g } t | | t | i | i t | | f } t i d d d d g d d d d d g d d d t f d t f g } t | | t | i | i t | | | f } t i d d d! d" d# d$ g d d% d& d' d( d) d* g d d+ d t f d t f g } t | | d S(, s0 Test combination of arrays w/ & w/o named fieldsi ii R R R0 i R RB s |S3t f2N( i ii( i ii( iR i ( iR i ( i i i ( i i i ( i i i ( i i i ( R s |S3( R i i( R i i( iii ( iii ( i i i ( i i i ( i i i ( i i i ( R s |S3( R i i( R i i( R i i( R i i( iii ( iii ( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( R s |S3( R RY R4 R R
R R R0 ( R RD R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_unnamed_and_named_fields s4
c C sj | i \ } } } } t i d d d g d
d d
t f d t f g } t | | f } t i d d d d d g d
d d
t f d t f g d d d! d" d# d$ g } t | | t | i | i t | | | f } d% d
t f d t f d t f g } t i d& d' d( d) d* d+ d, g d
| d d- d. d/ d0 d1 d2 d3 g } t | | t | i | i d S(4 s2 Test combination of arrays w/ matching field namesR g $@g Y@R g 4@g i@t cg >@g r@R R s |S3R R/ i ii R0 i t f3N( R g $@g Y@( R g 4@g i@( R^ g >@g r@( R s |S3( R i i( R i i( R g $@g Y@( R g 4@g i@( R^ g >@g r@( R s |S3( i i i ( i i i ( i i i ( i i i ( i i i ( R s |S3( R i ii( R i ii( R g $@g Y@i( R g 4@g i@i( R^ g >@g r@i( iiii ( iiii ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( R R
R R RY R4 R R0 R
( R RD R R t zzR R R& ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_matching_named_fields s.
$
c C s | i \ } } } } t i d d d g d
d d
t f d t f g } h d d 6d d
6d d 6d d 6} t | | f d | } t i d d d d! d" g d
d# d
t f d t f g d d$ d% d& d' d( g } t | | t | i | i t | i | i d S() sF Test defaults: no exception raised if keys of defaults are not fields.R g $@g Y@R g 4@g i@R^ g >@g r@R R s |S3R R/ s ???g 8g g it Dt defaultsi i R0 i N( R g $@g Y@( R g 4@g i@( R^ g >@g r@( R s |S3( R i g ( R i g ( R g $@g Y@( R g 4@g i@( R^ g >@g r@( R s |S3( i i i ( i i i ( i i i ( i i i ( i i i ( R R
R R RY R4 R R0 ( R RD R R` Rc R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt
test_defaults s "
c C s d t f d t f d t f g } t i d g d d g d | } d t f d t f d t f g } t i d g d | } t i d d g d d d g d | } t | | f d
t } t | | t | i | i y t | | f d
t } Wn t
j
o n Xt d S( s Tests autoconversionR R R/ i i i R0 i R i i i t autoconvertN( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( i i i ( R
t boolR R4 R RY R R R0 R t TypeErrort AssertionError( R t adtypeR t bdtypeR R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_autoconversion s !!!
c C s d t f d t f d t f g } t i d g d
d g d | } d t f d t f d t f g } t i d g d | } t | | f } t i d d g d
d d g d | } t | | t | i | i d S( s$ Test using titles in the field namesR R R R R^ R/ i i i R0 i R i i i N( R R ( R R ( R^ R/ ( i i i ( i i i ( R R ( R R ( R^ R/ ( i i i ( i i i ( i i i ( i i i ( i i i ( R
Rf R R4 R RY R R0 ( R Ri R Rj R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_checktitles s !!!
( R9 R: R; R RE R[ R] Ra Rd Rk Rl ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRX s t
TestJoinByc B s e Z d Z RS( c C s t i t t i d t i d d t i d d d d t f d t f d t f g } t i t t i d
d t i d d
t i d d d d t f d t f d t f g } t d | | d d } t i d@ dA dB dC dD g d d t f d' t f d( t f d t f d t f g } t | | t dE | | } t i dF dG dH dI dJ g d d t f d t f d t f d t f g } t dK | | d) } t i dL dM dN dO dP dQ dR dS dT dU dV dW dX dY dZ d[ d\ d] d^ d_ g d= d` da db dc dd de df dg dh di dj dk dl dm dn do dp dq dr ds g d d t f d t f d t f d t f g } t | | t dt | | d> } t i du dv dw dx dy dz d{ d| d} d~ g
d= d d d d d d d d d d g
d d t f d t f d t f d t f g } d? S( s Basic test of join_byi
i2 i< id in R R R R^ i i iA iK t dt jointypet inneri7 ii i i8 iB ij ie i i9 iC ik if i i: iD il ig i i; iE im ih t b1t b2t outeri ii i3 i i4 i i5 i i6 iF i iG i iH i
iI i iJ R0 t leftouterN( i i7 iA ii id ( i i8 iB ij ie ( i i9 iC ik if ( i i: iD il ig ( i i; iE im ih ( R R ( i i7 ii id ( i i8 ij ie ( i i9 ik if ( i i: il ig ( i i; im ih ( R R ( i i2 id i( i i3 ie i( i i4 if i( i i5 ig i( i i6 ih i( i i7 ii i( i iA iid ( i i8 ij i( i iB iie ( i i9 ik i( i iC iif ( i i: il i( i iD iig ( i i; im i( i iE iih ( i
iF iii ( i iG iij ( i iH iik ( i
iI iil ( i iJ iim ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( R R ( i i2 id i( i i3 ie i( i i4 if i( i i5 ig i( i i6 ih i( i i7 ii i( i i8 ij i( i i9 ik i( i i: il i( i i; im i( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( i i i i ( R
R t zipt arangeR
t join_byR R4 ( R R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_base sj 9$<$
( R9 R: Rx ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRm s t __main__( t syst numpyR
t numpy.maR4 t numpy.ma.testutilst numpy.ma.mrecordsR t numpy.lib.recfunctionst libt recfunctionsR% R( R t TestCaseR R< RA RR RX Rm R9 t run_module_suite( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt s
s6A