def test_update_depreciation_params_with_json():
expected_result = [{
'year': 2020,
'value': {
'method': 'Expensing',
'life': 5
},
'GDS_life': 3.0,
'ADS_life': 3.0,
'major_asset_group': 'Equipment',
'minor_asset_group': 'Computers and Software',
'system': 'GDS',
'asset_name': 'Custom software',
'BEA_code': 'ENS2'
}]
dp = DepreciationParams()
new_dp_json = """
{"asset": [
{"year": 2020,
"asset_name": "Custom software",
"value": {"life": 5, "method": "Expensing"}}]}
"""
dp.adjust(new_dp_json)
test_result = dp.select_eq(param="asset",
exact_match=False,
year=2020,
BEA_code="ENS2")
assert test_result == expected_result
def test_update_depreciation_params_as_a_group():
dp = DepreciationParams()
new_dp_dict = {
"asset": [{
"major_asset_group": "Intellectual Property",
"value": {
"life": 12,
"method": "DB 200%"
}
}]
}
dp.adjust(new_dp_dict)
test_result = dp.select_eq(param="asset",
exact_match=False,
year=2020,
major_asset_group="Intellectual Property")
assert test_result[0]['value']['life'] == 12
assert test_result[1]['value']['life'] == 12
assert test_result[2]['value']['life'] == 12