diff --git a/neurodsp/tests/sim/test_signals.py b/neurodsp/tests/sim/test_signals.py index c68e1147..e703b90b 100644 --- a/neurodsp/tests/sim/test_signals.py +++ b/neurodsp/tests/sim/test_signals.py @@ -4,6 +4,8 @@ import numpy as np +from neurodsp.sim.params import get_base_params + from neurodsp.sim.signals import * ################################################################################################### @@ -44,6 +46,16 @@ def test_simulations(): assert sims_full.has_signals assert sims_full.has_params + # Test pre-initialization + sims_pre = Simulations(n_sigs, params, 'sim_func') + assert len(sims_pre) == n_sigs + assert np.sum(sims_pre.signals) == 0 + assert sims_pre.has_signals and sims_pre.has_params + for ind, sig in enumerate(sigs): + sims_pre.add_signal(sig, ind) + assert len(sims_pre) == n_sigs + assert np.sum(sims_pre.signals) != 0 + def test_variable_simulations(): # Test empty initialization @@ -80,6 +92,16 @@ def test_variable_simulations(): assert sims_full.has_signals assert sims_full.has_params + # Test pre-initialization + sims_pre = VariableSimulations(n_sigs, get_base_params(params), 'sim_func') + assert len(sims_pre) == n_sigs + assert np.sum(sims_pre.signals) == 0 + assert sims_pre.has_signals and sims_pre.has_params + for ind, (sig, cparams) in enumerate(zip(sigs, params)): + sims_pre.add_signal(sig, cparams, ind) + assert len(sims_pre) == n_sigs + assert np.sum(sims_pre.signals) != 0 + def test_variable_simulations_add(): sig = np.array([1, 2, 3, 4, 5])