debug_data_frame = debug_data_frame.merge(
erf_individus_data_frame,
how = 'inner',
on = 'idmen',
)
suffixes = ["_erf", "_of", "_rel_diff", "_ratio"]
reordered_columns = [variable + suffixe for suffixe in suffixes] \
+ ["idmen", "quimen", "idfam", "quifam", "idfoy", "quifoy"]
reordered_columns = reordered_columns + list(set(kept_columns) - set(reordered_columns))
debug_data_frame = debug_data_frame[reordered_columns].copy()
print debug_data_frame.to_string()
if __name__ == '__main__':
import sys
from openfisca_plugin_aggregates.tests.test_aggregates import create_survey_scenario
logging.basicConfig(level = logging.INFO, stream = sys.stdout)
restart = True
if restart:
year = 2006
survey_scenario = create_survey_scenario(year)
survey_scenario.simulation = survey_scenario.new_simulation()
debugger = Debugger()
debugger.set_survey_scenario(survey_scenario = survey_scenario)
debugger.set_variable('af')
# debugger.show_aggregates()
debugger.describe_discrepancies(descending = False)
def test_inequality(year = 2009):
survey_scenario = create_survey_scenario(year)
inequality = Inequality()
inequality.set_survey_scenario(survey_scenario)
inequality.compute()
return inequality.inequality_data_frame, inequality.poverty
continue
extracted_indices = extracted_indices + \
input_data_frame.index[input_data_frame[entity.index_for_person_variable_name].isin(entity_index)]
extracted_input_data_frame = input_data_frame.loc[extracted_indices]
return extracted_input_data_frame
if __name__ == '__main__':
import sys
from openfisca_plugin_aggregates.tests.test_aggregates import create_survey_scenario
logging.basicConfig(level = logging.INFO, stream = sys.stdout)
restart = True
if restart:
year = 2009
survey_scenario = create_survey_scenario(year)
survey_scenario.simulation = survey_scenario.new_simulation()
debugger = Debugger()
debugger.set_survey_scenario(survey_scenario = survey_scenario)
debugger.set_variable('af')
debugger.build_columns_to_fetch()
debugger.build_openfisca_data_frames()
debugger.build_erf_data_frames()
# df_menage = debugger.data_frame_by_entity_key_plural['menages']
# df_famille = debugger.data_frame_by_entity_key_plural['familles']
# df_individus = debugger.data_frame_by_entity_key_plural['individus']
# df = debugger.get_major_differences()
def test_inequality(year=2009):
survey_scenario = create_survey_scenario(year)
inequality = Inequality()
inequality.set_survey_scenario(survey_scenario)
inequality.compute()
return inequality.inequality_data_frame, inequality.poverty
