Add flux span test

gyorilab · Jul 13, 2023 · 8951193 · 8951193
1 parent beccf19
commit 8951193
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_flux_span.py b/tests/test_flux_span.py
@@ -0,0 +1,15 @@
+import os
+import json
+from mira.sources.askenet.flux_span import reproduce_ode_semantics
+
+HERE = os.path.dirname(os.path.abspath(__file__))
+DATA = os.path.join(HERE, 'sir_flux_span.json')
+
+
+def test_flux_span_ode_semantics():
+ with open(DATA) as fh:
+ flux_span = json.load(fh)
+ tm = reproduce_ode_semantics(flux_span)
+ assert len(tm.templates) == 10
+ assert len(tm.parameters) == 4
+ assert all(t.rate_law for t in tm.templates)