update net_004_alt test

AMICI-dev · Dec 2, 2024 · ea1c75e · ea1c75e
1 parent b201f4e
commit ea1c75e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/tests/sciml/testsuite b/tests/sciml/testsuite
diff --git a/tests/sciml/testsuite.py b/tests/sciml/testsuite.py
@@ -59,7 +59,13 @@ def _test_net(test):
     ):
         return
 
-    ml_models = PetabScimlStandard.load_data(test / solutions["net_file"])
+    if test.stem.endswith("_alt"):
+        net_file = (
+            test.parent / test.stem.replace("_alt", "") / solutions["net_file"]
+        )
+    else:
+        net_file = test / solutions["net_file"]
+    ml_models = PetabScimlStandard.load_data(net_file)
 
     nets = {}
     outdir = Path(__file__).parent / "models" / test.stem
@@ -151,6 +157,7 @@ def _test_net(test):
                     "net_021",
                     "net_022",  # Conv layers
                     "net_004",
+                    "net_004_alt",
                     "net_005",
                     "net_006",
                     "net_007",
@@ -277,7 +284,6 @@ def _test_ude(test):
     )
 
     # gradient
-
     sllh, _ = eqx.filter_grad(run_simulations, has_aux=True)(
         jax_problem,
         solver=diffrax.Tsit5(),
+22 −0		test_cases/net_004_alt/create_testdata/net.jl
+21 −0		test_cases/net_004_alt/create_testdata/net.py
+151 −0		test_cases/net_004_alt/net_input_1.tsv
+151 −0		test_cases/net_004_alt/net_input_2.tsv
+151 −0		test_cases/net_004_alt/net_input_3.tsv
+21 −0		test_cases/net_004_alt/net_output_1.tsv
+21 −0		test_cases/net_004_alt/net_output_2.tsv
+21 −0		test_cases/net_004_alt/net_output_3.tsv
+74 −0		test_cases/net_004_alt/net_ps_1.tsv
+74 −0		test_cases/net_004_alt/net_ps_2.tsv
+74 −0		test_cases/net_004_alt/net_ps_3.tsv
+29 −0		test_cases/net_004_alt/solutions.yaml