def test_grouped_bar(): ''' Test grouped_bar 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.grouped_bar(calc2) assert fig fig = calc.grouped_bar(calc2, output_variable='rho', group_by_asset=False) assert fig
def test_grouped_bar(corporate): ''' Test grouped_bar method. ''' assets = Assets() p = Specification() dp = DepreciationParams() calc = Calculator(p, dp, assets) p2 = Specification(year=2026) p2.update_specification({'CIT_rate': 0.25}) calc2 = Calculator(p2, dp, assets) fig = calc.grouped_bar(calc2, corporate=corporate) assert fig fig = calc.grouped_bar(calc2, output_variable='rho', corporate=corporate, group_by_asset=False) assert fig