policyengine-us 1.403.1__py3-none-any.whl → 1.404.1__py3-none-any.whl

policyengine_us/tests/policy/baseline/gov/states/tx/dart/reduced_fare/eligibility/tx_dart_reduced_fare_eligible.yaml

@@ -0,0 +1,146 @@
+- name: Senior citizen eligible
+  period: 2025
+  input:
+    state_code: TX
+    age: 67
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: Child eligible through age
+  period: 2025
+  input:
+    state_code: TX
+    age: 8
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: Disabled person eligible
+  period: 2025
+  input:
+    state_code: TX
+    age: 45
+    is_disabled: true
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: Veteran eligible
+  period: 2025
+  input:
+    state_code: TX
+    age: 35
+    is_veteran: true
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: Full-time student eligible
+  period: 2025
+  input:
+    state_code: TX
+    age: 20
+    is_full_time_student: true
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: SNAP recipient eligible through program enrollment
+  period: 2025
+  input:
+    state_code: TX
+    age: 40
+    snap: 300
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: Medicaid recipient eligible
+  period: 2025
+  input:
+    state_code: TX
+    age: 25
+    medicaid: 5_000
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: CHIP recipient eligible
+  period: 2025
+  input:
+    state_code: TX
+    age: 12
+    chip: 3_000
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: WIC recipient eligible
+  period: 2025
+  input:
+    state_code: TX
+    age: 28
+    wic: 100
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: TANF recipient eligible
+  period: 2025
+  input:
+    state_code: TX
+    age: 32
+    tanf: 400
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: Medicare eligible person qualifies
+  period: 2025
+  input:
+    state_code: TX
+    age: 68
+    is_medicare_eligible: true
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: Non-eligible adult
+  period: 2025
+  input:
+    state_code: TX
+    age: 35
+    is_disabled: false
+    is_veteran: false
+    is_full_time_student: false
+    snap: 0
+    medicaid: 0
+  output:
+    tx_dart_reduced_fare_eligible: false
+
+- name: Multiple eligibility criteria (veteran and student)
+  period: 2025
+  input:
+    state_code: TX
+    age: 24
+    is_veteran: true
+    is_full_time_student: true
+  output:
+    tx_dart_reduced_fare_eligible: true
+
+- name: Family with mixed eligibility
+  period: 2025
+  input:
+    people:
+      veteran_parent:
+        age: 35
+        is_veteran: true
+      spouse:
+        age: 33
+        is_veteran: false
+      child:
+        age: 10
+        is_veteran: false
+      senior:
+        age: 70
+        is_veteran: false
+    spm_units:
+      spm_unit:
+        members: [veteran_parent, spouse, child, senior]
+        snap: 200
+    households:
+      household:
+        members: [veteran_parent, spouse, child, senior]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_eligible: [true, true, true, true]

policyengine_us/tests/policy/baseline/gov/states/tx/dart/reduced_fare/eligibility/tx_dart_reduced_fare_program_eligible.yaml

@@ -0,0 +1,206 @@
+- name: Person receiving SNAP benefits is eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      person1:
+        age: 30
+        employment_income: 15_000
+    spm_units:
+      spm_unit:
+        members: [person1]
+        snap: 200
+    households:
+      household:
+        members: [person1]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_program_eligible: true
+
+- name: Person receiving Medicaid is eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      person1:
+        age: 25
+        is_disabled: false
+        medicaid: 1
+    households:
+      household:
+        members: [person1]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_program_eligible: true
+
+- name: Child receiving CHIP is eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      child:
+        age: 10
+        chip: 1
+    households:
+      household:
+        members: [child]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_program_eligible: true
+
+- name: Person receiving WIC is eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      mother:
+        age: 28
+        wic: 100
+    households:
+      household:
+        members: [mother]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_program_eligible: true
+
+- name: Person receiving TANF is eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      person1:
+        age: 35
+    spm_units:
+      spm_unit:
+        members: [person1]
+        tanf: 250
+    households:
+      household:
+        members: [person1]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_program_eligible: true
+
+- name: Medicare-eligible person is eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      senior:
+        age: 67
+        is_disabled: false
+    households:
+      household:
+        members: [senior]
+        state_code: TX
+  output:
+    # Age 67 makes person Medicare eligible
+    is_medicare_eligible: true
+    tx_dart_reduced_fare_program_eligible: true
+
+- name: Disabled person receiving SSDI long enough for Medicare is eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      person1:
+        age: 45
+        is_disabled: true
+        social_security_disability: 12_000
+        months_receiving_social_security_disability: 24  # 24 months required for Medicare
+    households:
+      household:
+        members: [person1]
+        state_code: TX
+  output:
+    # Disability with 24+ months of SSDI makes person Medicare eligible
+    is_medicare_eligible: true
+    tx_dart_reduced_fare_program_eligible: true
+
+- name: Person with no qualifying programs is not eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      person1:
+        age: 40
+        employment_income: 75_000
+        is_disabled: false
+    households:
+      household:
+        members: [person1]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_program_eligible: false
+
+- name: Person with multiple qualifying programs is eligible for reduced fare program
+  period: 2025
+  input:
+    people:
+      person1:
+        age: 30
+        employment_income: 12_000
+        medicaid: 1
+    spm_units:
+      spm_unit:
+        members: [person1]
+        snap: 150
+        tanf: 200
+    households:
+      household:
+        members: [person1]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_program_eligible: true
+
+- name: Family with mixed eligibility - only program recipients are eligible
+  period: 2025
+  input:
+    people:
+      parent1:
+        age: 35
+        employment_income: 20_000
+      parent2:
+        age: 33
+        employment_income: 35_000
+      child1:
+        age: 8
+        chip: 1  # Child receives CHIP
+      child2:
+        age: 12
+        chip: 1  # Child receives CHIP
+    spm_units:
+      spm_unit:
+        members: [parent1, parent2, child1, child2]
+        snap: 400  # Family receives SNAP
+    households:
+      household:
+        members: [parent1, parent2, child1, child2]
+        state_code: TX
+  output:
+    # All family members get SNAP, so all are eligible
+    tx_dart_reduced_fare_program_eligible: [true, true, true, true]
+
+- name: Young professional with no assistance programs is not eligible
+  period: 2025
+  input:
+    people:
+      person1:
+        age: 28
+        employment_income: 55_000
+        is_disabled: false
+    households:
+      household:
+        members: [person1]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_program_eligible: false
+
+- name: Person outside Texas is not eligible regardless of programs
+  period: 2025
+  input:
+    people:
+      person1:
+        age: 30
+    spm_units:
+      spm_unit:
+        members: [person1]
+        snap: 200
+    households:
+      household:
+        members: [person1]
+        state_code: CA  # California, not Texas
+  output:
+    tx_dart_reduced_fare_program_eligible: false

