U f{j@sDddlZddlmZddlZddlmZddlmZeddZ dS)N)assert_array_equal)requires_sklearn) TimeFrequencyc Csddlm}d}dddg}t|dd}d D]}tjtt||d q*||}d \}}}tj|||}| |d t|dd}| |d | |} t | j ||||g| |d d dd d f} t | j |||gt|dd d}| |} t | j ||||d gd S)zTest TimeFrequency.r)cloned)sfreq)Z avg_powerZfooN)output) r Nr)r Zdecim)Z sklearn.baserrpytestZraises ValueErrornprandomZrandfitZ fit_transformZ transformrshape) rZn_freqsZfreqstfr Zn_epochsZn_chansZn_timesXZXtre/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/decoding/tests/test_time_frequency.pytest_timefrequencys&         r) numpyrZ numpy.testingrrZ mne.utilsrZmne.decoding.time_frequencyrrrrrrs