Fix Kansas TANF implementation issues from review

- Fix broken KEESM reference URLs (case sensitivity: EES/KEESM/Current)
- Optimize code pattern: combine add() calls in ks_tanf_gross_income_eligible.py
- Add integration tests: multiple earners (Case 7), large family of 5 (Case 8)
- Add boundary tests: work expense at $89 and $91 thresholds
- Apply black formatting

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: hua7450

policyengine_us/parameters/gov/states/ks/dcf/tanf/resource_limit/amount.yaml

@@ -8,7 +8,7 @@ metadata:
   period: month
   label: Kansas TANF resource limit
   reference:
+    - title: K.A.R. 30-4-105 - Resources
+      href: https://www.law.cornell.edu/regulations/kansas/K-A-R-30-4-105
     - title: KEESM 5110 - Resource Limitation
-      href: https://content.dcf.ks.gov/ees/keesm/current/keesm5110.htm
-    - title: KEESM 5000 - Resources
-      href: https://content.dcf.ks.gov/ees/keesm/current/keesm5000.htm
+      href: https://content.dcf.ks.gov/EES/KEESM/Current/keesm5110.htm

policyengine_us/parameters/gov/states/ks/dcf/tanf/work_expense_deduction/amount.yaml

@@ -8,7 +8,7 @@ metadata:
   period: month
   label: Kansas TANF work expense deduction
   reference:
-    - title: KEESM 8151 - Deduction From Earned Income
-      href: https://content.dcf.ks.gov/ees/keesm/robo10-17/keesm8151.htm
     - title: K.A.R. 30-4-110 - Income
       href: https://www.law.cornell.edu/regulations/kansas/K-A-R-30-4-110
+    - title: KEESM 8151 - Deduction From Earned Income
+      href: https://content.dcf.ks.gov/EES/KEESM/Current/keesm8151.htm

policyengine_us/tests/policy/baseline/gov/states/ks/dcf/tanf/integration.yaml

@@ -228,3 +228,98 @@
     ks_tanf_eligible: true
     # Benefit: 386 - 124 = 262
     ks_tanf: 262
+
+- name: Case 7, household with two earners.
+  period: 2025-01
+  input:
+    people:
+      person1:
+        age: 30
+        employment_income_before_lsr: 3_600  # $300/month
+      person2:
+        age: 28
+        employment_income_before_lsr: 2_400  # $200/month
+      person3:
+        age: 5
+        is_tax_unit_dependent: true
+    spm_units:
+      spm_unit:
+        members: [person1, person2, person3]
+        spm_unit_cash_assets: 800
+    tax_units:
+      tax_unit:
+        members: [person1, person2, person3]
+    households:
+      household:
+        members: [person1, person2, person3]
+        state_code: KS
+  output:
+    # Two earners - income is summed at SPM unit level
+    # Person1 gross: $300/month, Person2 gross: $200/month
+    # Combined gross earned: $500/month
+    # Work expense: $90 (applied once to combined total)
+    # Remainder: $500 - $90 = $410
+    # 60% disregard: $410 * 0.60 = $246
+    # Total deductions: $90 + $246 = $336
+    # Countable earned: $500 - $336 = $164
+    spm_unit_size: 3
+    ks_tanf_earned_income_deductions: 336
+    ks_tanf_countable_income: 164
+    ks_tanf_maximum_benefit: 386
+    # Gross ($500) < 30% FPL for family of 3 (~$670)
+    ks_tanf_gross_income_eligible: true
+    ks_tanf_income_eligible: true
+    ks_tanf_resources_eligible: true
+    ks_tanf_eligible: true
+    # Benefit: 386 - 164 = 222
+    ks_tanf: 222
+
+- name: Case 8, large family of 5.
+  period: 2025-01
+  input:
+    people:
+      adult1:
+        age: 35
+        employment_income_before_lsr: 7_200  # $600/month
+      adult2:
+        age: 32
+      child1:
+        age: 12
+        is_tax_unit_dependent: true
+      child2:
+        age: 8
+        is_tax_unit_dependent: true
+      child3:
+        age: 4
+        is_tax_unit_dependent: true
+    spm_units:
+      spm_unit:
+        members: [adult1, adult2, child1, child2, child3]
+        spm_unit_cash_assets: 1_500
+    tax_units:
+      tax_unit:
+        members: [adult1, adult2, child1, child2, child3]
+    households:
+      household:
+        members: [adult1, adult2, child1, child2, child3]
+        state_code: KS
+  output:
+    # Large family tests the additional-person formula for 5+ members
+    # Payment standard for 5: $454 + $61 = $515
+    # Gross earned: $7,200/year = $600/month
+    # Work expense: $90
+    # Remainder: $600 - $90 = $510
+    # 60% disregard: $510 * 0.60 = $306
+    # Total deductions: $90 + $306 = $396
+    # Countable: $600 - $396 = $204
+    spm_unit_size: 5
+    ks_tanf_earned_income_deductions: 396
+    ks_tanf_countable_income: 204
+    ks_tanf_maximum_benefit: 515
+    # Gross ($600) < 30% FPL for family of 5 (~$890)
+    ks_tanf_gross_income_eligible: true
+    ks_tanf_income_eligible: true
+    ks_tanf_resources_eligible: true
+    ks_tanf_eligible: true
+    # Benefit: 515 - 204 = 311
+    ks_tanf: 311

