def test_cyclical(self):
CyclicalExample = IndShockConsumerTypeFast(**CyclicalDict)
CyclicalExample.cycles = 0 # Make this consumer type have an infinite horizon
CyclicalExample.solve()
self.assertAlmostEqual(CyclicalExample.solution[3].cFunc(3).tolist(),
1.5958390056965004)
def test_lifecyle(self):
LifecycleExample = IndShockConsumerTypeFast(**LifecycleDict)
LifecycleExample.cycles = 1
LifecycleExample.solve()
self.assertEqual(len(LifecycleExample.solution), 11)
mMin = np.min([
LifecycleExample.solution[t].mNrmMin
for t in range(LifecycleExample.T_cycle)
])
self.assertAlmostEqual(LifecycleExample.solution[5].cFunc(3).tolist(),
2.129983771775666)
def test_infinite_horizon(self):
IndShockExample = IndShockConsumerTypeFast(**IdiosyncDict)
IndShockExample.cycles = 0 # Make this type have an infinite horizon
IndShockExample.solve()
self.assertAlmostEqual(IndShockExample.solution[0].mNrmStE, 1.5488165705077026)
self.assertAlmostEqual(
IndShockExample.solution[0].cFunc.functions[0].x_list[0], -0.25017509
)
IndShockExample.track_vars = ['aNrm', "mNrm", 'cNrm', 'pLvl']
IndShockExample.initialize_sim()
IndShockExample.simulate()
self.assertAlmostEqual(
IndShockExample.history["mNrm"][0][0], 1.0170176090252379
)
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)
def test_ConsIndShockSolverBasic(self):
LifecycleExample = IndShockConsumerTypeFast(**init_lifecycle)
LifecycleExample.cycles = 1
LifecycleExample.solve()
# test the solution_terminal
self.assertAlmostEqual(LifecycleExample.solution[-1].cFunc(2).tolist(), 2)
self.assertAlmostEqual(LifecycleExample.solution[9].cFunc(1), 0.79429538)
self.assertAlmostEqual(LifecycleExample.solution[8].cFunc(1), 0.79391692)
self.assertAlmostEqual(LifecycleExample.solution[7].cFunc(1), 0.79253095)
self.assertAlmostEqual(
LifecycleExample.solution[0].cFunc(1).tolist(), 0.7506184692092213
)
self.assertAlmostEqual(
LifecycleExample.solution[1].cFunc(1).tolist(), 0.7586358637239385
)
self.assertAlmostEqual(
LifecycleExample.solution[2].cFunc(1).tolist(), 0.7681247572911291
)
def test_ConsIndShockSolverBasic(self):
LifecycleExample = IndShockConsumerTypeFast(**init_lifecycle)
LifecycleExample.cycles = 1
LifecycleExample.solve()
# test the solution_terminal
self.assertAlmostEqual(LifecycleExample.solution[10].cFunc(2).tolist(), 2)
self.assertAlmostEqual(LifecycleExample.solution[9].cFunc(1), 0.97769632)
self.assertAlmostEqual(LifecycleExample.solution[8].cFunc(1), 0.96624445)
self.assertAlmostEqual(LifecycleExample.solution[7].cFunc(1), 0.95691449)
self.assertAlmostEqual(
LifecycleExample.solution[0].cFunc(1).tolist(), 0.87362789
)
self.assertAlmostEqual(
LifecycleExample.solution[1].cFunc(1).tolist(), 0.9081621
)
self.assertAlmostEqual(
LifecycleExample.solution[2].cFunc(1).tolist(), 0.9563899
)