def test_calc_depreciation_budget(default_btax_params):
"""
Test calcDep_budget method
"""
asset = Asset(default_btax_params)
asset.get_ccr_data()
asset.build_inv_matrix()
asset.build_deprLaw_matrices()
asset.calcDep_allyears()
def test_build_inv_matrix(default_btax_params):
"""
Test build_inv_matrix method with response
"""
resp_elasticities = {'inv_usercost_c': -1.0, 'inv_usercost_nc': -0.5}
resp = Response(resp_elasticities, default_btax_params, default_bax_params)
resp.calc_inv_response()
asset = Asset(default_btax_params, response=resp.investment_response)
asset.get_ccr_data()
asset.build_inv_matrix()
assert isinstance(asset.investment_history, pd.DataFrame)