Alexander/P-unit_model
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add error messages

Browse Source
This commit is contained in:
a.ott 2020-03-03 15:12:24 +01:00
parent 861557da0d
commit 3fa951624b
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
unittests/testSinusAmplitudeModulation.py
View File

@ -29,7 +29,8 @@ class SinusoidalStimulusTester(unittest.TestCase):
step_size = 0.00005
for base_freq in self.base_frequencies:
stimulus = SinusAmplitudeModulationStimulus(base_freq, contrast, mod_freq, 0, 8)
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size))
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size),
msg="Stimulus values inconsistent with base freq: {:.2f}".format(base_freq))
def test_consistency_contrast(self):
base_freq = 700
@ -39,7 +40,8 @@ class SinusoidalStimulusTester(unittest.TestCase):
step_size = 0.00005
for contrast in self.contrasts:
stimulus = SinusAmplitudeModulationStimulus(base_freq, contrast, mod_freq, 0, 8)
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size))
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size),
msg="Stimulus values inconsistent with contrast: {:.2f}".format(contrast))
def test_consistency_modulation_frequency(self):
contrast = 0.1
@ -48,9 +50,9 @@ class SinusoidalStimulusTester(unittest.TestCase):
duration = 10
step_size = 0.00005
for mod_freq in self.modulation_frequencies:
print(mod_freq)
stimulus = SinusAmplitudeModulationStimulus(base_freq, contrast, mod_freq, 0, 1)
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size))
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size),
msg="Stimulus values inconsistent with mod freq: {:.2f}".format(mod_freq))
def test_consistency_step_size(self):
contrast = 0.1
@ -60,7 +62,8 @@ class SinusoidalStimulusTester(unittest.TestCase):
mod_freq = 10
for step_size in self.step_sizes:
stimulus = SinusAmplitudeModulationStimulus(base_freq, contrast, mod_freq, 0, 8)
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size))
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size),
msg="Stimulus values inconsistent with step_size: {:.3f}ms".format(step_size)*1000)
def test_consistency_time_start(self):
contrast = 0.1
@ -70,7 +73,9 @@ class SinusoidalStimulusTester(unittest.TestCase):
step_size = 0.00005
for time_start in self.time_starts:
stimulus = SinusAmplitudeModulationStimulus(base_freq, contrast, mod_freq, 0, 8)
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size))
self.assertTrue(array_and_time_points_equal(stimulus, time_start, duration, step_size),
msg="Stimulus values inconsistent when the time starts at: {:.2f}s".format(time_start))
def array_and_time_points_equal(stimulus, start, duration, step_size):
precision = 15