import numpy as np

from fooof import FOOOFGroup
from fooof.objs import fit_fooof_3d
from fooof.sim import gen_group_power_spectra

np.random.seed(0)

ap_params = [-2, 0]
pe_params = [[10, 5, 5], [25, 5, 5]]

freqs, powers = gen_group_power_spectra(10, (1, 50), ap_params, pe_params)

# Shape: (100, 10, 99)
powers_3d = np.array([powers] * 100)

fg = FOOOFGroup(verbose=False)

%%timeit
fgs = fit_fooof_3d(fg, freqs, powers_3d, n_jobs=4)