tax_agency.py

"""
Contains functions to calculate taxes and welfare benefits owed.
"""

from data import *
from filing_status import *
from total_benefit import *

tax_credit_functions = [get_EITC, get_CTC]


# Fills and returns a dictionary with data mapping TaxUnits to their benefits and taxes.
def generate_data(max_income: int, max_kids: int, step: int) -> Dict[TaxUnit, TotalBenefit]:
    tax_unit_to_benefits: Dict[TaxUnit, TotalBenefit] = {}
    curr_unit: TaxUnit
    curr_benefit: TotalBenefit
    for curr_income in range(0, max_income, step):
        for num_kids in range(max_kids+1):
            for filing_status in FilingStatus:
                curr_unit = TaxUnit(curr_income, num_kids, filing_status)
                curr_unit.base_income_taxes = get_taxfare(curr_income, brackets[filing_status])
                refundable = 0
                non_refundable = 0
                for func in tax_credit_functions:
                    result = func(curr_unit)
                    refundable += result[0]
                    non_refundable += result[1]
                curr_benefit = TotalBenefit(refundable, non_refundable, 0)
                tax_unit_to_benefits[curr_unit] = curr_benefit
    return tax_unit_to_benefits
# TODO: Add in code for what a UCA and UBI would be. Can then have it auto-generated compared to norm