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