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 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