adding simulation tests on consumption for ConsPrefShockModel

econ-ark · Aug 18, 2020 · 547608c · 547608c
1 parent fe8ebdb
commit 547608c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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
+        )