Skip to content

Commit bd80504

Browse files
authored
Merge pull request #3070 from PolicyEngine/fix/fix-cps-tests
fix: Update tests to expect Enhanced CPS dataset by default
2 parents 0373c2d + 9a8c589 commit bd80504

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

changelog_entry.yaml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
- bump: patch
2+
changes:
3+
fixed:
4+
- Updated tests to expect Enhanced CPS dataset by default

tests/unit/services/test_economy_service.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -663,7 +663,8 @@ def test__given_us_nationwide__returns_correct_sim_options(self):
663663
assert sim_options["time_period"] == self.test_time_period
664664
assert sim_options["region"] == "us"
665665
assert (
666-
sim_options["data"] == "gs://policyengine-us-data/cps_2023.h5"
666+
sim_options["data"]
667+
== "gs://policyengine-us-data/enhanced_cps_2024.h5"
667668
)
668669

669670
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):
766767
assert sim_options["time_period"] == time_period
767768
assert sim_options["region"] == region
768769
assert (
769-
sim_options["data"] == "gs://policyengine-us-data/cps_2023.h5"
770+
sim_options["data"]
771+
== "gs://policyengine-us-data/enhanced_cps_2024.h5"
770772
)
771773
assert sim_options["include_cliffs"] is True
772774

@@ -942,7 +944,7 @@ def test__given_us_nationwide__returns_cps_dataset(self):
942944
# Test with US nationwide region
943945
service = EconomyService()
944946
result = service._setup_data("us", "us")
945-
assert result == "gs://policyengine-us-data/cps_2023.h5"
947+
assert result == "gs://policyengine-us-data/enhanced_cps_2024.h5"
946948

947949
def test__given_congressional_district__returns_district_dataset(self):
948950
# Test with congressional district - returns district-specific dataset

0 commit comments

Comments
 (0)