policyengine_us/tests/policy/baseline/gov/states/tx/dart/reduced_fare/tx_dart_reduced_fare_benefit.yaml

@@ -0,0 +1,145 @@
+- name: Eligible senior receives annual benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 68
+  output:
+    tx_dart_reduced_fare_benefit: 756  # ($126 - $63) * 12 months
+
+- name: Eligible child receives annual benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 10
+  output:
+    tx_dart_reduced_fare_benefit: 756
+
+- name: Eligible disabled person receives benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 45
+    is_disabled: true
+  output:
+    tx_dart_reduced_fare_benefit: 756
+
+- name: Eligible veteran receives benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 35
+    is_veteran: true
+  output:
+    tx_dart_reduced_fare_benefit: 756
+
+- name: Eligible student receives benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 20
+    is_full_time_student: true
+  output:
+    tx_dart_reduced_fare_benefit: 756
+
+- name: SNAP recipient receives benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 40
+    snap: 300
+  output:
+    tx_dart_reduced_fare_benefit: 756
+
+- name: Non-eligible adult receives no benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 35
+    is_disabled: false
+    is_veteran: false
+    is_full_time_student: false
+    snap: 0
+    medicaid: 0
+    chip: 0
+    tanf: 0
+    wic: 0
+    is_medicare_eligible: false
+  output:
+    tx_dart_reduced_fare_benefit: 0
+
+- name: Child under 5 not eligible for reduced fare benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 3
+    snap: 0
+    medicaid: 0
+    chip: 0
+    tanf: 0
+    wic: 0
+    is_medicare_eligible: false
+  output:
+    tx_dart_reduced_fare_benefit: 0
+
+- name: Family with mixed eligibility receives appropriate benefits
+  period: 2025
+  input:
+    people:
+      veteran_parent:
+        age: 35
+        is_veteran: true
+        is_medicare_eligible: false
+        medicaid: 0
+        chip: 0
+        wic: 0
+      spouse:
+        age: 33
+        is_veteran: false
+        is_medicare_eligible: false
+        medicaid: 0
+        chip: 0
+        wic: 0
+      child:
+        age: 12
+        is_veteran: false
+        is_medicare_eligible: false
+        medicaid: 0
+        chip: 0
+        wic: 0
+      senior:
+        age: 68
+        is_veteran: false
+        is_medicare_eligible: false
+        medicaid: 0
+        chip: 0
+        wic: 0
+    spm_units:
+      spm_unit:
+        members: [veteran_parent, spouse, child, senior]
+        snap: 0
+        tanf: 0
+    households:
+      household:
+        members: [veteran_parent, spouse, child, senior]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_benefit: [756, 0, 756, 756]
+
+- name: Household on TANF all members receive benefit
+  period: 2025
+  input:
+    people:
+      parent:
+        age: 30
+      child:
+        age: 8
+    spm_units:
+      spm_unit:
+        members: [parent, child]
+        tanf: 400
+    households:
+      household:
+        members: [parent, child]
+        state_code: TX
+  output:
+    tx_dart_reduced_fare_benefit: [756, 756]

policyengine_us/tests/policy/baseline/gov/states/tx/dart/tx_dart_benefit_person.yaml

