o h@sdZddlZddlZddlmZddlZddlmZ ddlm Z m Z m Z gdZ dZedd eDgZd d Ze ejjd d Ze ejjddZddZddZddZddZdS)zs Test used to verify PyWavelets Discrete Wavelet Transform computation accuracy against MathWorks Wavelet Toolbox. N)assert_)matlab_result_dict_dwt)size_setuses_precomputeduses_pymatbridge) )zerozpd)constantsp0) symmetricsym)reflectsymw)periodicppd)smoothsp1) periodizationper) antisymmetricasym) antireflectasymw)dbr coifbiorrbiocCsg|]}t|qS)pywtwavelist).0namerr_/var/www/html/scripts/venv/lib/python3.10/site-packages/pywt/tests/test_matlab_compatibility.py sr#cCs8tdkrtt|jdgd}|S|j|jdf}|S)z) Return the sizes to test for wavelet w. full()diiiP)rlistrangedec_len)w data_sizesrrr"_get_data_sizes!sr.c 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)pytest importorskipnprandom RandomStatestartwaveletsrWavelet set_variabler.randnmodes_compute_matlab_result_check_accuracy_load_matlab_result_pywt_coeffsstop) Matlabmlabrstateepsilonepsilon_pywt_coeffsr2r,Nr3pmodemmodemamdrrr"test_accuracy_pymatbridge+s,         rMc Cstjd}d}d}tD]>}t|}t|D]2}||}tD](\}}t |||\} } t |||| | ||t |||\} } t |||| | ||q qq dS)Nr/r0r1) r6r7r8r:rr;r.r=r>_load_matlab_resultr@rA) rErFrGr2r,rHr3rIrJrKrLrrr"test_accuracy_precomputedGs     rOc Cstj|tgdkddr(t|}|d|j|d|jd|d}nd|d}||}|d s;t 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. ) coif6coif7coif8coif9coif10coif11coif12coif13coif14coif15coif16coif17r)axisLo_DHi_Dz*[ma, md] = dwt(data, Lo_D, Hi_D, 'mode', 'z');z'[ma, md] = dwt(data, wavelet, 'mode', 'successzXMatlab failed to execute the provided code. Check that the wavelet toolbox is installed.rKrL) r6anyarrayrr;r<dec_lodec_hirun_code RuntimeErrorasarray get_variable) r3r2rJrDr, mlab_coderesrKrLrrr"r?Zs   r?cCvt|}d||t|dg}d||t|dg}|tvs"|tvr/td|d|d|t|}t|}||fS)" Load the precomputed result. _rKrL0Precompted Matlab result not found for wavelet: , mode: , size: lenjoinstrmatlab_result_dictKeyErrorr3r2rJrHma_keymd_keyrKrLrrr"rNq"rNcCrj)rkrl ma_pywtCoeffs md_pywtCoeffsrmrnrorprvrrr"rAryrAc 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)rdwtr6sqrtmeanrqr) r3r,rIrKrLr2rFpapdrms_arms_dr}rrr"r@sr@)__doc__numpyr6r4 numpy.testingrr pywt._pytestrrtrrrr>familiessumr:r.markslowrMrOr?rNrAr@rrrr"s*