def test_compute_gen_imbalance():
size_generation = 1
cohort = create_neutral_profiles_cohort(population=size_generation)
simulation = Simulation()
simulation.discount_rate = 0
simulation.growth_rate = 0
simulation.cohorts = cohort
simulation.create_present_values('tax')
gen_imbalance = simulation.compute_gen_imbalance(typ='tax')
# print gen_imbalance
assert gen_imbalance == -5000.0 / (2 * 199.0)
def test_compute_gen_imbalance():
size_generation = 1
cohort = create_neutral_profiles_cohort(population = size_generation)
simulation = Simulation()
simulation.discount_rate = 0
simulation.growth_rate = 0
simulation.cohorts = cohort
simulation.create_present_values('tax')
gen_imbalance = simulation.compute_gen_imbalance(typ = 'tax')
# print gen_imbalance
assert gen_imbalance[0] == -5000.0/(2*199.0), gen_imbalance[1] == -5000.0/(2*199.0)
def test_compute_gen_imbalance():
size_generation = 1
cohort = create_neutral_profiles_cohort(population = size_generation)
simulation = Simulation()
simulation.discount_rate = 0
simulation.growth_rate = 0
simulation.cohorts = cohort
simulation.create_present_values('tax')
print simulation.aggregate_pv.tail(20).to_string()
print simulation.cohorts.head(20).to_string()
gen_imbalance = simulation.compute_gen_imbalance(typ = 'tax', to_return='difference')
# print gen_imbalance, -5000.0/(2*199.0)
assert gen_imbalance == -5000.0/(2*199.0)
def test_save_simulation():
size_generation = 1
cohort = create_neutral_profiles_cohort(population = size_generation)
simulation = Simulation()
simulation.country = 'france'
simulation.discount_rate = 0
simulation.growth_rate = 0
simulation.cohorts = cohort
simulation.create_present_values('tax')
simulation.save_simulation(filename='test_save')
store = HDFStore(os.path.join(SRC_PATH,'countries','france','sources', 'Output_folder','test_save.h5'))
assert store['aggregate_pv'] is not None
