have amplitude be a optional value with default one

This commit is contained in:
a.ott 2020-01-27 14:13:19 +01:00
parent 5165046f20
commit 3ffa6d5dbe
2 changed files with 2 additions and 2 deletions

View File

@ -4,7 +4,7 @@ import numpy as np
class SinusAmplitudeModulationStimulus(AbstractStimulus): class SinusAmplitudeModulationStimulus(AbstractStimulus):
def __init__(self, carrier_frequency, amplitude, contrast, modulation_frequency, start_time=0, duration=np.inf): def __init__(self, carrier_frequency, contrast, modulation_frequency, amplitude=1, start_time=0, duration=np.inf):
self.contrast = contrast self.contrast = contrast
self.modulation_frequency = modulation_frequency self.modulation_frequency = modulation_frequency
self.amplitude = amplitude self.amplitude = amplitude

View File

@ -56,7 +56,7 @@ def test_lifac_noise():
total_time = duration + 2*start total_time = duration + 2*start
step_size = model.get_parameters()["step_size"]/1000 step_size = model.get_parameters()["step_size"]/1000
time = np.arange(0, total_time, step_size) time = np.arange(0, total_time, step_size)
stimulus = SinusAmplitudeModulationStimulus(700, 20, 0.2, 10, start, duration) stimulus = SinusAmplitudeModulationStimulus(700, 0.2, 10, 20, start, duration)
model.simulate(stimulus, total_time) model.simulate(stimulus, total_time)