def test_summary_table(): ''' Test summary_table method. ''' cyr = 2018 assets = Assets() p = Specification(year=cyr) calc1 = Calculator(p, assets) assert calc1.current_year == cyr p.update_specification({'CIT_rate': 0.38}) calc2 = Calculator(p, assets) assert calc2.current_year == cyr summary_df = calc1.summary_table(calc2) assert isinstance(summary_df, pd.DataFrame)
def test_summary_table(): ''' Test difference_table method. ''' yr = 2018 assets = Assets() p = Specifications(year=yr) calc1 = Calculator(p, assets) assert calc1.current_year == yr reform = {'CIT_rate': 0.38} p.update_specifications(reform) calc2 = Calculator(p, assets) assert calc2.current_year == yr summary_df = calc1.summary_table(calc2) assert isinstance(summary_df, pd.DataFrame)
def test_summary_table(include_land, include_inventories): ''' Test summary_table method. ''' cyr = 2018 assets = Assets() p = Specification(year=cyr) dp = DepreciationParams() calc1 = Calculator(p, dp, assets) assert calc1.current_year == cyr p.update_specification({'CIT_rate': 0.38}) calc2 = Calculator(p, dp, assets) assert calc2.current_year == cyr summary_df = calc1.summary_table(calc2, include_land=include_land, include_inventories=include_inventories) assert isinstance(summary_df, pd.DataFrame)