functional test for PerfectForesight simulation, econ-ark#1105

sbenthall · May 31, 2022 · 8953c1b · 8953c1b
1 parent 1f67d65
commit 8953c1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/HARK/ConsumptionSaving/tests/test_PerfForesightConsumerType.py b/HARK/ConsumptionSaving/tests/test_PerfForesightConsumerType.py
@@ -65,13 +65,13 @@ def test_simulation(self):
         )  # This implicitly uses the assign_parameters method of AgentType
 
         # Create PFexample object
-        self.agent_infinite.track_vars = ["mNrm"]
+        self.agent_infinite.track_vars = ["bNrm", "mNrm", "TranShk"]
         self.agent_infinite.initialize_sim()
         self.agent_infinite.simulate()
 
         self.assertAlmostEqual(
             np.mean(self.agent_infinite.history["mNrm"], axis=1)[40],
-            -23.008063500363942,
+            np.mean(self.agent_infinite.history["bNrm"], axis=1)[40] + np.mean(self.agent_infinite.history["TranShk"], axis=1)[40]
         )
 
         self.assertAlmostEqual(