extra test for correct behaviour of sim_time_series

darrenjw · Jan 2, 2025 · c69ea90 · c69ea90
1 parent 7bf3052
commit c69ea90
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_spn.py b/tests/test_spn.py
@@ -24,9 +24,19 @@ def test_bd():
     step = bd.step_gillespie()
     out = smfsb.sim_time_series(bd.m, 0, 100, 0.1, step)
     assert out.shape == (1000, 1)
+    assert out[0, :] == bd.m
     assert out[100, 0] >= 0
 
 
+def test_sim_time_series():
+    bd = smfsb.models.bd()
+    step = bd.step_gillespie()
+    out = smfsb.sim_time_series(bd.m, 0, 5, 1, step)
+    assert out.shape == (6, 1)
+    assert out[0, :] == bd.m
+    assert out[4, 0] >= 0
+
+
 def test_dimer():
     dimer = smfsb.models.dimer()
     step = dimer.step_gillespie()