def test_infinite_horizon(self):
baseEx_inf = IndShockConsumerTypeFast(cycles=0, **self.base_params)
baseEx_inf.solve()
baseEx_inf.unpack_cFunc()
m1 = np.linspace(
1, baseEx_inf.solution[0].mNrmStE, 50
) # m1 defines the plot range on the left of target m value (e.g. m <= target m)
c_m1 = baseEx_inf.cFunc[0](m1)
self.assertAlmostEqual(c_m1[0], 0.8527887545025995)
self.assertAlmostEqual(c_m1[-1], 1.0036279936408656)
x1 = np.linspace(0, 25, 1000)
cfunc_m = baseEx_inf.cFunc[0](x1)
self.assertAlmostEqual(cfunc_m[500], 1.8902146173138235)
self.assertAlmostEqual(cfunc_m[700], 2.1591451850267176)
m = np.linspace(0.001, 8, 1000)
# Use the HARK method derivative to get the derivative of cFunc, and the values are just the MPC
MPC = baseEx_inf.cFunc[0].derivative(m)
self.assertAlmostEqual(MPC[500], 0.08415000641504392)
self.assertAlmostEqual(MPC[700], 0.07173144137912524)
def test_GICRawFails(self):
GICRaw_fail_dictionary = dict(self.base_params)
GICRaw_fail_dictionary["Rfree"] = 1.08
GICRaw_fail_dictionary["PermGroFac"] = [1.00]
GICRawFailExample = IndShockConsumerTypeFast(
cycles=0, # cycles=0 makes this an infinite horizon consumer
**GICRaw_fail_dictionary
)
GICRawFailExample.solve()
GICRawFailExample.unpack_cFunc()
m = np.linspace(0, 5, 1000)
c_m = GICRawFailExample.cFunc[0](m)
self.assertAlmostEqual(c_m[500], 0.7772637042393458)
self.assertAlmostEqual(c_m[700], 0.8392649061916746)
self.assertFalse(GICRawFailExample.conditions["GICRaw"])
def test_baseEx(self):
baseEx = IndShockConsumerTypeFast(**self.base_params)
baseEx.cycles = 100 # Make this type have a finite horizon (Set T = 100)
baseEx.solve()
baseEx.unpack_cFunc()
m = np.linspace(0, 9.5, 1000)
c_m = baseEx.cFunc[0](m)
c_t1 = baseEx.cFunc[-2](m)
c_t5 = baseEx.cFunc[-6](m)
c_t10 = baseEx.cFunc[-11](m)
self.assertAlmostEqual(c_m[500], 1.4008090582203356)
self.assertAlmostEqual(c_t1[500], 2.9227437159255216)
self.assertAlmostEqual(c_t5[500], 1.7350607327187664)
self.assertAlmostEqual(c_t10[500], 1.4991390649979213)
self.assertAlmostEqual(c_t10[600], 1.6101476268581576)
self.assertAlmostEqual(c_t10[700], 1.7196531041366991)