ó î&]\c@`sšddlmZmZmZddlZddlZddlmZm Z m Z m Z ddl Z ddl m Z mZddlmZddlmZmZmZmZmZmZmZmZddlmZmZd„Zd „Zd „Zd e fd „ƒYZ!d e fd„ƒYZ"de fd„ƒYZ#de fd„ƒYZ$de fd„ƒYZ%d„Z&de fd„ƒYZ'de fd„ƒYZ(dS(i(tdivisiontprint_functiontabsolute_importN(tassert_t assert_equaltassert_allclosetassert_array_equal(traisestwarns(txrange(t argrelmaxt argrelmintpeak_prominencest peak_widthst_unpack_condition_argst find_peakstfind_peaks_cwtt_identify_ridge_lines(t_local_maxima_1dtPeakPropertyWarningcC`s{tjd|ƒjtƒ}tj|dtƒ}xDt|ƒD]6\}}||||}|tj|d ƒ7}q=W|S(Nitdtypei(tnptarangetastypetfloattzerost enumeratetexp(t center_locstsigmast total_lengthtxdatatout_datatindtsigmattmp((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyt_gen_gaussianss cC`s[t|ƒ}||d}tj|||d|ƒjtƒ}t|||ƒ}||fS(Nitnum(tlenRtlinspaceRtintR$(RRt num_peakstdeltaRR ((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyt_gen_gaussians_even&s  %c C`sÛd„}tj|ƒ}tj|ƒ}tj|dgdtƒ}||ddd…f<|d|dt|ƒ}||krtdƒ‚n|t|ƒd}|t|ƒd} x÷td|ƒD]æ} || ddf} || ddfd} | |dkr6t|ƒdkr6| d| |j ƒ7} n| | dkrkt|ƒdkrk| |j ƒ7} n|| |dƒ} || |dƒ} | | g|| dd…feZd„Zd„Zd„Zd„Zd„Zd„ZRS(cC`sNtjddgƒ}t|dtjdƒdƒ}tt|ƒdkƒdS(Niidiii(RRRRSRR&(RFt test_matrtlines((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyRH–scC`s¸tjddgƒ}d|d Verify that appropriate warnings are raised. s!some peaks have a prominence of 0iiiRQNR˜(RRR (RFtmsgRš((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyt test_warnings¬s #( RVRWRHRzR”R–R™RURœ(((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyR‹Cs    tTestPeakWidthscB`skeZd„Zejjdƒd„ƒZd„Zd„Zd„Z d„Z ejjdƒd„ƒZ RS(cC`sÅtggƒd}tt|tjƒƒt|jdƒtdddggƒd}tt|tjƒƒt|jdƒtggƒ}x4|D],}tt|tjƒƒt|jdƒq‘WdS(sN Test if an empty array is returned if no peaks are provided. iiiiN(R Rt isinstanceRtndarrayRRŽ(RFtwidthsR/R((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyRHºs s#ignore:some peaks have a width of 0c C`s½tjdddddddgƒ}d}xŒdddddddgD]o\}}}}t|dg|ƒ\}}} } t||ƒt|d||ƒt| |ƒt| |ƒqFWdS(sk Test a simple use case with easy to verify results at different relative heights. iiiiÿÿÿÿgg@gÐ?gð?g@g @gà?g@g@gè?gø?g@g@g@iN(ggg@g@(gÐ?gð?g@g @(gà?g@g@g@(gè?g@gø?g@(gð?g@gð?g@(g@g@gð?g@(g@g@gð?g@(RRBR R( RFRGt prominencet rel_heightt width_truetlip_truetrip_truet width_calctheighttlip_calctrip_calc((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyRzÉs $  cC`svtjdddgdƒ}tjdgdƒ}t|ddd…|ddd…ƒ}t|dd dd gƒdS( s: Test with non-C-contiguous input arrays. iidi2iiiNgè?iKgø?(RR•R R(RFRGRtresult((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyR–ás)cC`sättddƒ$ttjdƒtjdƒƒWdQXttddƒtddgƒWdQXttddƒ-ttjdƒtjdd tjƒƒWdQXttddƒttjdƒdƒWdQXttdd ƒ!ttjdƒd d gƒWdQXttdd ƒtgd dgƒWdQXttddƒ!ttjdƒddgƒWdQXttddƒ-tdd dd dgd dgddƒWdQXttddƒ$td dd gd gddƒWdQXdS(sD Verify that argument validation works as intended. RQs1D arrayiiNii iRsnot a valid indexii iscannot safely castgš™™™™™ñ?gffffff@R¢iÿÿÿÿREtprominence_data(ii(ii(NNN( RR4R RRRSRRRTRE(RF((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyRUês$%."".cC`s¿d}ttd|ƒ$tdddgdgddƒWdQXttd|ƒitdddddgdgdtjd gtjƒtjdgtjƒtjdgtjƒfƒWdQXdS( s> Verify that appropriate warnings are raised. ssome peaks have a width of 0RQiiR¢NiR«g(RRR RRBRCR(RFR›((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyRœ s%c C`s"dddg}dg}xtd ddfdddfdddfdddfdddfdddfdd d!fgƒD]¡\}\}}}tj|dtjƒtj|dtjƒtj|dtjƒf}|dkrèd}nd }ttd |ƒt||d |ƒWd QXqyWd S("s4Test with mismatching peak and / or prominence data.iigð?iÿÿÿÿiiRs#prominence data is invalid for peaks4arrays in `prominence_data` must have the same shapeRQR«N(gð?(iÿÿÿÿ(i(gð?(i(i(gð?(i(i(gð?gð?(ii(ii(gð?gð?(i(i(gð?(ii(i(gð?(i(ii(RRRBRCRRR4R ( RFRGR—Rtt prominencest left_basest right_basesR«RQ((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyt test_mismatching_prominence_datas$       (  c C`s‡dddddddddddg }tt|ddgddƒddddgƒtt|ddgddƒddddgƒdS(s3Test if x == eval_height counts as an intersection.iiiiRiR¢gg@g@g@gð?g@N(g(g@(g@(g@gUUUUUUå?(g@(gð?(g@(g@(RR (RFRG((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyttest_intersection_rules4s '( RVRWRHRXRYtfilterwarningsRzR–RURœR¯R°(((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyR¸s   c C`sltjdƒ}|}|d}|ddd…}td td ||ƒkƒtd td||ƒkƒtdtd||ƒkƒtdtd||ƒkƒtdtd||ƒkƒt||f||ƒ\}}t|||ƒt|||ƒttddƒt|tjd ƒ|ƒWdQXttdd ƒ$td|ftjd ƒ|ƒWdQXdS(sW Verify parsing of condition arguments for `scipy.signal.find_peaks` function. i iNig@g@RQsarray size of loweri sarray size of upper(NN(NN(iN(iN(iN(Ni(Ni(g@g@(g@g@(RRRRERRRR4(RGt amin_truet amax_trueRt amin_calct amax_calc((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyttest_unpack_condition_argsAs  t TestFindPeaksc B`s›eZdddddddddd h Zd „Zd „Zd „Zd „Zd„Zd„Zd„Z d„Z d„Z e j jddƒd„ƒZRS(t peak_heightstleft_thresholdstright_thresholdsR¬R­R®R t width_heightstleft_ipst right_ipsc C`s{d}ttjdƒd|d|d|d|ƒ\}}t|jdkƒx(|jD]}t||jdkƒqVWdS( s@ Test behavior for signal without local maxima. i R§t thresholdR¡twidthiN(NN(RERRRSRRŽt property_keys(RFt open_intervalRtpropstkey((sClib/python2.7/site-packages/scipy/signal/tests/test_peak_finding.pyt test_constantds c C`s’tjdddddddgƒ}tj|jddƒ}||ddd…s&  " :   +;\Su‰ «