Python Asset.build_inv_matrix Examples

Example #1

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()

Example #2

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)