Test for the simulate util function.

blakeaw · Jan 30, 2025 · ed54ccc · ed54ccc
1 parent db8fac5
commit ed54ccc
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_util.py b/tests/test_util.py
@@ -0,0 +1,14 @@
+import numpy as np
+from pysb.pkpd.pk_models import twocomp
+from pysb.pkpd import simulate
+
+
+def test_simulate():
+    tspan = np.arange(10)
+    sim_traj = simulate(twocomp, tspan)
+    assert len(sim_traj) == 10
+    assert len(sim_traj[0]) == 2
+
+
+if __name__ == "__main__":
+    test_simulate()