Merge pull request #802 from sbenthall/sb-pref-tests

adding simulation tests on consumption for ConsPrefShockModel
econ-ark · Aug 19, 2020 · 64a2b04 · 64a2b04
2 parents fe8ebdb + 547608c
commit 64a2b04
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
+        )