示例#1
0
def test_from_person_variable_to_group(tax_benefit_system):
    persons_ids: Iterable = [2, 0, 1, 4, 3]
    households_ids: Iterable = ['c', 'a', 'b']

    persons_households: Iterable = ['c', 'a', 'a', 'b', 'a']

    persons_salaries: Iterable = [6000, 2000, 1000, 1500, 1500]
    households_rents = [1036.6667, 781.6667, 271.6667]

    period = '2018-12'

    simulation_builder = SimulationBuilder()
    simulation_builder.create_entities(tax_benefit_system)
    simulation_builder.declare_person_entity('person', persons_ids)

    household_instance = simulation_builder.declare_entity(
        'household', households_ids)
    simulation_builder.join_with_persons(household_instance,
                                         persons_households,
                                         ['first_parent'] * 5)

    simulation = simulation_builder.build(tax_benefit_system)
    simulation.set_input('salary', period, persons_salaries)
    simulation.set_input('rent', period, households_rents)

    total_taxes = simulation.calculate('total_taxes', period)
    assert total_taxes == pytest.approx(households_rents)
    assert total_taxes / simulation.calculate('rent',
                                              period) == pytest.approx(1)
示例#2
0
def test_nb_persons_in_households(tax_benefit_system):
    persons_ids: Iterable = [2, 0, 1, 4, 3]
    households_ids: Iterable = ['c', 'a', 'b']
    persons_households: Iterable = ['c', 'a', 'a', 'b', 'a']

    simulation_builder = SimulationBuilder()
    simulation_builder.create_entities(tax_benefit_system)
    simulation_builder.declare_person_entity('person', persons_ids)
    household_instance = simulation_builder.declare_entity(
        'household', households_ids)
    simulation_builder.join_with_persons(household_instance,
                                         persons_households,
                                         ['first_parent'] * 5)

    persons_in_households = simulation_builder.nb_persons('household')

    assert persons_in_households.tolist() == [1, 3, 1]
示例#3
0
def test_integral_roles(tax_benefit_system):
    persons_ids: Iterable = [2, 0, 1, 4, 3]
    households_ids: Iterable = ['c', 'a', 'b']
    persons_households: Iterable = ['c', 'a', 'a', 'b', 'a']
    # Same roles as test_nb_persons_by_role
    persons_households_roles: Iterable = [2, 0, 1, 0, 2]

    simulation_builder = SimulationBuilder()
    simulation_builder.create_entities(tax_benefit_system)
    simulation_builder.declare_person_entity('person', persons_ids)
    household_instance = simulation_builder.declare_entity(
        'household', households_ids)

    simulation_builder.join_with_persons(household_instance,
                                         persons_households,
                                         persons_households_roles)
    parents_in_households = household_instance.nb_persons(
        role=entities.Household.FIRST_PARENT)

    assert parents_in_households.tolist() == [0, 1, 1]
示例#4
0
def test_nb_persons_no_role(tax_benefit_system):
    persons_ids: Iterable = [2, 0, 1, 4, 3]
    households_ids: Iterable = ['c', 'a', 'b']
    persons_households: Iterable = ['c', 'a', 'a', 'b', 'a']

    simulation_builder = SimulationBuilder()
    simulation_builder.create_entities(tax_benefit_system)
    simulation_builder.declare_person_entity('person', persons_ids)
    household_instance = simulation_builder.declare_entity(
        'household', households_ids)

    simulation_builder.join_with_persons(household_instance,
                                         persons_households,
                                         ['first_parent'] * 5)
    parents_in_households = household_instance.nb_persons(
        role=entities.Household.PARENT)

    assert parents_in_households.tolist() == [
        1, 3, 1
    ]  # household member default role is first_parent