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)