def test_dsct(): population = create_empty_population_dataframe(2001, 2061) cohorts = Cohorts(data = population, columns = ['pop']) cohorts.gen_dsct(0.05) test_value = cohorts.get_value((0,0,2060), 'dsct') # print test_value assert test_value <= 1
def test_empty_frame_generation(): year_start = 2001 year_end = 2061 population_dataframe = create_empty_population_dataframe(year_start, year_end) test_value = population_dataframe.get_value((0,0,2043), "pop") assert test_value == 1
def test_fill_cohort(): population = create_empty_population_dataframe(2001, 2061) profiles = create_constant_profiles_dataframe(population, tax = -1, subsidies = 0.5) cohorts_test = DataCohorts(data = population, columns = ['pop']) cohorts_test.fill(profiles, year = None) test_value = cohorts_test.get_value((0,0,2060), 'tax') assert test_value == -1
def test_dsct(): population = create_empty_population_dataframe(2001, 2061) cohorts = Cohorts(data=population, columns=["pop"]) cohorts.gen_dsct(0.05) test_value = cohorts.get_value((0, 0, 2060), "dsct") # print test_value assert test_value <= 1
def test_empty_frame_generation(): year_start = 2001 year_end = 2061 population_dataframe = create_empty_population_dataframe(year_start, year_end) test_value = population_dataframe.get_value((0, 0, 2043), "pop") assert test_value == 1
def test_column_combination(): year_start = 2001 pop_dataframe = create_empty_population_dataframe(year_start, 2061) # print pop_dataframe profiles_dataframe = create_constant_profiles_dataframe(pop_dataframe, tax=-1.0, sub=0.5) # print profiles_dataframe profiles_dataframe['net'] = profiles_dataframe.sum(axis=1) # print profiles_dataframe['net'] test_value = profiles_dataframe.get_value((0,0,2001), 'net') # print test_value assert test_value == -0.5
def test_column_combination(): year_start = 2001 pop_dataframe = create_empty_population_dataframe(year_start, 2061) # print pop_dataframe profiles_dataframe = create_constant_profiles_dataframe(pop_dataframe, tax=-1.0, sub=0.5) # print profiles_dataframe profiles_dataframe["net"] = profiles_dataframe.sum(axis=1) # print profiles_dataframe['net'] test_value = profiles_dataframe.get_value((0, 0, 2001), "net") # print test_value assert test_value == -0.5
def test_compute_net_transfers(): population = create_empty_population_dataframe(2001, 2061) profiles = create_constant_profiles_dataframe(population, tax = 1, subsidies = 0.5) tax = ['tax'] subsidy = ['subsidies'] cohorts_test = DataCohorts(data = population, columns = ['pop']) cohorts_test.fill(profiles, year = None) cohorts_test.compute_net_transfers(taxes_list = tax, payments_list = subsidy) test_value = cohorts_test.get_value((0,0,2060), 'net_transfers') assert test_value == 0.5 pass
def test_population_projection(): # Create cohorts start_data = 2001 end_data = 2061 population = create_empty_population_dataframe(start_data, end_data) cohorts = DataCohorts(data=population, columns=['pop']) # Complete population projection year_length = 100 end_project = start_data + year_length method = 'exp_growth' growth_rate = n = 0.05 cohorts.population_project(year_length, method=method, growth_rate=n) year_control = 2082 control_value = (1 + n)**(year_control - end_data - 1) test_value = cohorts.get_value((0, 0, 2081), "pop") assert test_value == control_value
def test_population_projection(): # Create cohorts start_data = 2001 end_data = 2061 population = create_empty_population_dataframe(start_data, end_data) cohorts = DataCohorts(data = population, columns = ['pop']) # Complete population projection year_length = 100 end_project = start_data + year_length method = 'exp_growth' growth_rate = n = 0.05 cohorts.population_project(year_length, method = method, growth_rate = n) year_control = 2082 control_value = (1+n)**(year_control - end_data - 1) test_value = cohorts.get_value((0,0,2081), "pop") assert test_value == control_value
def test_tax_projection(): population = create_empty_population_dataframe(2001, 2061) profile = create_constant_profiles_dataframe(population, tax=-1, sub=0.5) g = 0.05 r = 0 cohort = DataCohorts(population) year_length = 200 method = 'stable' cohort.population_project(year_length, method=method) cohort._fill(profile) typ = None cohort.proj_tax(g, r, typ, method='per_capita') # print cohort test_value = cohort.get_value((0, 1, 2002), 'tax') test_value2 = cohort.get_value((0, 1, 2002), 'sub') # print test_value, test_value2 # TODO: I don't understand the following <- Just wanted to check if the for loop works by changing the value of typ in the test assert test_value2 > 0.5 and test_value < -1
def test_tax_projection(): population = create_empty_population_dataframe(2001, 2061) profile = create_constant_profiles_dataframe(population, tax = -1, sub=0.5) g = 0.05 r = 0 cohort = DataCohorts(population) year_length = 200 method = 'stable' cohort.population_project(year_length, method = method) cohort.fill(profile) typ = None cohort.proj_tax(g, r, typ, method = 'per_capita') # print cohort test_value = cohort.get_value((0,1,2002), 'tax') test_value2 = cohort.get_value((0,1,2002), 'sub') # print test_value, test_value2 # TODO: I don't understand the following <- Just wanted to check if the for loop works by changing the value of typ in the test assert test_value2 > 0.5 and test_value < -1