ó ˜í Yc@sdZddlZddlZddlmZmZmZddlZ ddl m Z m Z d„Z d„Zd„Zd „Zd „ZdS( sTest replace().i˙˙˙˙N(t assert_equaltassert_array_equalt assert_raisesi(tarrayst array_orderccs)x"tjdƒD]}t|fVqWdS(stest nonreduce functionst nonreduceN(tbnt get_functionst unit_maker(tfunc((s>lib/python2.7/site-packages/bottleneck/tests/nonreduce_test.pyttest_nonreduce sc Cs]d}|d7}|j}td|ƒ}tjjdddgƒ}ddtjtj g}xůtt|ƒƒD]ĺ\}}xÖt dƒD]Č}|j dkr§d} n(|j t |j dƒƒ}|j |} x|D]w} t|jjtjƒs!tj| ƒs qÖntj| ƒs!qÖq!n|jƒ} tjƒ"tjdƒ|| | | ƒWd QX|jƒ} tjƒ"tjdƒ|| | | ƒWd QX|d t|ƒt|jƒt|jƒ| | t|ƒ|f} || }t| | d |ƒ|d 7}t| d ƒst| d ƒrÖ| j}| j}t|||||fƒqÖqÖWq‰WqpWd S(s1Test that bn.xxx gives the same output as np.xxx.sA func %s | input %s (%s) | shape %s | old %f | new %f | order %s s Input array: %s s bn.slow.%siiiitignoreNtaterr_msgs dtype mismatch %s %stdtype(t__name__tevaltnptrandomt RandomStatetnantinft enumerateRtrangetsizetrandinttmaxtflatt issubclassRttypetinexacttisfinitetcopytwarningstcatch_warningst simplefiltertstrtshapeRRthasattrR(R tmsgtnametfunc0trstnewstitarrtidxtoldtnewtactualtdesiredttupR tdatdd((s>lib/python2.7/site-packages/bottleneck/tests/nonreduce_test.pyRsF            $    cCs¸ddg}xĽ|D]}tjdd|ƒ}tttj|jƒddƒtttj|jƒddƒtttjj|jƒddƒtttjj|jƒddƒqWdS(sTest replace for unsafe caststint32tint64iRgš™™™™™š?iN(RtzerosRt ValueErrorRtreplaceR tslow(tdtypesRR ((s>lib/python2.7/site-packages/bottleneck/tests/nonreduce_test.pyttest_replace_unsafe_cast>s  "cCsKdddg}tttj|ddƒd}tttj|ddƒdS(s Test that non-array input raisesiiiiN(iii(Rt TypeErrorRR:(R ((s>lib/python2.7/site-packages/bottleneck/tests/nonreduce_test.pyttest_non_arrayIscCs§tjd ƒjdddƒ}|jƒ}tj|tjdƒ|jƒ}d}t||d|ƒ|jƒ}tjj|tjdƒd}t||d|ƒdS( s'Test replace, int array, old=nan, new=0iiiis,replace failed on int input looking for nansR s1slow.replace failed on int input looking for nansNii( RtarangetreshapeR RR:RRR;(R R1R2R'((s>lib/python2.7/site-packages/bottleneck/tests/nonreduce_test.pyttest_replace_nan_intUs   (t__doc__R!tnumpyRt numpy.testingRRRt bottleneckRtutilRRR RR=R?RB(((s>lib/python2.7/site-packages/bottleneck/tests/nonreduce_test.pyts     -