We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 72f46c5 commit 0f4e131Copy full SHA for 0f4e131
1 file changed
tests/pipelines/test_mp300k_gate_inputs.py
@@ -24,6 +24,14 @@ def _write_minimal_policyengine_dataset(path: Path, *, period: int = 2024) -> Pa
24
"household_weight": {str(period): np.asarray([10.0, 20.0])},
25
"person_id": {str(period): np.asarray([1, 2, 3])},
26
"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])},
35
}
36
return write_policyengine_us_time_period_dataset(arrays, path)
37
0 commit comments