policyengine-us

from policyengine_us import system from policyengine_core.entities import Person, Household from policyengine_core.simulation import Simulation # Define a simple household # A person named 'alice' who is 40 years old and has wage income alice = Person(age=40, wages=50000, employment_status='EMPLOYED') # A household with Alice household = Household(members=[alice]) # Get the current US tax-benefit system # By default, uses the latest available policy year (e.g., 2025 as of recent updates) sim_system = system.get_system() # Create a simulation for the household simulation = Simulation(household=household, system=sim_system) # Calculate a variable, e.g., Alice's net income in 2025 alice_net_income = simulation.get_person_variable(person_id=alice.id, variable_name='net_income', period='2025') print(f"Alice's net income in 2025: ${alice_net_income:,.2f}") # Example of applying a simple reform (e.g., a universal basic income) reform_code = { 'universal_basic_income': [ { 'period': '2025', 'value': 12000, 'description': 'A universal basic income of $12,000 per year for all adults.' } ] } # Apply the reform to the system reformed_system = system.get_system(reform=reform_code) # Run simulation with the reformed system reformed_simulation = Simulation(household=household, system=reformed_system) alice_net_income_reformed = reformed_simulation.get_person_variable( person_id=alice.id, variable_name='net_income', period='2025' ) print(f"Alice's net income in 2025 with UBI reform: ${alice_net_income_reformed:,.2f}")