From ac96fc154c4ddde3a861a287809ec11297c31ad6 Mon Sep 17 00:00:00 2001 From: Anthony Volk Date: Wed, 14 Jan 2026 23:56:06 +0300 Subject: [PATCH 1/2] fix: Update tests to expect Enhanced CPS dataset by default MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Update three tests that expected `cps_2023.h5` to now expect `enhanced_cps_2024.h5`, reflecting the change in policyengine-us 1.499.0. Fixes #3069 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 --- tests/unit/services/test_economy_service.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/unit/services/test_economy_service.py b/tests/unit/services/test_economy_service.py index ba4a4e586..1220c24b8 100644 --- a/tests/unit/services/test_economy_service.py +++ b/tests/unit/services/test_economy_service.py @@ -663,7 +663,8 @@ def test__given_us_nationwide__returns_correct_sim_options(self): assert sim_options["time_period"] == self.test_time_period assert sim_options["region"] == "us" assert ( - sim_options["data"] == "gs://policyengine-us-data/cps_2023.h5" + sim_options["data"] + == "gs://policyengine-us-data/enhanced_cps_2024.h5" ) def test__given_us_state_ca__returns_correct_sim_options(self): @@ -766,7 +767,8 @@ def test__given_cliff_target__returns_correct_sim_options(self): assert sim_options["time_period"] == time_period assert sim_options["region"] == region assert ( - sim_options["data"] == "gs://policyengine-us-data/cps_2023.h5" + sim_options["data"] + == "gs://policyengine-us-data/enhanced_cps_2024.h5" ) assert sim_options["include_cliffs"] is True @@ -942,7 +944,7 @@ def test__given_us_nationwide__returns_cps_dataset(self): # Test with US nationwide region service = EconomyService() result = service._setup_data("us", "us") - assert result == "gs://policyengine-us-data/cps_2023.h5" + assert result == "gs://policyengine-us-data/enhanced_cps_2024.h5" def test__given_congressional_district__returns_district_dataset(self): # Test with congressional district - returns district-specific dataset From 9a8c589fabb67e8bf763aae8aaa92fc84fe1cadd Mon Sep 17 00:00:00 2001 From: Anthony Volk Date: Thu, 15 Jan 2026 00:12:34 +0300 Subject: [PATCH 2/2] chore: Add changelog entry MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 --- changelog_entry.yaml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/changelog_entry.yaml b/changelog_entry.yaml index e69de29bb..c98a6de16 100644 --- a/changelog_entry.yaml +++ b/changelog_entry.yaml @@ -0,0 +1,4 @@ +- bump: patch + changes: + fixed: + - Updated tests to expect Enhanced CPS dataset by default