policyengine-us 1.381.3__py3-none-any.whl → 1.383.0__py3-none-any.whl

policyengine_us/parameters/gov/states/id/tax/income/credits/2022_rebate/floor.yaml

@@ -0,0 +1,10 @@
+description: Idaho provides the following minimum 2022 rebate amount for each filer and dependent.
+values:
+  2022-01-01: 75
+metadata:
+  unit: currency-USD
+  period: year
+  label: Idaho 2022 rebate floor
+  reference:
+    - title: Idaho Statutes 63-3024B (3)
+      href: https://legislature.idaho.gov/statutesrules/idstat/title63/t63ch30/sect63-3024b/

policyengine_us/parameters/gov/states/id/tax/income/credits/2022_rebate/rate.yaml

@@ -0,0 +1,10 @@
+description: Idaho provides a 2022 rebate amount of this percentage of the state income tax liability.
+values:
+  2022-01-01: 0.12
+metadata:
+  unit: currency-USD
+  period: year
+  label: Idaho 2022 rebate floor
+  reference:
+    - title: Idaho Statutes 63-3024B (3)
+      href: https://legislature.idaho.gov/statutesrules/idstat/title63/t63ch30/sect63-3024b/

policyengine_us/parameters/gov/states/id/tax/income/credits/refundable.yaml

@@ -12,7 +12,14 @@ metadata:
 
 values:
   2002-01-01:
-    - id_aged_or_disabled_credit # Idaho aged or disabled credit
+    - id_aged_or_disabled_credit
   2021-01-01:
-    - id_grocery_credit  # Idaho grocery credit
-    - id_aged_or_disabled_credit # Idaho aged or disabled credit
+    - id_grocery_credit  
+    - id_aged_or_disabled_credit
+  2022-01-01:
+    - id_grocery_credit
+    - id_aged_or_disabled_credit
+    - id_2022_rebate
+  2023-01-01:
+    - id_grocery_credit
+    - id_aged_or_disabled_credit

policyengine_us/parameters/gov/states/id/tax/income/credits/special_seasonal_rebate/floor.yaml

@@ -0,0 +1,22 @@
+description: Idaho provides the following minimum special seasonal rebate amount, based on filing status.
+
+metadata:
+  unit: currency-USD
+  period: year
+  breakdown:
+    - filing_status
+  label: Idaho special seasonal rebate floor
+  reference:
+    - title: Idaho Statutes 63-3024B (4)
+      href: https://legislature.idaho.gov/statutesrules/idstat/title63/t63ch30/sect63-3024b/
+
+SINGLE:
+  2022-01-01: 300
+JOINT:
+  2022-01-01: 600
+SEPARATE:
+  2022-01-01: 300
+SURVIVING_SPOUSE:
+  2022-01-01: 300
+HEAD_OF_HOUSEHOLD:
+  2022-01-01: 300

policyengine_us/parameters/gov/states/id/tax/income/credits/special_seasonal_rebate/rate.yaml

@@ -0,0 +1,10 @@
+description: Idaho provides a special seasonal rebate amount of this percentage of the state income tax liability.
+values:
+  2022-01-01: 0.1
+metadata:
+  unit: currency-USD
+  period: year
+  label: Idaho special seasonal rebate rate
+  reference:
+    - title: Idaho Statutes 63-3024B (4)
+      href: https://legislature.idaho.gov/statutesrules/idstat/title63/t63ch30/sect63-3024b/

policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_2022_rebate.yaml

@@ -0,0 +1,17 @@
+- name: Floor over rate
+  period: 2022
+  input:
+    id_income_tax_before_non_refundable_credits: 1_000
+    state_code: ID
+    id_2022_rebate_floor: 300
+  output:
+    id_2022_rebate: 300
+
+- name: Rate over floor
+  period: 2022
+  input:
+    id_income_tax_before_non_refundable_credits: 1_000
+    state_code: ID
+    id_2022_rebate_floor: 0
+  output:
+    id_2022_rebate: 120

policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_2022_rebate_floor.yaml

@@ -0,0 +1,21 @@
+- name: One person in ID
+  period: 2022
+  input:
+    state_code: ID
+  output:
+    id_2022_rebate_floor: 75
+
+- name: Two people
+  period: 2022
+  input:
+    people: 
+      person1:
+        age: 40
+      person2:
+        age: 40
+    households:
+      household:
+        members: [person1, person2]
+        state_code: ID
+  output:
+    id_2022_rebate_floor: [75, 75]

policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_special_season_rebate.yaml

@@ -0,0 +1,17 @@
+- name: Floor over rate
+  period: 2022
+  input:
+    id_income_tax_before_non_refundable_credits: 1_000
+    state_code: ID
+    filing_status: SINGLE
+  output:
+    id_special_season_rebate: 300
+
+- name: Rate over floor
+  period: 2022
+  input:
+    id_income_tax_before_non_refundable_credits: 10_000
+    state_code: ID
+    filing_status: SINGLE
+  output:
+    id_special_season_rebate: 1_000

policyengine_us/variables/gov/states/id/tax/income/credits/2022_rebate/id_2022_rebate.py

@@ -0,0 +1,18 @@
+from policyengine_us.model_api import *
+
+
+class id_2022_rebate(Variable):
+    value_type = float
+    entity = TaxUnit
+    label = "Idaho 2022 rebate"
+    definition_period = YEAR
+    defined_for = StateCode.ID
+
+    def formula(tax_unit, period, parameters):
+        p = parameters(period).gov.states.id.tax.income.credits["2022_rebate"]
+        income_tax_before_credits = tax_unit(
+            "id_income_tax_before_non_refundable_credits", period
+        )
+        percentage_credit = p.rate * income_tax_before_credits
+        floor = add(tax_unit, period, ["id_2022_rebate_floor"])
+        return max_(percentage_credit, floor)

policyengine_us/variables/gov/states/id/tax/income/credits/2022_rebate/id_2022_rebate_floor.py

@@ -0,0 +1,11 @@
+from policyengine_us.model_api import *
+
+
+class id_2022_rebate_floor(Variable):
+    value_type = float
+    entity = Person
+    label = "Idaho 2022 rebate floor"
+    definition_period = YEAR
+    defined_for = StateCode.ID
+
+    adds = ["gov.states.id.tax.income.credits.2022_rebate.floor"]

policyengine_us/variables/gov/states/id/tax/income/credits/id_special_season_rebate.py

@@ -0,0 +1,21 @@
+from policyengine_us.model_api import *
+
+
+class id_special_season_rebate(Variable):
+    value_type = float
+    entity = TaxUnit
+    label = "Idaho special seasonal rebate"
+    definition_period = YEAR
+    defined_for = StateCode.ID
+
+    def formula(tax_unit, period, parameters):
+        p = parameters(
+            period
+        ).gov.states.id.tax.income.credits.special_seasonal_rebate
+        income_tax_before_credits = tax_unit(
+            "id_income_tax_before_non_refundable_credits", period
+        )
+        percentage_credit = p.rate * income_tax_before_credits
+        filing_status = tax_unit("filing_status", period)
+        floor = p.floor[filing_status]
+        return max_(percentage_credit, floor)

