def test_instantiation_and_usage():
"""
Test ParametersBase instantiation and usage.
"""
pbase = ParametersBase()
assert pbase
assert pbase.inflation_rates() is None
assert pbase.wage_growth_rates() is None
syr = 2010
nyrs = 10
pbase.initialize(start_year=syr, num_years=nyrs)
# pylint: disable=protected-access
with pytest.raises(ValueError):
pbase.set_year(syr - 1)
with pytest.raises(NotImplementedError):
pbase._params_dict_from_json_file()
with pytest.raises(ValueError):
pbase._update([])
with pytest.raises(ValueError):
pbase._update({})
with pytest.raises(ValueError):
pbase._update({(syr + nyrs): {}})
with pytest.raises(ValueError):
pbase._update({syr: []})
# pylint: disable=no-member
with pytest.raises(ValueError):
ParametersBase._expand_array({}, True, True, [0.02], 1)
threedarray = np.array([[[1, 1]], [[1, 1]], [[1, 1]]])
with pytest.raises(ValueError):
ParametersBase._expand_array(threedarray, True, True, [0.02, 0.02], 2)
def test_expand_2d_accept_none():
# pylint doesn't like caps in var name, so pylint: disable=invalid-name
_II_brk2 = [[36000, 72250, 36500, 48600, 72500],
[38000, 74000, 36900, 49400, 73800],
[40000, 74900, 37450, 50200, 74900],
[41000, None, None, None, None]]
exp1 = 74900 * 1.02
exp2 = 37450 * 1.02
exp3 = 50200 * 1.02
exp4 = 74900 * 1.02
exp = [[36000, 72250, 36500, 48600, 72500],
[38000, 74000, 36900, 49400, 73800],
[40000, 74900, 37450, 50200, 74900],
[41000, exp1, exp2, exp3, exp4]]
exp = np.array(exp).astype('i4', casting='unsafe')
res = ParametersBase._expand_array(_II_brk2,
inflate=True,
inflation_rates=[0.02] * 5,
num_years=4)
assert_equal(res, exp)
syr = 2013
pol = Policy(start_year=syr)
irates = pol.inflation_rates()
reform = {2016: {u'_II_brk2': _II_brk2}}
pol.implement_reform(reform)
pol.set_year(2019)
# The 2019 policy should be the combination of the user-defined
# value and CPI-inflated values from 2018
exp_2019 = [41000.] + [(1.0 + irates[2018 - syr]) * i
for i in _II_brk2[2][1:]]
exp_2019 = np.array(exp_2019)
assert_equal(pol.II_brk2, exp_2019)
def test_expand_1d_accept_none():
lst = [4., 5., None]
irates = [0.02, 0.02, 0.03, 0.035]
exp = []
cur = 5.0 * 1.02
exp = [4., 5., cur]
cur *= 1.03
exp.append(cur)
cur *= 1.035
exp.append(cur)
exp = np.array(exp)
res = ParametersBase._expand_array(lst,
inflate=True,
inflation_rates=irates,
num_years=5)
assert np.allclose(exp.astype('f4', casting='unsafe'), res)
def test_expand_2d_accept_none_add_row():
# pylint doesn't like caps in var name, so pylint: disable=invalid-name
_II_brk2 = [[36000, 72250, 36500, 48600, 72500],
[38000, 74000, 36900, 49400, 73800],
[40000, 74900, 37450, 50200, 74900],
[41000, None, None, None, None],
[43000, None, None, None, None]]
exx = [0.0]
exx.append(74900 * 1.02) # exx[1]
exx.append(37450 * 1.02) # exx[2]
exx.append(50200 * 1.02) # exx[3]
exx.append(74900 * 1.02) # exx[4]
exx.append(0.0)
exx.append(exx[1] * 1.03) # exx[6]
exx.append(exx[2] * 1.03) # exx[7]
exx.append(exx[3] * 1.03) # exx[8]
exx.append(exx[4] * 1.03) # exx[9]
exp = [[36000, 72250, 36500, 48600, 72500],
[38000, 74000, 36900, 49400, 73800],
[40000, 74900, 37450, 50200, 74900],
[41000, exx[1], exx[2], exx[3], exx[4]],
[43000, exx[6], exx[7], exx[8], exx[9]]]
inflation_rates = [0.015, 0.02, 0.02, 0.03]
res = ParametersBase._expand_array(_II_brk2,
inflate=True,
inflation_rates=inflation_rates,
num_years=5)
assert_equal(res, exp)
user_mods = {2016: {'_II_brk2': _II_brk2}}
syr = 2013
pol = Policy(start_year=syr)
irates = pol.inflation_rates()
pol.implement_reform(user_mods)
pol.set_year(2020)
irates = pol.inflation_rates()
# The 2020 policy should be the combination of the user-defined
# value and CPI-inflated values from 2018
exp_2020 = [43000.] + [(1 + irates[2019 - syr]) *
(1 + irates[2018 - syr]) * i
for i in _II_brk2[2][1:]]
exp_2020 = np.array(exp_2020)
assert np.allclose(pol.II_brk2, exp_2020)