change TractableBufferstockModel to use functional simulation test, see

econ-ark#1105
sbenthall · May 31, 2022 · 1f67d65 · 1f67d65
1 parent ea73795
commit 1f67d65
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/HARK/ConsumptionSaving/tests/test_TractableBufferStockModel.py b/HARK/ConsumptionSaving/tests/test_TractableBufferStockModel.py
@@ -16,8 +16,8 @@ def test_simulation(self):
         self.tct.aLvlInitMean = 0.0  # Mean of log initial assets for new agents
         self.tct.aLvlInitStd = 1.0 # stdev of log initial assets for new agents
         self.tct.T_cycle = 1
-        self.tct.track_vars += ["mLvl"]
+        self.tct.track_vars += ["mLvl", "cLvlNow", "aLvl"]
         self.tct.initialize_sim()
         self.tct.simulate()
 
-        self.assertAlmostEqual(self.tct.history["mLvl"][15][0], 5.820630251772332)
+        self.assertAlmostEqual(self.tct.history["mLvl"][15][0] - self.tct.history["cLvlNow"][15][0], self.tct.history["aLvl"][15][0] )