policyengine_us/tests/policy/baseline/gov/states/ks/dcf/tanf/ks_tanf_earned_income_deductions.yaml

@@ -69,3 +69,27 @@
     # 60% disregard: 910 * 0.60 = 546
     # Total: 90 + 546 = 636
     ks_tanf_earned_income_deductions: 636
+
+- name: Case 7, earned income at $89 (just below $90 threshold).
+  period: 2025-01
+  input:
+    state_code: KS
+    tanf_gross_earned_income: 89
+  output:
+    # Work expense capped at gross: min(89, 90) = 89
+    # Remainder: max(89 - 89, 0) = 0
+    # 60% disregard: 0 * 0.60 = 0
+    # Total: 89 + 0 = 89
+    ks_tanf_earned_income_deductions: 89
+
+- name: Case 8, earned income at $91 (just above $90 threshold).
+  period: 2025-01
+  input:
+    state_code: KS
+    tanf_gross_earned_income: 91
+  output:
+    # Work expense: min(91, 90) = 90
+    # Remainder: max(91 - 90, 0) = 1
+    # 60% disregard: 1 * 0.60 = 0.60
+    # Total: 90 + 0.60 = 90.60
+    ks_tanf_earned_income_deductions: 90.6

policyengine_us/variables/gov/states/ks/dcf/tanf/ks_tanf_eligible.py

@@ -4,7 +4,9 @@
 class ks_tanf_eligible(Variable):
     value_type = bool
     entity = SPMUnit
-    label = "Eligible for Kansas Temporary Assistance for Needy Families (TANF)"
+    label = (
+        "Eligible for Kansas Temporary Assistance for Needy Families (TANF)"
+    )
     definition_period = MONTH
     reference = (
         "https://www.law.cornell.edu/regulations/kansas/K-A-R-30-4-70",
@@ -19,7 +21,9 @@ def formula(spm_unit, period, parameters):
         has_eligible_member = (
             add(spm_unit, period, ["is_person_demographic_tanf_eligible"]) > 0
         )
-        gross_income_eligible = spm_unit("ks_tanf_gross_income_eligible", period)
+        gross_income_eligible = spm_unit(
+            "ks_tanf_gross_income_eligible", period
+        )
         net_income_eligible = spm_unit("ks_tanf_income_eligible", period)
         resources_eligible = spm_unit("ks_tanf_resources_eligible", period)
         return (

policyengine_us/variables/gov/states/ks/dcf/tanf/ks_tanf_gross_income_eligible.py

@@ -15,9 +15,11 @@ class ks_tanf_gross_income_eligible(Variable):
     def formula(spm_unit, period, parameters):
         # Per K.S.A. 39-709 and Kansas TANF State Plan:
         # Gross income must be less than 30% of Federal Poverty Level
-        gross_earned = add(spm_unit, period, ["tanf_gross_earned_income"])
-        gross_unearned = add(spm_unit, period, ["tanf_gross_unearned_income"])
-        gross_income = gross_earned + gross_unearned
+        gross_income = add(
+            spm_unit,
+            period,
+            ["tanf_gross_earned_income", "tanf_gross_unearned_income"],
+        )
         # Get FPL for family size
         family_size = spm_unit("spm_unit_size", period.this_year)
         state_group = spm_unit.household("state_group_str", period.this_year)