def test_expand_xd_errors(): """ One of several _expand_?D tests. """ dct = dict() with pytest.raises(ValueError): Parameters._expand_1d(dct, inflate=False, inflation_rates=[], num_years=10) with pytest.raises(ValueError): Parameters._expand_2d(dct, inflate=False, inflation_rates=[], num_years=10)
def test_expand_xd_errors(): """ One of several _expand_?D tests. """ dct = dict() with pytest.raises(ValueError): Parameters._expand_1d(dct, inflate=False, inflation_rates=[], num_years=10) with pytest.raises(ValueError): Parameters._expand_2d(dct, inflate=False, inflation_rates=[], num_years=10)
def test_expand_1d_scalar(): """ One of several _expand_?D tests. """ yrs = 12 val = 10.0 exp = np.array([val * math.pow(1.02, i) for i in range(0, yrs)]) res = Parameters._expand_1d(np.array([val]), inflate=True, inflation_rates=[0.02] * yrs, num_years=yrs) assert np.allclose(exp, res, atol=0.01, rtol=0.0) res = Parameters._expand_1d(np.array([val]), inflate=True, inflation_rates=[0.02] * yrs, num_years=1) assert np.allclose(np.array([val]), res, atol=0.01, rtol=0.0)
def test_expand_1d_scalar(): """ One of several _expand_?D tests. """ yrs = 12 val = 10.0 exp = np.array([val * math.pow(1.02, i) for i in range(0, yrs)]) res = Parameters._expand_1d(np.array([val]), inflate=True, inflation_rates=[0.02] * yrs, num_years=yrs) assert np.allclose(exp, res, atol=0.01, rtol=0.0) res = Parameters._expand_1d(np.array([val]), inflate=True, inflation_rates=[0.02] * yrs, num_years=1) assert np.allclose(np.array([val]), res, atol=0.01, rtol=0.0)