U f{j @sddlZddlmZddlZddlmZddlmZm Z m Z dZ dZ de Z dZee e ee eZed eefZee d gd gd Zeededdeed<d ZeeeZejd ZddZeeddgddgddgZddZdS)N)assert_array_equal)morlet)ctps _prob_kuiper_compute_normalized_phaseg333333ӿg@@?X)Zn_cycles#*cCstt|}|ddf}t|}t| ||}tdd|D}t||||||gd}|j |ddfks|t |S)z'Generate ground truth and testing data.r rcSsg|]}tt|qS)nproll single_trial).0ir r `/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/tests/test_ctps.py &szget_data..r ) rZtilerrngZ random_samplerandintarrayZ concatenateZreshapeshapeAssertionError)n_trialsj_extentZ ground_truthZmy_shapeZ random_dataZ rand_intsZ jittered_datadatar r rget_data s    ri,g333333?g?c CstD]z\}\}}}t||}t|\}}}t|}t|dd\} } } t|||g| | |gD]R\} } t| | | dks~t| dkst| dkst| dks\tq\|dks|dkst|j |j kst|j |j ddkst|ddks tt t |ddks&t|d|kss$  "