Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions changelog.d/codex-org-inputs.added.md
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Add ORG-backed `hourly_wage` and `is_union_member_or_covered` input variables.
15 changes: 15 additions & 0 deletions policyengine_us/tests/core/test_org_input_variables.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
from policyengine_us import CountryTaxBenefitSystem


def test_org_input_variables_exist():
system = CountryTaxBenefitSystem()

assert "hourly_wage" in system.variables
assert "is_union_member_or_covered" in system.variables


def test_org_input_variable_types():
system = CountryTaxBenefitSystem()

assert system.variables["hourly_wage"].value_type is float
assert system.variables["is_union_member_or_covered"].value_type is bool
10 changes: 10 additions & 0 deletions policyengine_us/variables/household/income/person/hourly_wage.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
from policyengine_us.model_api import *


class hourly_wage(Variable):
value_type = float
entity = Person
label = "hourly wage"
documentation = "Reported hourly wage, imputed from CPS ORG donors."
unit = USD
definition_period = YEAR
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
from policyengine_us.model_api import *


class is_union_member_or_covered(Variable):
value_type = bool
entity = Person
label = "is union member or covered by a union contract"
definition_period = YEAR
Loading