\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 j p t 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 t Nonet AssertionError( 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 @R1 ( 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 R2 i R3 R4 i ii N( R
R6 R R$ R7 R R R8 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+ R0 R9 R: ( ( ( 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@R2 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 ( R6 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 ( R; R< R= RA RB ( ( ( 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 RE 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 RD t f1R2 i N( i i
( i i ( ii ( i i
( i i ( ii ( i i ( i i ( i i (
R RE R R
R R
R R R6 R2 ( R RF 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 RD s |S3RJ N( i R g ?( i R g @( R s |S3( R g ?( R g @( R s |S3( R RE R R
R R
R R R ( R RF 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 RD R R R RJ R g ?N( i i i g @( i i i g @( i g @( i g @( R RE R R
R R
R R R ( R R R RF 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 | | t | t p t d S( s Test merge_arrays masked arraysi i i R2 i RI iR RD RJ t
asrecarrayN( i i ( i i ( ii ( i i ( i i ( i i (
R R6 R RE R R
R t
isinstanceR R ( R RF 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 >@iR2 i R RJ N( i g $@( i g 4@( ig >@( i i ( i i ( i i ( RE R
R R$ R
R6 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 R1 R g ?R g @s -1R s |S3N( R g ?i
( R g @i ( s -1ii ( R s |S3( R RE 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( R; R< R= R RG RH RK RL RN RR RS RT ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRC 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 iR2 i R RD N( i i
( i i ( ii ( i i ( i i ( i i ( R t
append_fieldsR6 R R
R ( R RF 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 iR2 i R RD N( R R ( i i
id ( i i i ( ii i( i i i ( i i i ( i i i ( R RV R6 R R
R ( R RF 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 arraysiR1 R i
i i R g ?R g @g R2 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 RV R6 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 R1 R i
i i i i g @i i g @ig R2 R R R R R N( i g @( i g @( ig ( i i ( i i ( i i ( R RV R6 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
* ( R; R< R= R RW RX RY RZ ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRU 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 RF R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRG 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 fieldsRI i i i
i i N( R R\ R R
R R ( R RF 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 R2 i R RD 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 R\ R6 R R
R R R2 ( R RF 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 R1 i ii R2 i t f3N( R g $@g Y@( R g 4@g i@( Ra g >@g r@( R s |S3( R i i( R i i( R g $@g Y@( R g 4@g i@( Ra 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( Ra 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 R\ R6 R R2 R
( R RF 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@Ra g >@g r@R R s |S3R R1 s ???g 8g g it Dt defaultsi i R2 i N( R g $@g Y@( R g 4@g i@( Ra g >@g r@( R s |S3( R i g ( R i g ( R g $@g Y@( R g 4@g i@( Ra 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 R\ R6 R R2 ( R RF R Rc Rf 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 R1 i i i R2 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 R6 R R\ R R R2 R t TypeErrorR ( 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 Ra R1 i i i R2 i R i i i N( R R ( R R ( Ra R1 ( i i i ( i i i ( R R ( R R ( Ra R1 ( i i i ( i i i ( i i i ( i i i ( i i i ( R
Ri R R6 R R\ R R2 ( R Rk R Rl R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_checktitles s !!!
( R; R< R= R RG R^ R` Rd Rg Rm Rn ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyR[ 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 Ra 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 R2 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 R6 ( R R R R R ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt test_base sj 9$<$
( R; R< Rz ( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyRo s t __main__( t syst numpyR
t numpy.maR6 t numpy.ma.testutilst numpy.ma.mrecordsR t numpy.lib.recfunctionst libt recfunctionsR' R* R t TestCaseR R> RC RU R[ Ro R; t run_module_suite( ( ( sG /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_recfunctions.pyt s
s6A