def test_asset_bubble(): ''' Test asset bubble 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.asset_bubble(calc2) assert fig fig = calc.asset_bubble(calc2, output_variable='rho_mix') assert fig
def test_asset_bubble(): ''' Test asset bubble plot 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.asset_bubble(calc2, include_title=True) assert fig fig = calc.asset_bubble(calc2, output_variable='rho_mix', include_title=True) assert fig