def produce_graph(name="test", apply_reform=False, reforme=False, conj = False, sal_conj=False, kids = 0, save_figure = False, destination_dir = None, bareme=True, tax_rates=False, year=2011): app = QApplication(sys.argv) win = ApplicationWindow() ax = win.mplwidget.axes simulation = run_simulation(year=year, apply_reform=apply_reform, reforme=reforme, conj = conj, sal_conj=sal_conj, kids=kids) if bareme: draw_simulation_bareme(simulation, ax, legend = True, position = 4) if tax_rates: draw_simulation_taux(simulation, ax, legend=True) win.resize(1400,700) win.mplwidget.draw() win.show() df = simulation.get_results_dataframe() # print df.to_string() if save_figure and destination_dir is not None: win.mplwidget.print_figure(os.path.join(destination_dir, name + '.png')) del ax, simulation
def test_bareme(x_axis="sali"): """ Use to test and debug bareme mode test-case """ yr = 2012 # Changes in individualized characteristics # salaires: sali # retraites: choi # intérêts: f2ee intpfl; f2tr intb # dividendes: f2da divplf; f2dc divb # foncier f4ba fon (micro foncier f4be) maxrev = 500000 year = 2012 simulation = ScenarioSimulation() # Changes in individualized caracteristics # salaires: sali # retraites: choi # intérêts: f2ee intpfl; f2tr intb # dividendes: f2da divplf; f2dc divb # foncier f4ba fon (micro foncier f4be) simulation.set_config(year = yr, nmen = 101, x_axis = x_axis, maxrev=maxrev, reforme = False, mode ='bareme', decomp_file="decomp_contrib.xml") simulation.set_param() # Hack to get rid of missing parameters in 2012 if yr == 2012: complete_2012_param(simulation.P) test_case = simulation.scenario if SHOW_OPENFISCA: app = QApplication(sys.argv) win = ApplicationWindow() ax = win.mplwidget.axes title ="Barème openfisca" ax.set_title(title) graph_x_axis = simulation.get_varying_revenues(x_axis) draw_simulation_bareme(simulation, ax, graph_x_axis = graph_x_axis, legend = True, position = 4) win.resize(1400,700) win.mplwidget.draw() win.show() if EXPORT: win.mplwidget.print_figure(DESTINATION_DIR + title + '.png') if ax: del ax del simulation sys.exit(app.exec_())
def produce_graph(name="test", apply_reform=False, reforme=False, conj=False, sal_conj=False, kids=0, save_figure=False, destination_dir=None, bareme=True, tax_rates=False, year=2011): app = QApplication(sys.argv) win = ApplicationWindow() ax = win.mplwidget.axes simulation = run_simulation(year=year, apply_reform=apply_reform, reforme=reforme, conj=conj, sal_conj=sal_conj, kids=kids) if bareme: draw_simulation_bareme(simulation, ax, legend=True, position=4) if tax_rates: draw_simulation_taux(simulation, ax, legend=True) win.resize(1400, 700) win.mplwidget.draw() win.show() df = simulation.get_results_dataframe() # print df.to_string() if save_figure and destination_dir is not None: win.mplwidget.print_figure(os.path.join(destination_dir, name + '.png')) del ax, simulation
def test_bareme(x_axis="sali"): """ Use to test and debug bareme mode test-case """ yr = 2012 # Changes in individualized characteristics # salaires: sali # retraites: choi # intérêts: f2ee intpfl; f2tr intb # dividendes: f2da divplf; f2dc divb # foncier f4ba fon (micro foncier f4be) maxrev = 500000 year = 2012 simulation = ScenarioSimulation() # Changes in individualized caracteristics # salaires: sali # retraites: choi # intérêts: f2ee intpfl; f2tr intb # dividendes: f2da divplf; f2dc divb # foncier f4ba fon (micro foncier f4be) simulation.set_config(year=yr, nmen=101, x_axis=x_axis, maxrev=maxrev, reforme=False, mode='bareme', decomp_file="decomp_contrib.xml") simulation.set_param() # Hack to get rid of missing parameters in 2012 if yr == 2012: complete_2012_param(simulation.P) test_case = simulation.scenario if SHOW_OPENFISCA: app = QApplication(sys.argv) win = ApplicationWindow() ax = win.mplwidget.axes title = "Barème openfisca" ax.set_title(title) graph_x_axis = simulation.get_varying_revenues(x_axis) draw_simulation_bareme(simulation, ax, graph_x_axis=graph_x_axis, legend=True, position=4) win.resize(1400, 700) win.mplwidget.draw() win.show() if EXPORT: win.mplwidget.print_figure(DESTINATION_DIR + title + '.png') if ax: del ax del simulation sys.exit(app.exec_())
reforme=False, mode='bareme', same_rev_couple=True) simu.set_param() simu.scenario.addIndiv(1, datetime(1975, 1, 1).date(), 'conj', 'part') if marginal is True: for i in range(1, n_enf_final): print 'adding %s kid(s)' % i simu.scenario.addIndiv(1 + i, datetime(2000, 1, 1).date(), 'pac', 'enf') simu.set_marginal_alternative_scenario() for i in range(1, n_enf_final + 1): if (marginal is False) or (i == n_enf_final): print 'adding %s kid(s)' % i simu.alternative_scenario.addIndiv(1 + i, datetime(2000, 1, 1).date(), 'pac', 'enf') ax.set_title(title) draw_simulation_bareme(simu, ax, legend=True, position=4) win.resize(1400, 700) win.mplwidget.draw() win.show() win.mplwidget.print_figure(destination_dir + title + '.png') del ax, simu sys.exit(app.exec_())