U h @sdZddlmZmZmZddlZddlZddlm Z ddl Z ddl m Z m Z mZddl mZddd d d d d ddg ZdZeddeDgZddZe ejjddZe ejjddZddZddZddZdd ZdS)!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)ZzeroZzpd)ZconstantZsp0)Z symmetricsym)ZreflectZsymw)ZperiodicZppd)ZsmoothZsp1)Z periodizationZper)Z antisymmetricZasym)Z antireflectZasymw)dbr ZcoifZbiorZrbiocCsg|]}t|qS)pywtZwavelist).0namer r H/tmp/pip-unpacked-wheel-mu97bu1x/pywt/tests/test_matlab_compatibility.py srcCs<tdkr(tt|jddddddg}n|j|jdf}|S) z) Return the sizes to test for wavelet w. full(diiiP)rlistrangeZdec_len)wZ data_sizesr r r_get_data_sizes"s  rc Cstd}|}tjd}d}d}|ztD]}t |}| d|t |D]r}| |}| d|t D]R\} } t||| |\} } t||| | | ||t||| \} } t||| | | ||qrqTq2W5|XdS)Nzpymatbridge.Matlab-C6 ?绽|=waveletdata)pytestZ importorskipnprandom RandomStatestartstopwaveletsr Wavelet set_variablerrandnmodes_compute_matlab_result_check_accuracy_load_matlab_result_pywt_coeffs) ZMatlabmlabrstateepsilonepsilon_pywt_coeffsrrNrpmodemmodemamdr r rtest_accuracy_pymatbridge,s&        r7c Cstjd}d}d}tD]|}t|}t|D]d}||}tD]P\}}t |||\} } t |||| | ||t |||\} } t |||| | ||q@q.qdS)Nrrr) r!r"r#r&r r'rr)r*_load_matlab_resultr,r-) r/r0r1rrr2rr3r4r5r6r r rtest_accuracy_precomputedHs     r9c Cstj|tddddddddd d d d g kd dr`t|}|d|j|d|jd|}nd|}||}|dst dt | 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.r5r6) r!anyarrayr r'r(Zdec_loZdec_hiZrun_code RuntimeErrorZasarrayZ get_variable) rrr4r.rZ mlab_coderesr5r6r r rr+[s0   r+cCsnt|}d||t|dg}d||t|dg}|tksD|tkrVtd|||t|}t|}||fS)" Load the precomputed result. _r5r6IPrecompted Matlab result not found for wavelet: {0}, mode: {1}, size: {2}lenjoinstrmatlab_result_dictKeyErrorformatrrr4r2Zma_keyZmd_keyr5r6r r rr8rs r8cCsnt|}d||t|dg}d||t|dg}|tksD|tkrVtd|||t|}t|}||fS)r?r@Z ma_pywtCoeffsZ md_pywtCoeffsrArBrIr r rr-s r-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)r Zdwtr!sqrtZmeanrCr) rrr3r5r6rr0papdZrms_aZrms_drKr r rr,sr,)__doc__ __future__rrrZnumpyr!r Z numpy.testingrr Z pywt._pytestrrrr rFr*Zfamiliessumr&rmarkZslowr7r9r+r8r-r,r r r rs<