format

PolicyEngine · Dec 14, 2024 · f621a61 · f621a61
1 parent f4c3f28
commit f621a61
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/policyengine_us/tests/policy/baseline/gov/states/or/tax/income/or_liheap_eligibility.yaml b/policyengine_us/tests/policy/baseline/gov/states/or/tax/income/or_liheap_eligibility.yaml
@@ -2,7 +2,7 @@
   period: 2021
   input:
     spm_unit_size: 1
-    state_code_str: OR
+    state_code: OR
     adjusted_gross_income: 25_000
   output:
     or_liheap_eligibility: true
@@ -11,16 +11,16 @@
   period: 2023
   input:
     spm_unit_size: 2
-    state_code_str: OR
-    adjusted_gross_income: 50_000
+    state_code: OR
+    adjusted_gross_income: 90_000
   output:
     or_liheap_eligibility: false
 
 - name: oregon liheap eligibility 2024 eligible four-person household
   period: 2024
   input:
     spm_unit_size: 4
-    state_code_str: OR
+    state_code: OR
     adjusted_gross_income: 60_000
   output:
     or_liheap_eligibility: true
diff --git a/...yengine_us/tests/policy/baseline/gov/states/or/tax/income/or_liheap_income_threshold.yaml b/...yengine_us/tests/policy/baseline/gov/states/or/tax/income/or_liheap_income_threshold.yaml
@@ -3,7 +3,7 @@
   absolute_error_margin: 0.01
   input:
     spm_unit_size: 1  # Single-person household
-    state_code_str: OR  # Oregon state code
+    state_code: OR  # Oregon state code
     hhs_smi: 48_906  # State Median Income for a single-person household in Oregon for 2021
   output:
     or_liheap_income_threshold: 29_343.6  # 0.6 * $48_906
@@ -12,7 +12,7 @@
   period: 2023
   input:
     spm_unit_size: 2  # Two-person household
-    state_code_str: OR  # Oregon state code
+    state_code: OR  # Oregon state code
     hhs_smi: 72_852.48  # State Median Income for Oregon in 2023
   output:
     or_liheap_income_threshold: 43_711.488  # 0.6 * $72_852.48
@@ -22,7 +22,7 @@
   absolute_error_margin: 0.01
   input:
     spm_unit_size: 4  # Four-person household
-    state_code_str: OR  # Oregon state code
+    state_code: OR  # Oregon state code
     hhs_smi: 117_983  # State Median Income for Oregon in 2024
   output:
     or_liheap_income_threshold: 70_789.8  # 0.6 * $117_983
diff --git a/policyengine_us/variables/gov/states/or/tax/income/liheap/or_liheap_eligibility.py b/policyengine_us/variables/gov/states/or/tax/income/liheap/or_liheap_eligibility.py
@@ -9,6 +9,6 @@ class or_liheap_eligibility(Variable):
         defined_for = StateCode.OR
 
         def formula(spm_unit, period, parameters):
-            income = tax_unit.spm_unit("adjusted_gross_income", period)  
+            income = spm_unit.tax_unit("adjusted_gross_income", period)  
             threshold = spm_unit("or_liheap_income_threshold", period)
             return income <= threshold