def simulate_pac_default(n_points=128,
fs=200.,
high_fq=50.,
low_fq=3.,
low_fq_width=1.,
noise_level=0.1,
random_state=42,
*args,
**kwargs):
"""Call simulate_pac with default values, used for testing only."""
return simulate_pac(n_points,
fs,
high_fq,
low_fq,
low_fq_width,
noise_level,
random_state=random_state,
*args,
**kwargs)
from pactools.simulate_pac import simulate_pac
# Parameters used for the simulated signal in the test
low_fq_range = [1., 3., 5., 7.]
high_fq_range = [25., 50., 75.]
n_low = len(low_fq_range)
n_high = len(high_fq_range)
high_fq = high_fq_range[1]
low_fq = low_fq_range[1]
n_points = 1024
fs = 200.
signal = simulate_pac(n_points=n_points,
fs=fs,
high_fq=high_fq,
low_fq=low_fq,
low_fq_width=1.,
noise_level=0.1,
random_state=0)
def fast_peak_locking(low_sig=signal,
high_sig=None,
mask=None,
fs=fs,
low_fq=low_fq,
*args,
**kwargs):
return PeakLocking(fs=fs, low_fq=low_fq, *args,
**kwargs).fit(low_sig=low_sig,
high_sig=high_sig,
ALL_MODELS = [DAR, AR, HAR, StableDAR]
# Parameters used for the simulated sigin in the test
low_fq_range = [1., 3., 5., 7.]
high_fq_range = [25., 50., 75.]
n_low = len(low_fq_range)
n_high = len(high_fq_range)
high_fq = high_fq_range[1]
low_fq = low_fq_range[1]
n_points = 1024
fs = 200.
_sigin = simulate_pac(n_points=n_points,
fs=fs,
high_fq=high_fq,
low_fq=low_fq,
low_fq_width=1.,
noise_level=0.3,
random_state=0)
_sigdriv = simulate_pac(n_points=n_points,
fs=fs,
high_fq=high_fq,
low_fq=low_fq,
low_fq_width=1.,
noise_level=0.,
random_state=0,
high_fq_amp=0,
return_driver=True)
_sigdriv_imag = np.imag(_sigdriv)
_sigdriv = np.real(_sigdriv)
_noise = np.random.RandomState(0).randn(n_points)