Merge pull request #428 from PolicyEngine/codex/policyengine-0x-simulation

[codex] Prepare simulation service for policyengine 0.13.0

Author: anth-volk

projects/policyengine-api-simulation/pyproject.toml

@@ -16,7 +16,7 @@ dependencies = [
     "pydantic-settings (>=2.7.1,<3.0.0)",
     "opentelemetry-instrumentation-fastapi (>=0.51b0,<0.52)",
     "policyengine-fastapi",
-    "policyengine==0.12.1",
+    "policyengine==0.13.0",
     "policyengine-core>=3.23.5",
     "policyengine-uk>=2.22.8",
     "policyengine-us>=1.370.2",

projects/policyengine-api-simulation/src/modal/app.py

@@ -47,7 +47,7 @@ def get_app_name(us_version: str, uk_version: str) -> str:
     .pip_install(
         f"policyengine-us=={US_VERSION}",
         f"policyengine-uk=={UK_VERSION}",
-        "policyengine>=0.10.1,<1",
+        "policyengine==0.13.0",
         "tables>=3.10.2",
         "logfire",
     )

projects/policyengine-api-simulation/tests/test_policyengine_dependency_source.py

@@ -0,0 +1,40 @@
+"""Regression tests for the policyengine dependency version configuration."""
+
+import re
+import tomllib
+from pathlib import Path
+
+REPO_ROOT = Path(__file__).parent.parent
+PYPROJECT_PATH = REPO_ROOT / "pyproject.toml"
+MODAL_APP_PATH = REPO_ROOT / "src" / "modal" / "app.py"
+POLICYENGINE_DEPENDENCY_PREFIX = "policyengine=="
+
+
+def _load_toml(path: Path) -> dict:
+    with path.open("rb") as file:
+        return tomllib.load(file)
+
+
+def _get_pyproject_policyengine_dependency(pyproject: dict) -> str:
+    dependencies = pyproject["project"]["dependencies"]
+    return next(
+        dep for dep in dependencies if dep.startswith(POLICYENGINE_DEPENDENCY_PREFIX)
+    )
+
+
+def _get_modal_policyengine_dependency(modal_source: str) -> str:
+    match = re.search(
+        r'"(policyengine==[^"]+)"',
+        modal_source,
+    )
+    assert match is not None, "Modal app should install a pinned policyengine version"
+    return match.group(1)
+
+
+def test_policyengine_dependency_version_is_pinned_consistently():
+    pyproject = _load_toml(PYPROJECT_PATH)
+    pyproject_dependency = _get_pyproject_policyengine_dependency(pyproject)
+    modal_dependency = _get_modal_policyengine_dependency(MODAL_APP_PATH.read_text())
+
+    assert pyproject_dependency.startswith(POLICYENGINE_DEPENDENCY_PREFIX)
+    assert modal_dependency == pyproject_dependency