Skip to content

Commit 0f4e131

Browse files
committed
Fix packaged mp-300k gate fixture
1 parent 72f46c5 commit 0f4e131

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

tests/pipelines/test_mp300k_gate_inputs.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,14 @@ def _write_minimal_policyengine_dataset(path: Path, *, period: int = 2024) -> Pa
2424
"household_weight": {str(period): np.asarray([10.0, 20.0])},
2525
"person_id": {str(period): np.asarray([1, 2, 3])},
2626
"person_household_id": {str(period): np.asarray([1, 1, 2])},
27+
"tax_unit_id": {str(period): np.asarray([10, 20])},
28+
"person_tax_unit_id": {str(period): np.asarray([10, 10, 20])},
29+
"spm_unit_id": {str(period): np.asarray([100, 200])},
30+
"person_spm_unit_id": {str(period): np.asarray([100, 100, 200])},
31+
"family_id": {str(period): np.asarray([1000, 2000])},
32+
"person_family_id": {str(period): np.asarray([1000, 1000, 2000])},
33+
"marital_unit_id": {str(period): np.asarray([10000, 10001, 20000])},
34+
"person_marital_unit_id": {str(period): np.asarray([10000, 10001, 20000])},
2735
}
2836
return write_policyengine_us_time_period_dataset(arrays, path)
2937

0 commit comments

Comments
 (0)