Add SSA disability screen input

Author: MaxGhenis

changelog.d/added/ssa-disability-screen.md

@@ -0,0 +1 @@
+Added an SSA disability-screen variable for data-backed SSI disability modeling.

policyengine_us/tests/policy/baseline/gov/ssa/ss/would_pass_ssa_disability_screen.yaml

@@ -0,0 +1,13 @@
+- name: Broad disability input remains the household fallback
+  period: 2024
+  input:
+    is_disabled: true
+  output:
+    would_pass_ssa_disability_screen: true
+
+- name: No disability signal defaults to not passing screen
+  period: 2024
+  input:
+    is_disabled: false
+  output:
+    would_pass_ssa_disability_screen: false

policyengine_us/tests/policy/baseline/gov/ssa/ssi/is_ssi_disabled.yaml

@@ -68,6 +68,24 @@
   output:
     is_ssi_disabled: false
 
+- name: Data-provided SSA disability screen can identify SSI disability
+  period: 2024
+  input:
+    would_pass_ssa_disability_screen: true
+    is_disabled: false
+    ssi_engaged_in_sga: false
+  output:
+    is_ssi_disabled: true
+
+- name: Data-provided SSA disability screen remains subject to SGA
+  period: 2024
+  input:
+    would_pass_ssa_disability_screen: true
+    is_disabled: false
+    ssi_engaged_in_sga: true
+  output:
+    is_ssi_disabled: false
+
 - name: Data-provided SSI disability criteria can exclude broad disability
   period: 2024
   input:

policyengine_us/tests/policy/baseline/gov/ssa/ssi/meets_ssi_disability_criteria.yaml

@@ -21,3 +21,19 @@
     ssi_engaged_in_sga: false
   output:
     meets_ssi_disability_criteria: false
+
+- name: Data-provided SSA disability screen can identify SSI disability criteria
+  period: 2024
+  input:
+    would_pass_ssa_disability_screen: true
+    is_disabled: false
+  output:
+    meets_ssi_disability_criteria: true
+
+- name: Data-provided SSA disability screen can exclude broad disability
+  period: 2024
+  input:
+    would_pass_ssa_disability_screen: false
+    is_disabled: true
+  output:
+    meets_ssi_disability_criteria: false

policyengine_us/variables/gov/ssa/ss/would_pass_ssa_disability_screen.py

@@ -0,0 +1,15 @@
+from policyengine_us.model_api import *
+
+
+class would_pass_ssa_disability_screen(Variable):
+    value_type = bool
+    entity = Person
+    label = "Would pass SSA disability screen"
+    definition_period = YEAR
+    reference = [
+        "https://www.law.cornell.edu/uscode/text/42/423#d_1_A",
+        "https://www.law.cornell.edu/uscode/text/42/1382c#a_3_A",
+    ]
+
+    def formula(person, period, parameters):
+        return person("is_disabled", period)

policyengine_us/variables/gov/ssa/ssi/eligibility/status/meets_ssi_disability_criteria.py

@@ -13,4 +13,4 @@ class meets_ssi_disability_criteria(Variable):
     reference = "https://www.law.cornell.edu/uscode/text/42/1382c#a_3_A"
 
     def formula(person, period, parameters):
-        return person("is_disabled", period)
+        return person("would_pass_ssa_disability_screen", period)