test fixes

biosustain · Jul 19, 2023 · e7b3abc · e7b3abc
1 parent da9a7a9
commit e7b3abc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -74,6 +74,7 @@ packages = [
     "maud.data.example_inputs.example_ode",
     "maud.data.example_outputs",
     "maud.data.example_outputs.linear",
+    "maud.data.example_outputs.linear.user_input",
 ]
 [tool.setuptools.package-data]
 "*" = ["*.stan", "*.json", "*.toml"]

diff --git a/tests/test_unit/test_model_ode.py b/tests/test_unit/test_model_ode.py
@@ -4,13 +4,12 @@
 from pathlib import Path
 from typing import List, Tuple
 
-import cmdstanpy
 import importlib_resources
 import pytest
 from numpy import isclose
 
-from maud import stan
 from maud.data.example_inputs import example_ode
+from maud.running_stan import load_stan_model
 
 
 @dataclass
@@ -58,9 +57,7 @@ class TestCase:
 @pytest.mark.parametrize("test_case", TEST_CASES)
 def test_model_ode(test_case):
     """Test that the function get_input_data behaves as expected."""
-    model = cmdstanpy.CmdStanModel(
-        stan_file=importlib_resources.files(stan).joinpath("model.stan")
-    )
+    model = load_stan_model("model", stanc_options={}, cpp_options={})
     mcmc = model.sample(
         data=str(test_case.input_data_path),
         inits=str(test_case.inits_path),