o e@sdZddlmZmZmZddlZddlZddlm Z ddl Z ddl m Z m Z mZddl mZgdZdZed d eDgZd d Ze ejjd dZe ejjddZddZddZddZddZdS)zs Test used to verify PyWavelets Discrete Wavelet Transform computation accuracy against MathWorks Wavelet Toolbox. )divisionprint_functionabsolute_importN)assert_)uses_pymatbridgeuses_precomputedsize_set)matlab_result_dict_dwt) )zeroZzpd)ZconstantZsp0)Z symmetricsym)ZreflectZsymw)ZperiodicZppd)ZsmoothZsp1)Z periodizationZper)Z antisymmetricZasym)Z antireflectZasymw)dbr ZcoifZbiorZrbiocCsg|]}t|qS)pywtZwavelist).0namer r TD:\Projects\ConvertPro\env\Lib\site-packages\pywt/tests/test_matlab_compatibility.py srcCs8tdkrtt|jdgd}|S|j|jdf}|S)z) Return the sizes to test for wavelet w. full()diiiP)rlistrangeZdec_len)wZ data_sizesr r r_get_data_sizes"src Cstd}|}tjd}d}d}|zUtD]K}t|}| d|t |D]9}| |}| d|t D])\} } t ||| |\} } t||| | | ||t||| \} } t||| | | ||q9q*qW|dS|w)Nzpymatbridge.Matlab-C6 ?绽|=waveletdata)pytestZ importorskipnprandom RandomStatestartwaveletsrWavelet set_variablerrandnmodes_compute_matlab_result_check_accuracy_load_matlab_result_pywt_coeffsstop) Matlabmlabrstateepsilonepsilon_pywt_coeffsrrNr pmodemmodemamdr r rtest_accuracy_pymatbridge,s,         r9c Cstjd}d}d}tD]>}t|}t|D]2}||}tD](\}}t |||\} } t |||| | ||t |||\} } t |||| | ||q qq dS)Nrrr) r"r#r$r&rr'rr)r*_load_matlab_resultr,r-) r1r2r3rrr4r r5r6r7r8r r rtest_accuracy_precomputedHs     r;c Cstj|tgdkddr&t|}|d|j|d|jd|}nd|}||}|ds7t d t | d }t | d }||fS) z Compute the result using MATLAB. This function assumes that the Matlab variables `wavelet` and `data` have already been set externally. ) Zcoif6Zcoif7Zcoif8Zcoif9Zcoif10Zcoif11Zcoif12Zcoif13Zcoif14Zcoif15Zcoif16Zcoif17r)ZaxisZLo_DZHi_Dz/[ma, md] = dwt(data, Lo_D, Hi_D, 'mode', '%s');z,[ma, md] = dwt(data, wavelet, 'mode', '%s');successzXMatlab failed to execute the provided code. Check that the wavelet toolbox is installed.r7r8) r"anyarrayrr'r(Zdec_loZdec_hiZrun_code RuntimeErrorZasarrayZ get_variable) r rr6r0rZ mlab_coderesr7r8r r rr+[s   r+cCnt|}d||t|dg}d||t|dg}|tvs"|tvr+td|||t|}t|}||fS)" Load the precomputed result. _r7r8IPrecompted Matlab result not found for wavelet: {0}, mode: {1}, size: {2}lenjoinstrmatlab_result_dictKeyErrorformatr rr6r4Zma_keyZmd_keyr7r8r r rr:r r:cCrA)rBrCZ ma_pywtCoeffsZ md_pywtCoeffsrDrErLr r rr-rMr-c Cst|||\}}tt||d} tt||d} d||t|| f} t| |k| dd||t|| f} t| |k| ddS)NzA[RMS_A > EPSILON] for Mode: %s, Wavelet: %s, Length: %d, rms=%.3g)msgzA[RMS_D > EPSILON] for Mode: %s, Wavelet: %s, Length: %d, rms=%.3g)rZdwtr"sqrtmeanrFr) r rr5r7r8rr2papdZrms_aZrms_drOr r rr,sr,)__doc__ __future__rrrnumpyr"r!Z numpy.testingrrZ pywt._pytestrrrr rIr*Zfamiliessumr&rmarkZslowr9r;r+r:r-r,r r r rs,