diff --git a/HARK/ConsumptionSaving/tests/test_ConsPrefShockModel.py b/HARK/ConsumptionSaving/tests/test_ConsPrefShockModel.py index 329de95d7..125bc484d 100644 --- a/HARK/ConsumptionSaving/tests/test_ConsPrefShockModel.py +++ b/HARK/ConsumptionSaving/tests/test_ConsPrefShockModel.py @@ -39,6 +39,11 @@ def test_simulation(self): self.agent.initializeSim() self.agent.simulate() + self.assertAlmostEqual( + self.agent.history['cNrmNow'][0][5], + 1.21699840367737 + ) + class testKinkyPrefConsumerType(unittest.TestCase): def setUp(self): @@ -72,3 +77,8 @@ def test_simulation(self): self.agent.track_vars = ["cNrmNow", "PrefShkNow"] self.agent.initializeSim() self.agent.simulate() + + self.assertAlmostEqual( + self.agent.history['cNrmNow'][0][5], + 1.3282880084862532 + )