policyengine_us/variables/input/geography.py

@@ -142,3 +142,27 @@ class state_fips(Variable):
     definition_period = YEAR
     documentation = "State FIPS code"
     default_value = 6
+
+
+class congressional_district_geoid(Variable):
+    value_type = int
+    label = "Congressional district GEOID"
+    entity = Household
+    definition_period = YEAR
+    documentation = """Congressional district geographic identifier stored as an integer.
+    
+    Format: SSDD where SS is the 2-digit state FIPS code and DD is the 2-digit district number.
+    
+    Examples:
+    - Alabama (FIPS 01) district 01 is stored as 101
+    - Alabama (FIPS 01) district 07 is stored as 107  
+    - North Carolina (FIPS 37) district 01 is stored as 3701
+    - California (FIPS 06) district 52 is stored as 652
+    
+    To extract components:
+    - State FIPS: geoid // 100
+    - District number: geoid % 100
+    
+    Note: Leading zeros are not preserved in the integer storage (e.g., 0101 becomes 101),
+    but the value remains unique and unambiguous since the format is always interpreted as SSDD."""
+    default_value = 0

{policyengine_us-1.381.3.dist-info → policyengine_us-1.383.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: policyengine-us
-Version: 1.381.3
+Version: 1.383.0
 Summary: Add your description here.
 Author-email: PolicyEngine <hello@policyengine.org>
 License-File: LICENSE

{policyengine_us-1.381.3.dist-info → policyengine_us-1.383.0.dist-info}/RECORD

@@ -1531,7 +1531,9 @@ policyengine_us/parameters/gov/states/ia/tax/income/taxable_income/additions.yam
 policyengine_us/parameters/gov/states/ia/tax/income/taxable_income/subtractions.yaml,sha256=xvzYg1pZwnbhPCJnCkpe0EQ1uM_BQqPciyJzD29oizE,1311
 policyengine_us/parameters/gov/states/id/README.md,sha256=kvtjNiUwpLs4_9k916xh9jLgIzvTylItPpMJLmJZREY,7
 policyengine_us/parameters/gov/states/id/tax/income/credits/non_refundable.yaml,sha256=DFvGtl8YZyzYPxNmbFr_rhuG8Ts59YJ55sePKS6HJxg,641
-policyengine_us/parameters/gov/states/id/tax/income/credits/refundable.yaml,sha256=-V1eiVSjS3cRglMuUvQKH8RLfvy_2GkBKelcxK9Q71U,907
+policyengine_us/parameters/gov/states/id/tax/income/credits/refundable.yaml,sha256=mfvsOxpHPh2EIyTBX9_61SVA90ZRk9HkA9oa4Vu7yA4,984
+policyengine_us/parameters/gov/states/id/tax/income/credits/2022_rebate/floor.yaml,sha256=G0o8nti3DsIBGskQGaYutFXQTh08-2_lv32W9G_BeEs,350
+policyengine_us/parameters/gov/states/id/tax/income/credits/2022_rebate/rate.yaml,sha256=mUlgcUQUDEIuXoTBUtc4lOp4RPGat6zQwaJSOoIadqA,356
 policyengine_us/parameters/gov/states/id/tax/income/credits/aged_or_disabled/age_threshold.yaml,sha256=WKPHZLm1j4umAXhJOHzWEEMag94u4Auhnk5QYWwk22Y,1269
 policyengine_us/parameters/gov/states/id/tax/income/credits/aged_or_disabled/amount.yaml,sha256=vDZ9mCNywpPArOj1npful07Ldh9D0D8Wg9cKKaBAGI8,1282
 policyengine_us/parameters/gov/states/id/tax/income/credits/aged_or_disabled/person_cap.yaml,sha256=aVT8BHsHPy6OBUQ1e6Cc_dERBw9-ZW7ATC63-pFefKo,1290
@@ -1542,6 +1544,8 @@ policyengine_us/parameters/gov/states/id/tax/income/credits/grocery/aged/age_thr
 policyengine_us/parameters/gov/states/id/tax/income/credits/grocery/aged/amount.yaml,sha256=J_NorjR2OQrMx5MRtlCFqtUDKlkOEEt2eB9VZ1wMLfA,1470
 policyengine_us/parameters/gov/states/id/tax/income/credits/grocery/aged/in_effect.yaml,sha256=LX-UMYdBk-PUxAve5ClqOwvvsHakJn7tHxIv7oObCJM,1504
 policyengine_us/parameters/gov/states/id/tax/income/credits/grocery/base/amount.yaml,sha256=2JDsT4NJqI3dvq4DAeo2D1fa_7X80Je54_TZebaac4c,1574
+policyengine_us/parameters/gov/states/id/tax/income/credits/special_seasonal_rebate/floor.yaml,sha256=yH00DE9jZgXxLHeaHPEmLLN9Jms0eKejz9yWwt_UVEo,531
+policyengine_us/parameters/gov/states/id/tax/income/credits/special_seasonal_rebate/rate.yaml,sha256=TsdFTB2Ye3_EkgOFXRtnvEN9JbSjgdxDyKbC_xLyIuo,378
 policyengine_us/parameters/gov/states/id/tax/income/deductions/capital_gains/percentage.yaml,sha256=ycrvMenEutCPG-4iVRRjcvG0q2uZRS3a3TDHJlbk1qM,1021
 policyengine_us/parameters/gov/states/id/tax/income/deductions/dependent_care_expenses/cap.yaml,sha256=NEoYQo8rglFbkVmKjRK93B6f2O-z_Cn-jqb9MW_5cuc,1053
 policyengine_us/parameters/gov/states/id/tax/income/deductions/retirement_benefits/cap.yaml,sha256=a9c3vtiGMnIfRLJyoBmUUcP_gEzBL-abEzJE-g36MhI,2043
@@ -4010,9 +4014,12 @@ policyengine_us/tests/policy/baseline/gov/states/id/tax/income/id_salt_deduction
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/id_taxable_income.yaml,sha256=CzTxtnk9qVtJoY30V2jVOk3O40KDtTHSnbVSnDpN050,579
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/id_withheld_income_tax.yaml,sha256=hZGxW9CVVHUfVnlrpQZpix5LYkgRu3hh8zk0eX1C1Do,720
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/integration.yaml,sha256=FjFVklhKYM_3vc9dvWETfRaDE-hBxdsWMKQsD6MGFqc,4766
+policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_2022_rebate.yaml,sha256=9d8oNWSDIYMsp-uDCOplCU9JJcfFCS1922L8qzM05pA,371
+policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_2022_rebate_floor.yaml,sha256=DcNyo7Wrhimm-ojCC0RiT3XtAc9WUDOdglx6YahPfBs,363
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_aged_or_disabled_credit.yaml,sha256=ypo7VOLXqf5UNF9nxTXD32M5Ow6GHg__cn7lqv8sWsI,2108
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_aged_or_disabled_credit_eligible_person.yaml,sha256=_K8Ku_C5eM1nfS99xLtxj5o5CHut8XhqMuK76mgXcx4,6195
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_ctc.yaml,sha256=k9dtZYJCiDXIbWQqf6dWhxgdIINQWvu4XL_DFnyMm48,370
+policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/id_special_season_rebate.yaml,sha256=QaUYpsFKzrrVo9appi0UOfmTKc3sE7xxu3nE1Yjdj2M,388
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/grocery/id_grocery_credit.yaml,sha256=vXX3ybmcX0idJicBBjwfsc8EK3mLXQQrMj1xPa2IYiE,2335
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/grocery/id_grocery_credit_aged.yaml,sha256=DQBH7vI1lfl8W_IPy4WSbT84A4ngTqpnLr3QHKsoe4A,1810
 policyengine_us/tests/policy/baseline/gov/states/id/tax/income/credits/grocery/id_grocery_credit_base.yaml,sha256=7mZ2mI7yXQsit2ZcVghgZAxsPlLL0E4RQj0pQrBjI0w,447
@@ -5179,7 +5186,6 @@ policyengine_us/tests/policy/contrib/ubi_center/flat_tax.yaml,sha256=O6krfJ2rovW
 policyengine_us/tests/policy/reform/ctc_expansion.yaml,sha256=soq-LqWwbb7fNQJFphx_1gSc8uDGOhtr-5P93oY2Mpg,1596
 policyengine_us/tests/policy/reform/winship.yaml,sha256=OPfsoZ0NrlPU25hJadZXycrYLaM0wPUGXjBrn6_8gR4,1706
 policyengine_us/tests/utilities/test_load_county_fips_dataset.py,sha256=QAURDL6IMN5gafUOcTri1PUet6khLNZ_Bov6NdW_pMM,4397
-policyengine_us/tests/utilities/test_ucgid_hierarchical.py,sha256=WZYqWHw6-QQAv0Wd3bGgAcgPgZ0ILhOxpC-kgo4HRkU,4532
 policyengine_us/tests/variables/gov/states/vt/tax/income/vt_eitc.yaml,sha256=gE-ZtclGqoFKs87k9VUgKEoVO2iZkFBq4k-ixGtde14,2341
 policyengine_us/tests/variables/gov/states/vt/tax/income/adjusted_gross_income/subtractions/retirement_income_exemption/vt_military_retirement_income_based_exemption.yaml,sha256=b5I5ekAoDCrojCLwEZMoBEDdUX9zDw1bM-4iQLmyTng,3267
 policyengine_us/tests/variables/gov/states/vt/tax/income/credits/vt_veteran_tax_credit.yaml,sha256=51NtThz2bNwVD7o-2VXZqplDjKUAYzIiIzyL1PfXK94,3244
@@ -6432,6 +6438,9 @@ policyengine_us/variables/gov/states/id/tax/income/id_withheld_income_tax.py,sha
 policyengine_us/variables/gov/states/id/tax/income/credits/id_aged_or_disabled_credit.py,sha256=PJHgc8u8-l38_L15zPIRPkdCeMkqLlSpbBPYPaQuNhs,1128
 policyengine_us/variables/gov/states/id/tax/income/credits/id_aged_or_disabled_credit_eligible_person.py,sha256=R7Cjs8LqLatmO3GzCNnLzUNKbfDlrIvT_2v3v4kKy4w,1253
 policyengine_us/variables/gov/states/id/tax/income/credits/id_ctc.py,sha256=f7c3cnPzeUDddGq04y9bePYGMEO5LgGf7X68u_vLQAY,723
+policyengine_us/variables/gov/states/id/tax/income/credits/id_special_season_rebate.py,sha256=NSG10Fs-Q353bxUjz7Xkjlcwp46NLci8ShBBgYIqRGI,718
+policyengine_us/variables/gov/states/id/tax/income/credits/2022_rebate/id_2022_rebate.py,sha256=YqeOhvCLtYkqsSORenn-jUvomeGshB7ooPhmjq0hXzc,632
+policyengine_us/variables/gov/states/id/tax/income/credits/2022_rebate/id_2022_rebate_floor.py,sha256=2eiEbuiXBKTSccvwRYeoRWXijsMtosEnND0yHN8QkME,288
 policyengine_us/variables/gov/states/id/tax/income/credits/grocery/id_grocery_credit.py,sha256=UTAAx1T9gIFWPCjFXs2vhQ2rXlEOM0RnvkjXRzBQC9k,1064
 policyengine_us/variables/gov/states/id/tax/income/credits/grocery/id_grocery_credit_aged.py,sha256=Ounu9koUqG6guzJJpxHRPr3XSV5t6qjhduTJq_hbVLM,850
 policyengine_us/variables/gov/states/id/tax/income/credits/grocery/id_grocery_credit_base.py,sha256=0LIGHITHapIepN2WpJvnX8FDr9T-dm49SI6OwGacnVw,606
@@ -7845,9 +7854,6 @@ policyengine_us/variables/household/demographic/geographic/county/county_str.py,
 policyengine_us/variables/household/demographic/geographic/county/first_county_in_state.py,sha256=yXCj-LVjxPeILIg42ijefTmi_j4NxJwX0kMtoE47xic,1578
 policyengine_us/variables/household/demographic/geographic/state/average_home_energy_use_in_state.py,sha256=aaGYAXLU7tsx-A9NW4nyLij5J5mp0WTRXsyObngekvY,319
 policyengine_us/variables/household/demographic/geographic/state/in_state.py,sha256=o3ksZCOHtlJ1JGHIXAPlqcs5-rVbTgOCcDdN1-fYqmA,551
-policyengine_us/variables/household/demographic/geographic/ucgid/ucgid.py,sha256=SwONHVRvHmUFHvPZQMYY7Y9e7gz74kisLjJICpHdyKg,2874
-policyengine_us/variables/household/demographic/geographic/ucgid/ucgid_enum.py,sha256=dkeusTBI1j4xAGmRFkfSsyT3Hq5gHZwRUkSeISeNbx0,15636
-policyengine_us/variables/household/demographic/geographic/ucgid/ucgid_str.py,sha256=GnIRg2tc0376YazPX7wm7YQgMQaYPm2Au47vc1E5JcM,914
 policyengine_us/variables/household/demographic/geographic/zip_code/three_digit_zip_code.py,sha256=7q0FQRahTOINiK2vlAyHEa5xIKowM5Kgl60-kR3ZuWU,426
 policyengine_us/variables/household/demographic/geographic/zip_code/zip_code.py,sha256=3J8RfnIIsmw4_5gF5yZHmATt2cNskyKQR0lncCvOQx0,1691
 policyengine_us/variables/household/demographic/household/bedrooms.py,sha256=4ltdIRLdmeuTKAskn4pdgxZuxEE-4rz0ffPnIKobc-o,164
@@ -8295,11 +8301,11 @@ policyengine_us/variables/household/income/tax_unit/us_govt_interest.py,sha256=L
 policyengine_us/variables/input/employment_income.py,sha256=f85M5Jy8kHmkDId0j9lCE30R6htNQ-Hg74ko9cLupmI,505
 policyengine_us/variables/input/employment_income_before_lsr.py,sha256=X4PBfylDs40hw6jwHPL_B0JcBkqrKCFU-beQEDFoJTE,364
 policyengine_us/variables/input/farm_income.py,sha256=BEKxYmHNNnWJAAvULl5qZJigy58ap_bmQB6iHdXLGuo,427
-policyengine_us/variables/input/geography.py,sha256=XmBlgXhzBoLRKk6R8taVZHqUw1eU8MbNeGS9iJ7_l44,4506
+policyengine_us/variables/input/geography.py,sha256=Ux0ueAf0rhZaflyEqz81UuXP3xKCKBDvoO3CrKhiQEc,5421
 policyengine_us/variables/input/self_employment_income.py,sha256=PwsGz8R4lRikKWUYOhsC0qosNNLXq4f5SQmfw4S3mk8,511
 policyengine_us/variables/input/self_employment_income_before_lsr.py,sha256=E8fcX9Nlyqz8dziHhQv_euutdmoIwFMMWePUwbbwv_w,379
-policyengine_us-1.381.3.dist-info/METADATA,sha256=jsYKFk9M_6GcAwoo8Y13OnzIKFQ_Nwc9tjnuN2gqUyY,1649
-policyengine_us-1.381.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-policyengine_us-1.381.3.dist-info/entry_points.txt,sha256=MLaqNyNTbReALyKNkde85VkuFFpdPWAcy8VRG1mjczc,57
-policyengine_us-1.381.3.dist-info/licenses/LICENSE,sha256=2N5ReRelkdqkR9a-KP-y-shmcD5P62XoYiG-miLTAzo,34519
-policyengine_us-1.381.3.dist-info/RECORD,,
+policyengine_us-1.383.0.dist-info/METADATA,sha256=OHHi_sCwaJcw2LBrsw8bG5acWDrtPQB6Yn6DKyxNSec,1649
+policyengine_us-1.383.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+policyengine_us-1.383.0.dist-info/entry_points.txt,sha256=MLaqNyNTbReALyKNkde85VkuFFpdPWAcy8VRG1mjczc,57
+policyengine_us-1.383.0.dist-info/licenses/LICENSE,sha256=2N5ReRelkdqkR9a-KP-y-shmcD5P62XoYiG-miLTAzo,34519
+policyengine_us-1.383.0.dist-info/RECORD,,

policyengine_us/tests/utilities/test_ucgid_hierarchical.py

@@ -1,121 +0,0 @@
-"""
-Test script demonstrating UCGID hierarchical functionality.
-"""
-
-import numpy as np
-from policyengine_us import Microsimulation
-from policyengine_us.variables.household.demographic.geographic.ucgid.ucgid_enum import (
-    UCGID,
-)
-
-
-class TestUCGIDHierarchical:
-    """Test class for UCGID hierarchical functionality."""
-
-    def test_ucgid_enum_hierarchical_methods(self):
-        """Test the hierarchical methods on UCGID enum values directly."""
-        print("\n=== Testing UCGID Enum Hierarchical Methods ===")
-
-        # Test Congressional District (CA_23)
-        ca_23 = UCGID.CA_23
-        ca_23_codes = ca_23.get_hierarchical_codes()
-        assert ca_23_codes == ["5001800US0623", "0400000US06", "0100000US"]
-
-        # Test State (CA)
-        ca = UCGID.CA
-        ca_codes = ca.get_hierarchical_codes()
-        assert ca_codes == ["0400000US06", "0100000US"]
-
-        # Test US
-        us = UCGID.US
-        us_codes = us.get_hierarchical_codes()
-        assert us_codes == ["0100000US"]
-
-        # Test hierarchy matching
-        assert ca_23.matches_hierarchy("0400000US06") == True
-        assert ca_23.matches_hierarchy("0100000US") == True
-        assert ca.matches_hierarchy("5001800US0623") == False
-
-    def test_ucgid_simulation_usage(self):
-        """Test how UCGID is used in a PolicyEngine simulation."""
-        simulation = Microsimulation()
-        ucgid_values = simulation.calculate("ucgid", "2024")
-
-        # The UCGID value should be a string (enum name)
-        ucgid_string = ucgid_values.iloc[0]  # First household
-        assert type(ucgid_string) == str
-
-        # Convert the string back to enum to access hierarchical methods
-        ucgid_enum = UCGID[ucgid_string]
-        hierarchical_codes = ucgid_enum.get_hierarchical_codes()
-
-        # Should have 2 hierarchical codes for state-level UCGIDs: [state, US]
-        assert len(hierarchical_codes) == 2
-        assert (
-            hierarchical_codes[1] == "0100000US"
-        )  # Second code should always be US
-        assert hierarchical_codes[0].startswith(
-            "0400000US"
-        )  # First should be state-level
-
-        # Test hierarchy matching in simulation context
-        is_in_us = ucgid_enum.matches_hierarchy("0100000US")
-        assert is_in_us == True
-
-    def test_ucgid_str_variable(self):
-        """Test the UCGID string variable functionality."""
-        simulation = Microsimulation()
-        ucgid_str_values = simulation.calculate("ucgid_str", "2024")
-
-        # The UCGID string variable should return a string representation
-        ucgid_str = ucgid_str_values.iloc[0]
-        assert type(ucgid_str) == str
-        assert (
-            ucgid_str.startswith("0100000US")
-            | ucgid_str.startswith("0400000US")
-            | ucgid_str.startswith("5001800US")
-        )
-
-        # Create a basic simulation to test with specific input values
-        from policyengine_us import Simulation
-
-        simulation = Simulation(
-            situation={
-                "people": {"person": {}},
-                "households": {"household": {"members": ["person"]}},
-            }
-        )
-
-        # Set a specific UCGID value for testing (CA_23)
-        simulation.set_input("ucgid", 2024, UCGID.CA_23)
-
-        # Calculate the ucgid_str value
-        ucgid_str_values = simulation.calculate("ucgid_str", 2024)
-        ucgid_str = ucgid_str_values[0]
-
-        # Verify it contains all three hierarchical codes
-        assert ucgid_str == "5001800US0623,0400000US06,0100000US"
-
-    def test_ucgid_microsimulation_input_override(self):
-        """Test setting UCGID input for all households in a microsimulation."""
-        microsim = Microsimulation()
-
-        # Get initial values
-        ucgid_before = microsim.calculate("ucgid", 2024)
-        num_households = len(ucgid_before)
-
-        # Set all households to CA_23
-        input_array = np.array([UCGID.CA_23] * num_households)
-        microsim.set_input("ucgid", 2024, input_array)
-
-        # Verify the input was set
-        ucgid_after = microsim.calculate("ucgid", 2024)
-        after_values = ucgid_after.values
-        assert all(val == "CA_23" for val in after_values)
-        assert len(set(after_values)) == 1
-
-        # Test that ucgid_str now returns hierarchical codes for all households
-        ucgid_str_values = microsim.calculate("ucgid_str", 2024)
-        expected_str = "5001800US0623,0400000US06,0100000US"
-        str_values = ucgid_str_values.values
-        assert all(str_val == expected_str for str_val in str_values)

policyengine_us/variables/household/demographic/geographic/ucgid/ucgid.py

@@ -1,94 +0,0 @@
-from policyengine_us.model_api import *
-from policyengine_core.simulations import Simulation
-from policyengine_us.variables.household.demographic.geographic.ucgid.ucgid_enum import (
-    UCGID,
-)
-import numpy as np
-
-
-class ucgid(Variable):
-    value_type = Enum
-    possible_values = UCGID
-    default_value = UCGID.US
-    entity = Household
-    label = "Unified Congressional Geographic Identifier (UCGID)"
-    definition_period = YEAR
-    documentation = """
-    Unified Congressional Geographic Identifier (UCGID) for the household as defined by the U.S. Census Bureau.
-    """
-
-    def formula(household, period, parameters):
-        simulation: Simulation = household.simulation
-
-        # Check if this variable has been set as input
-        if simulation.get_holder("ucgid").get_known_periods():
-            return household("ucgid", period)
-
-        # Try to derive from state_code
-        state_code_enum = household("state_code", period)
-        state_code_strings = state_code_enum.decode_to_str()
-
-        # Map state codes to UCGID state values
-        state_mapping = {
-            "AL": UCGID.AL,
-            "AK": UCGID.AK,
-            "AZ": UCGID.AZ,
-            "AR": UCGID.AR,
-            "CA": UCGID.CA,
-            "CO": UCGID.CO,
-            "CT": UCGID.CT,
-            "DE": UCGID.DE,
-            "DC": UCGID.DC,
-            "FL": UCGID.FL,
-            "GA": UCGID.GA,
-            "HI": UCGID.HI,
-            "ID": UCGID.ID,
-            "IL": UCGID.IL,
-            "IN": UCGID.IN,
-            "IA": UCGID.IA,
-            "KS": UCGID.KS,
-            "KY": UCGID.KY,
-            "LA": UCGID.LA,
-            "ME": UCGID.ME,
-            "MD": UCGID.MD,
-            "MA": UCGID.MA,
-            "MI": UCGID.MI,
-            "MN": UCGID.MN,
-            "MS": UCGID.MS,
-            "MO": UCGID.MO,
-            "MT": UCGID.MT,
-            "NE": UCGID.NE,
-            "NV": UCGID.NV,
-            "NH": UCGID.NH,
-            "NJ": UCGID.NJ,
-            "NM": UCGID.NM,
-            "NY": UCGID.NY,
-            "NC": UCGID.NC,
-            "ND": UCGID.ND,
-            "OH": UCGID.OH,
-            "OK": UCGID.OK,
-            "OR": UCGID.OR,
-            "PA": UCGID.PA,
-            "RI": UCGID.RI,
-            "SC": UCGID.SC,
-            "SD": UCGID.SD,
-            "TN": UCGID.TN,
-            "TX": UCGID.TX,
-            "UT": UCGID.UT,
-            "VT": UCGID.VT,
-            "VA": UCGID.VA,
-            "WA": UCGID.WA,
-            "WV": UCGID.WV,
-            "WI": UCGID.WI,
-            "WY": UCGID.WY,
-        }
-
-        # Convert state code strings to UCGID values
-        result = np.empty(len(state_code_strings), dtype=object)
-        for i, state_str in enumerate(state_code_strings):
-            if state_str in state_mapping:
-                result[i] = state_mapping[state_str]
-            else:
-                result[i] = UCGID.US
-
-        return result

policyengine_us/variables/household/demographic/geographic/ucgid/ucgid_enum.py

@@ -1,551 +0,0 @@
-from policyengine_us.model_api import *
-from typing import List
-
-
-class UCGID(Enum):
-    US = "0100000US"
-    AL = "0400000US01"
-    AK = "0400000US02"
-    AZ = "0400000US04"
-    AR = "0400000US05"
-    CA = "0400000US06"
-    CO = "0400000US08"
-    CT = "0400000US09"
-    DE = "0400000US10"
-    DC = "0400000US11"
-    FL = "0400000US12"
-    GA = "0400000US13"
-    HI = "0400000US15"
-    ID = "0400000US16"
-    IL = "0400000US17"
-    IN = "0400000US18"
-    IA = "0400000US19"
-    KS = "0400000US20"
-    KY = "0400000US21"
-    LA = "0400000US22"
-    ME = "0400000US23"
-    MD = "0400000US24"
-    MA = "0400000US25"
-    MI = "0400000US26"
-    MN = "0400000US27"
-    MS = "0400000US28"
-    MO = "0400000US29"
-    MT = "0400000US30"
-    NE = "0400000US31"
-    NV = "0400000US32"
-    NH = "0400000US33"
-    NJ = "0400000US34"
-    NM = "0400000US35"
-    NY = "0400000US36"
-    NC = "0400000US37"
-    ND = "0400000US38"
-    OH = "0400000US39"
-    OK = "0400000US40"
-    OR = "0400000US41"
-    PA = "0400000US42"
-    RI = "0400000US44"
-    SC = "0400000US45"
-    SD = "0400000US46"
-    TN = "0400000US47"
-    TX = "0400000US48"
-    UT = "0400000US49"
-    VT = "0400000US50"
-    VA = "0400000US51"
-    WA = "0400000US53"
-    WV = "0400000US54"
-    WI = "0400000US55"
-    WY = "0400000US56"
-    AL_01 = "5001800US0101"
-    AL_02 = "5001800US0102"
-    AL_03 = "5001800US0103"
-    AL_04 = "5001800US0104"
-    AL_05 = "5001800US0105"
-    AL_06 = "5001800US0106"
-    AL_07 = "5001800US0107"
-    AK_01 = "5001800US0200"
-    AZ_01 = "5001800US0401"
-    AZ_02 = "5001800US0402"
-    AZ_03 = "5001800US0403"
-    AZ_04 = "5001800US0404"
-    AZ_05 = "5001800US0405"
-    AZ_06 = "5001800US0406"
-    AZ_07 = "5001800US0407"
-    AZ_08 = "5001800US0408"
-    AZ_09 = "5001800US0409"
-    AR_01 = "5001800US0501"
-    AR_02 = "5001800US0502"
-    AR_03 = "5001800US0503"
-    AR_04 = "5001800US0504"
-    CA_01 = "5001800US0601"
-    CA_02 = "5001800US0602"
-    CA_03 = "5001800US0603"
-    CA_04 = "5001800US0604"
-    CA_05 = "5001800US0605"
-    CA_06 = "5001800US0606"
-    CA_07 = "5001800US0607"
-    CA_08 = "5001800US0608"
-    CA_09 = "5001800US0609"
-    CA_10 = "5001800US0610"
-    CA_11 = "5001800US0611"
-    CA_12 = "5001800US0612"
-    CA_13 = "5001800US0613"
-    CA_14 = "5001800US0614"
-    CA_15 = "5001800US0615"
-    CA_16 = "5001800US0616"
-    CA_17 = "5001800US0617"
-    CA_18 = "5001800US0618"
-    CA_19 = "5001800US0619"
-    CA_20 = "5001800US0620"
-    CA_21 = "5001800US0621"
-    CA_22 = "5001800US0622"
-    CA_23 = "5001800US0623"
-    CA_24 = "5001800US0624"
-    CA_25 = "5001800US0625"
-    CA_26 = "5001800US0626"
-    CA_27 = "5001800US0627"
-    CA_28 = "5001800US0628"
-    CA_29 = "5001800US0629"
-    CA_30 = "5001800US0630"
-    CA_31 = "5001800US0631"
-    CA_32 = "5001800US0632"
-    CA_33 = "5001800US0633"
-    CA_34 = "5001800US0634"
-    CA_35 = "5001800US0635"
-    CA_36 = "5001800US0636"
-    CA_37 = "5001800US0637"
-    CA_38 = "5001800US0638"
-    CA_39 = "5001800US0639"
-    CA_40 = "5001800US0640"
-    CA_41 = "5001800US0641"
-    CA_42 = "5001800US0642"
-    CA_43 = "5001800US0643"
-    CA_44 = "5001800US0644"
-    CA_45 = "5001800US0645"
-    CA_46 = "5001800US0646"
-    CA_47 = "5001800US0647"
-    CA_48 = "5001800US0648"
-    CA_49 = "5001800US0649"
-    CA_50 = "5001800US0650"
-    CA_51 = "5001800US0651"
-    CA_52 = "5001800US0652"
-    CO_01 = "5001800US0801"
-    CO_02 = "5001800US0802"
-    CO_03 = "5001800US0803"
-    CO_04 = "5001800US0804"
-    CO_05 = "5001800US0805"
-    CO_06 = "5001800US0806"
-    CO_07 = "5001800US0807"
-    CO_08 = "5001800US0808"
-    CT_01 = "5001800US0901"
-    CT_02 = "5001800US0902"
-    CT_03 = "5001800US0903"
-    CT_04 = "5001800US0904"
-    CT_05 = "5001800US0905"
-    DE_01 = "5001800US1000"
-    DC_01 = "5001800US1198"
-    FL_01 = "5001800US1201"
-    FL_02 = "5001800US1202"
-    FL_03 = "5001800US1203"
-    FL_04 = "5001800US1204"
-    FL_05 = "5001800US1205"
-    FL_06 = "5001800US1206"
-    FL_07 = "5001800US1207"
-    FL_08 = "5001800US1208"
-    FL_09 = "5001800US1209"
-    FL_10 = "5001800US1210"
-    FL_11 = "5001800US1211"
-    FL_12 = "5001800US1212"
-    FL_13 = "5001800US1213"
-    FL_14 = "5001800US1214"
-    FL_15 = "5001800US1215"
-    FL_16 = "5001800US1216"
-    FL_17 = "5001800US1217"
-    FL_18 = "5001800US1218"
-    FL_19 = "5001800US1219"
-    FL_20 = "5001800US1220"
-    FL_21 = "5001800US1221"
-    FL_22 = "5001800US1222"
-    FL_23 = "5001800US1223"
-    FL_24 = "5001800US1224"
-    FL_25 = "5001800US1225"
-    FL_26 = "5001800US1226"
-    FL_27 = "5001800US1227"
-    FL_28 = "5001800US1228"
-    GA_01 = "5001800US1301"
-    GA_02 = "5001800US1302"
-    GA_03 = "5001800US1303"
-    GA_04 = "5001800US1304"
-    GA_05 = "5001800US1305"
-    GA_06 = "5001800US1306"
-    GA_07 = "5001800US1307"
-    GA_08 = "5001800US1308"
-    GA_09 = "5001800US1309"
-    GA_10 = "5001800US1310"
-    GA_11 = "5001800US1311"
-    GA_12 = "5001800US1312"
-    GA_13 = "5001800US1313"
-    GA_14 = "5001800US1314"
-    HI_01 = "5001800US1501"
-    HI_02 = "5001800US1502"
-    ID_01 = "5001800US1601"
-    ID_02 = "5001800US1602"
-    IL_01 = "5001800US1701"
-    IL_02 = "5001800US1702"
-    IL_03 = "5001800US1703"
-    IL_04 = "5001800US1704"
-    IL_05 = "5001800US1705"
-    IL_06 = "5001800US1706"
-    IL_07 = "5001800US1707"
-    IL_08 = "5001800US1708"
-    IL_09 = "5001800US1709"
-    IL_10 = "5001800US1710"
-    IL_11 = "5001800US1711"
-    IL_12 = "5001800US1712"
-    IL_13 = "5001800US1713"
-    IL_14 = "5001800US1714"
-    IL_15 = "5001800US1715"
-    IL_16 = "5001800US1716"
-    IL_17 = "5001800US1717"
-    IN_01 = "5001800US1801"
-    IN_02 = "5001800US1802"
-    IN_03 = "5001800US1803"
-    IN_04 = "5001800US1804"
-    IN_05 = "5001800US1805"
-    IN_06 = "5001800US1806"
-    IN_07 = "5001800US1807"
-    IN_08 = "5001800US1808"
-    IN_09 = "5001800US1809"
-    IA_01 = "5001800US1901"
-    IA_02 = "5001800US1902"
-    IA_03 = "5001800US1903"
-    IA_04 = "5001800US1904"
-    KS_01 = "5001800US2001"
-    KS_02 = "5001800US2002"
-    KS_03 = "5001800US2003"
-    KS_04 = "5001800US2004"
-    KY_01 = "5001800US2101"
-    KY_02 = "5001800US2102"
-    KY_03 = "5001800US2103"
-    KY_04 = "5001800US2104"
-    KY_05 = "5001800US2105"
-    KY_06 = "5001800US2106"
-    LA_01 = "5001800US2201"
-    LA_02 = "5001800US2202"
-    LA_03 = "5001800US2203"
-    LA_04 = "5001800US2204"
-    LA_05 = "5001800US2205"
-    LA_06 = "5001800US2206"
-    ME_01 = "5001800US2301"
-    ME_02 = "5001800US2302"
-    MD_01 = "5001800US2401"
-    MD_02 = "5001800US2402"
-    MD_03 = "5001800US2403"
-    MD_04 = "5001800US2404"
-    MD_05 = "5001800US2405"
-    MD_06 = "5001800US2406"
-    MD_07 = "5001800US2407"
-    MD_08 = "5001800US2408"
-    MA_01 = "5001800US2501"
-    MA_02 = "5001800US2502"
-    MA_03 = "5001800US2503"
-    MA_04 = "5001800US2504"
-    MA_05 = "5001800US2505"
-    MA_06 = "5001800US2506"
-    MA_07 = "5001800US2507"
-    MA_08 = "5001800US2508"
-    MA_09 = "5001800US2509"
-    MI_01 = "5001800US2601"
-    MI_02 = "5001800US2602"
-    MI_03 = "5001800US2603"
-    MI_04 = "5001800US2604"
-    MI_05 = "5001800US2605"
-    MI_06 = "5001800US2606"
-    MI_07 = "5001800US2607"
-    MI_08 = "5001800US2608"
-    MI_09 = "5001800US2609"
-    MI_10 = "5001800US2610"
-    MI_11 = "5001800US2611"
-    MI_12 = "5001800US2612"
-    MI_13 = "5001800US2613"
-    MN_01 = "5001800US2701"
-    MN_02 = "5001800US2702"
-    MN_03 = "5001800US2703"
-    MN_04 = "5001800US2704"
-    MN_05 = "5001800US2705"
-    MN_06 = "5001800US2706"
-    MN_07 = "5001800US2707"
-    MN_08 = "5001800US2708"
-    MS_01 = "5001800US2801"
-    MS_02 = "5001800US2802"
-    MS_03 = "5001800US2803"
-    MS_04 = "5001800US2804"
-    MO_01 = "5001800US2901"
-    MO_02 = "5001800US2902"
-    MO_03 = "5001800US2903"
-    MO_04 = "5001800US2904"
-    MO_05 = "5001800US2905"
-    MO_06 = "5001800US2906"
-    MO_07 = "5001800US2907"
-    MO_08 = "5001800US2908"
-    MT_01 = "5001800US3001"
-    MT_02 = "5001800US3002"
-    NE_01 = "5001800US3101"
-    NE_02 = "5001800US3102"
-    NE_03 = "5001800US3103"
-    NV_01 = "5001800US3201"
-    NV_02 = "5001800US3202"
-    NV_03 = "5001800US3203"
-    NV_04 = "5001800US3204"
-    NH_01 = "5001800US3301"
-    NH_02 = "5001800US3302"
-    NJ_01 = "5001800US3401"
-    NJ_02 = "5001800US3402"
-    NJ_03 = "5001800US3403"
-    NJ_04 = "5001800US3404"
-    NJ_05 = "5001800US3405"
-    NJ_06 = "5001800US3406"
-    NJ_07 = "5001800US3407"
-    NJ_08 = "5001800US3408"
-    NJ_09 = "5001800US3409"
-    NJ_10 = "5001800US3410"
-    NJ_11 = "5001800US3411"
-    NJ_12 = "5001800US3412"
-    NM_01 = "5001800US3501"
-    NM_02 = "5001800US3502"
-    NM_03 = "5001800US3503"
-    NY_01 = "5001800US3601"
-    NY_02 = "5001800US3602"
-    NY_03 = "5001800US3603"
-    NY_04 = "5001800US3604"
-    NY_05 = "5001800US3605"
-    NY_06 = "5001800US3606"
-    NY_07 = "5001800US3607"
-    NY_08 = "5001800US3608"
-    NY_09 = "5001800US3609"
-    NY_10 = "5001800US3610"
-    NY_11 = "5001800US3611"
-    NY_12 = "5001800US3612"
-    NY_13 = "5001800US3613"
-    NY_14 = "5001800US3614"
-    NY_15 = "5001800US3615"
-    NY_16 = "5001800US3616"
-    NY_17 = "5001800US3617"
-    NY_18 = "5001800US3618"
-    NY_19 = "5001800US3619"
-    NY_20 = "5001800US3620"
-    NY_21 = "5001800US3621"
-    NY_22 = "5001800US3622"
-    NY_23 = "5001800US3623"
-    NY_24 = "5001800US3624"
-    NY_25 = "5001800US3625"
-    NY_26 = "5001800US3626"
-    NC_01 = "5001800US3701"
-    NC_02 = "5001800US3702"
-    NC_03 = "5001800US3703"
-    NC_04 = "5001800US3704"
-    NC_05 = "5001800US3705"
-    NC_06 = "5001800US3706"
-    NC_07 = "5001800US3707"
-    NC_08 = "5001800US3708"
-    NC_09 = "5001800US3709"
-    NC_10 = "5001800US3710"
-    NC_11 = "5001800US3711"
-    NC_12 = "5001800US3712"
-    NC_13 = "5001800US3713"
-    NC_14 = "5001800US3714"
-    ND_01 = "5001800US3800"
-    OH_01 = "5001800US3901"
-    OH_02 = "5001800US3902"
-    OH_03 = "5001800US3903"
-    OH_04 = "5001800US3904"
-    OH_05 = "5001800US3905"
-    OH_06 = "5001800US3906"
-    OH_07 = "5001800US3907"
-    OH_08 = "5001800US3908"
-    OH_09 = "5001800US3909"
-    OH_10 = "5001800US3910"
-    OH_11 = "5001800US3911"
-    OH_12 = "5001800US3912"
-    OH_13 = "5001800US3913"
-    OH_14 = "5001800US3914"
-    OH_15 = "5001800US3915"
-    OK_01 = "5001800US4001"
-    OK_02 = "5001800US4002"
-    OK_03 = "5001800US4003"
-    OK_04 = "5001800US4004"
-    OK_05 = "5001800US4005"
-    OR_01 = "5001800US4101"
-    OR_02 = "5001800US4102"
-    OR_03 = "5001800US4103"
-    OR_04 = "5001800US4104"
-    OR_05 = "5001800US4105"
-    OR_06 = "5001800US4106"
-    PA_01 = "5001800US4201"
-    PA_02 = "5001800US4202"
-    PA_03 = "5001800US4203"
-    PA_04 = "5001800US4204"
-    PA_05 = "5001800US4205"
-    PA_06 = "5001800US4206"
-    PA_07 = "5001800US4207"
-    PA_08 = "5001800US4208"
-    PA_09 = "5001800US4209"
-    PA_10 = "5001800US4210"
-    PA_11 = "5001800US4211"
-    PA_12 = "5001800US4212"
-    PA_13 = "5001800US4213"
-    PA_14 = "5001800US4214"
-    PA_15 = "5001800US4215"
-    PA_16 = "5001800US4216"
-    PA_17 = "5001800US4217"
-    RI_01 = "5001800US4401"
-    RI_02 = "5001800US4402"
-    SC_01 = "5001800US4501"
-    SC_02 = "5001800US4502"
-    SC_03 = "5001800US4503"
-    SC_04 = "5001800US4504"
-    SC_05 = "5001800US4505"
-    SC_06 = "5001800US4506"
-    SC_07 = "5001800US4507"
-    SD_01 = "5001800US4600"
-    TN_01 = "5001800US4701"
-    TN_02 = "5001800US4702"
-    TN_03 = "5001800US4703"
-    TN_04 = "5001800US4704"
-    TN_05 = "5001800US4705"
-    TN_06 = "5001800US4706"
-    TN_07 = "5001800US4707"
-    TN_08 = "5001800US4708"
-    TN_09 = "5001800US4709"
-    TX_01 = "5001800US4801"
-    TX_02 = "5001800US4802"
-    TX_03 = "5001800US4803"
-    TX_04 = "5001800US4804"
-    TX_05 = "5001800US4805"
-    TX_06 = "5001800US4806"
-    TX_07 = "5001800US4807"
-    TX_08 = "5001800US4808"
-    TX_09 = "5001800US4809"
-    TX_10 = "5001800US4810"
-    TX_11 = "5001800US4811"
-    TX_12 = "5001800US4812"
-    TX_13 = "5001800US4813"
-    TX_14 = "5001800US4814"
-    TX_15 = "5001800US4815"
-    TX_16 = "5001800US4816"
-    TX_17 = "5001800US4817"
-    TX_18 = "5001800US4818"
-    TX_19 = "5001800US4819"
-    TX_20 = "5001800US4820"
-    TX_21 = "5001800US4821"
-    TX_22 = "5001800US4822"
-    TX_23 = "5001800US4823"
-    TX_24 = "5001800US4824"
-    TX_25 = "5001800US4825"
-    TX_26 = "5001800US4826"
-    TX_27 = "5001800US4827"
-    TX_28 = "5001800US4828"
-    TX_29 = "5001800US4829"
-    TX_30 = "5001800US4830"
-    TX_31 = "5001800US4831"
-    TX_32 = "5001800US4832"
-    TX_33 = "5001800US4833"
-    TX_34 = "5001800US4834"
-    TX_35 = "5001800US4835"
-    TX_36 = "5001800US4836"
-    TX_37 = "5001800US4837"
-    TX_38 = "5001800US4838"
-    UT_01 = "5001800US4901"
-    UT_02 = "5001800US4902"
-    UT_03 = "5001800US4903"
-    UT_04 = "5001800US4904"
-    VT_01 = "5001800US5000"
-    VA_01 = "5001800US5101"
-    VA_02 = "5001800US5102"
-    VA_03 = "5001800US5103"
-    VA_04 = "5001800US5104"
-    VA_05 = "5001800US5105"
-    VA_06 = "5001800US5106"
-    VA_07 = "5001800US5107"
-    VA_08 = "5001800US5108"
-    VA_09 = "5001800US5109"
-    VA_10 = "5001800US5110"
-    VA_11 = "5001800US5111"
-    WA_01 = "5001800US5301"
-    WA_02 = "5001800US5302"
-    WA_03 = "5001800US5303"
-    WA_04 = "5001800US5304"
-    WA_05 = "5001800US5305"
-    WA_06 = "5001800US5306"
-    WA_07 = "5001800US5307"
-    WA_08 = "5001800US5308"
-    WA_09 = "5001800US5309"
-    WA_10 = "5001800US5310"
-    WV_01 = "5001800US5401"
-    WV_02 = "5001800US5402"
-    WI_01 = "5001800US5501"
-    WI_02 = "5001800US5502"
-    WI_03 = "5001800US5503"
-    WI_04 = "5001800US5504"
-    WI_05 = "5001800US5505"
-    WI_06 = "5001800US5506"
-    WI_07 = "5001800US5507"
-    WI_08 = "5001800US5508"
-    WY_01 = "5001800US5600"
-
-    def get_hierarchical_codes(self) -> List[str]:
-        """
-        Returns all applicable UCGID codes for this enum value in hierarchical order.
-
-        Returns:
-            List of UCGID codes from most specific to least specific
-
-        Examples:
-            UCGID.CA_23.get_hierarchical_codes() -> ["5001800US0623", "0400000US06", "0100000US"]
-            UCGID.CA.get_hierarchical_codes() -> ["0400000US06", "0100000US"]
-            UCGID.US.get_hierarchical_codes() -> ["0100000US"]
-        """
-        code = self.value
-        codes_list = [code]
-
-        # If it's a district code (Congressional District format: 5001800US + state + district)
-        if code.startswith("5001800US"):
-            # Extract state code from district code
-            state_code_num = code[9:11]  # Extract 2-digit state code
-            state_code = f"0400000US{state_code_num}"
-
-            # Add state code if it exists in the enum
-            for state_enum in UCGID:
-                if state_enum.value == state_code:
-                    codes_list.append(state_code)
-                    break
-
-            # Add US code
-            codes_list.append("0100000US")
-
-        # If it's a state code (State format: 0400000US + state code)
-        elif code.startswith("0400000US"):
-            # Add US code
-            codes_list.append("0100000US")
-
-        # If it's already the US code, return as-is
-        # (US code: "0100000US")
-
-        return codes_list
-
-    def matches_hierarchy(self, target_code: str) -> bool:
-        """
-        Check if this UCGID code matches a target code in the hierarchy.
-
-        Args:
-            target_code: The target code to match against
-
-        Returns:
-            True if this code belongs to the target code's hierarchy
-
-        Examples:
-            UCGID.CA_23.matches_hierarchy("0400000US06") -> True (CA_23 is in CA)
-            UCGID.CA_23.matches_hierarchy("0100000US") -> True (CA_23 is in US)
-            UCGID.CA.matches_hierarchy("5001800US0623") -> False (CA is not CA_23)
-        """
-        hierarchical_codes = self.get_hierarchical_codes()
-        return target_code in hierarchical_codes

policyengine_us/variables/household/demographic/geographic/ucgid/ucgid_str.py

@@ -1,29 +0,0 @@
-from policyengine_us.model_api import *
-from policyengine_us.variables.household.demographic.geographic.ucgid.ucgid_enum import (
-    UCGID,
-)
-
-
-class ucgid_str(Variable):
-    value_type = str
-    entity = Household
-    label = "UCGID (string)"
-    documentation = "UCGID variable, stored as a string"
-    definition_period = YEAR
-
-    def formula(household, period, parameters):
-        import numpy as np
-
-        ucgid_enum_names = household("ucgid", period).decode_to_str()
-
-        # Convert each enum name to its hierarchical codes
-        result = []
-        for enum_name in ucgid_enum_names:
-            # Get the enum instance from its name
-            ucgid_enum = UCGID[enum_name]
-
-            # Get all hierarchical codes and join with commas
-            hierarchical_codes = ucgid_enum.get_hierarchical_codes()
-            result.append(",".join(hierarchical_codes))
-
-        return np.array(result)