def test_range_plot(): ''' Test range_plot method. ''' assets = Assets() p = Specifications() calc = Calculator(p, assets) p2 = Specifications(year=2026) p2.update_specifications({'CIT_rate': 0.25}) calc2 = Calculator(p2, assets) fig = calc.range_plot(calc2) assert fig fig = calc.range_plot(calc2, output_variable='rho') assert fig
def test_range_plot(corporate): ''' Test range_plot method. ''' assets = Assets() p = Specification() calc = Calculator(p, assets) p2 = Specification(year=2026) p2.update_specification({'CIT_rate': 0.25}) calc2 = Calculator(p2, assets) fig = calc.range_plot(calc2, corporate=corporate, include_title=True) assert fig fig = calc.range_plot(calc2, output_variable='rho', corporate=corporate, include_title=True) assert fig
# specify baseline and reform Calculator objects for 2019 calculations assets = Assets() baseline_parameters = Specification(year=cyr) calc1 = Calculator(baseline_parameters, assets) reform_parameters = Specification(year=cyr) reform_parameters.update_specification(business_tax_reform) calc2 = Calculator(reform_parameters, assets) # do calculations by asset and by industry baseln_assets_df = calc1.calc_by_asset() reform_assets_df = calc2.calc_by_asset() baseln_industry_df = calc1.calc_by_industry() reform_industry_df = calc2.calc_by_industry() # generate dataframes with reform-minus-baseline differences diff_assets_df = diff_two_tables(reform_assets_df, baseln_assets_df) diff_industry_df = diff_two_tables(reform_industry_df, baseln_industry_df) # save dataframes to disk as csv files in this directory baseln_industry_df.to_csv('baseline_byindustry.csv', float_format='%.5f') reform_industry_df.to_csv('reform_byindustry.csv', float_format='%.5f') baseln_assets_df.to_csv('baseline_byasset.csv', float_format='%.5f') reform_assets_df.to_csv('reform_byasset.csv', float_format='%.5f') diff_industry_df.to_csv('changed_byindustry.csv', float_format='%.5f') diff_assets_df.to_csv('changed_byasset.csv', float_format='%.5f') # create and show in browser a range plot p = calc1.range_plot(calc2) show(p)