@@ -0,0 +1,141 @@
+- name: Child under 5 gets free ride benefit (full monthly pass value)
+  period: 2025
+  input:
+    state_code: TX
+    age: 3
+  output:
+    tx_dart_benefit_person: 1_512  # $126 * 12 months
+
+- name: Senior gets reduced fare benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 68
+  output:
+    tx_dart_benefit_person: 756  # ($126 - $63) * 12 months
+
+- name: Child age 10 gets reduced fare benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 10
+  output:
+    tx_dart_benefit_person: 756
+
+- name: Disabled person gets reduced fare benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 45
+    is_disabled: true
+  output:
+    tx_dart_benefit_person: 756
+
+- name: Veteran gets reduced fare benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 35
+    is_veteran: true
+  output:
+    tx_dart_benefit_person: 756
+
+- name: Student gets reduced fare benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 20
+    is_full_time_student: true
+  output:
+    tx_dart_benefit_person: 756
+
+- name: SNAP recipient gets reduced fare benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 40
+    snap: 300
+  output:
+    tx_dart_benefit_person: 756
+
+- name: Non-eligible adult gets no benefit
+  period: 2025
+  input:
+    state_code: TX
+    age: 35
+    is_disabled: false
+    is_veteran: false
+    is_full_time_student: false
+    snap: 0
+    medicaid: 0
+    chip: 0
+    tanf: 0
+    wic: 0
+    is_medicare_eligible: false
+  output:
+    tx_dart_benefit_person: 0
+
+- name: Family with toddler and senior
+  period: 2025
+  input:
+    people:
+      parent1:
+        age: 35
+        is_veteran: false
+        is_medicare_eligible: false
+        medicaid: 0
+        chip: 0
+        wic: 0
+      parent2:
+        age: 33
+        is_veteran: false
+        is_medicare_eligible: false
+        medicaid: 0
+        chip: 0
+        wic: 0
+      toddler:
+        age: 2
+        is_veteran: false
+        is_medicare_eligible: false
+        medicaid: 0
+        chip: 0
+        wic: 0
+      senior:
+        age: 70
+        is_veteran: false
+        is_medicare_eligible: false
+        medicaid: 0
+        chip: 0
+        wic: 0
+    spm_units:
+      spm_unit:
+        members: [parent1, parent2, toddler, senior]
+        snap: 0
+        tanf: 0
+    households:
+      household:
+        members: [parent1, parent2, toddler, senior]
+        state_code: TX
+  output:
+    tx_dart_benefit_person: [0, 0, 1_512, 756]
+
+- name: Family on TANF with child under 5
+  period: 2025
+  input:
+    people:
+      parent:
+        age: 30
+      older_child:
+        age: 8
+      toddler:
+        age: 3
+    spm_units:
+      spm_unit:
+        members: [parent, older_child, toddler]
+        tanf: 400
+    households:
+      household:
+        members: [parent, older_child, toddler]
+        state_code: TX
+  output:
+    tx_dart_benefit_person: [756, 756, 1_512]  # Parents get reduced fare, toddler gets free ride (higher benefit)

policyengine_us/tests/test_batched.py

@@ -47,7 +47,8 @@ def split_into_batches(base_path: Path, num_batches: int) -> List[List[str]]:
     Special handling for contrib tests to divide by folder count.
     """
     # Special handling for contrib tests - each folder is its own batch
-    if "contrib" in str(base_path):
+    # Only apply to policy/contrib (structural tests), not baseline/contrib
+    if str(base_path).endswith("policy/contrib"):
         # Get all subdirectories and sort them alphabetically
         subdirs = sorted(
             [item for item in base_path.iterdir() if item.is_dir()]
@@ -79,17 +80,20 @@ def split_into_batches(base_path: Path, num_batches: int) -> List[List[str]]:
             # Batch 1: Only states
             batch1 = [str(states_path)]
 
-            # Batch 2: Everything else (excluding states)
+            # Batch 2: Everything else (excluding states, household, and contrib)
             batch2 = []
 
             # Add root level files if any
             for yaml_file in base_path.glob("*.yaml"):
                 batch2.append(str(yaml_file))
 
-            # Add all directories except gov/states
+            # Add all directories except gov/states, household, and contrib
             for item in base_path.iterdir():
                 if item.is_dir():
-                    if item.name == "gov":
+                    # Skip household and contrib directories (they'll be run separately)
+                    if item.name in ["household", "contrib"]:
+                        continue
+                    elif item.name == "gov":
                         # Add gov subdirectories except states
                         for gov_item in item.iterdir():
                             if gov_item.is_dir() and gov_item.name != "states":
@@ -97,7 +101,7 @@ def split_into_batches(base_path: Path, num_batches: int) -> List[List[str]]:
                             elif gov_item.suffix == ".yaml":
                                 batch2.append(str(gov_item))
                     else:
-                        # Non-gov directories
+                        # Other non-gov directories
                         batch2.append(str(item))
 
             return [batch1, batch2] if batch2 else [batch1]

policyengine_us/variables/gov/states/mt/tax/income/base/{mt_agi.py → mt_agi_indiv.py}

@@ -1,7 +1,7 @@
 from policyengine_us.model_api import *
 
 
-class mt_agi(Variable):
+class mt_agi_indiv(Variable):
     value_type = float
     entity = Person
     label = "Montana Adjusted Gross Income for each individual"