Make test inputs floats, not ints

econ-ark · Aug 27, 2021 · fd6e6b2 · fd6e6b2
1 parent fd8d69a
commit fd6e6b2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/HARK/ConsumptionSaving/tests/test_ConsRiskyContribModel.py b/HARK/ConsumptionSaving/tests/test_ConsRiskyContribModel.py
@@ -52,13 +52,13 @@ def test_finite_cont_share(self):
         # Independent solver
         fin_cont_agent.solve()
         self.assertAlmostEqual(
-            fin_cont_agent.solution[0].stage_sols["Reb"].dfracFunc_Adj(3, 4), -0.87671241
+            fin_cont_agent.solution[0].stage_sols["Reb"].dfracFunc_Adj(3., 4.), -0.87671241
         )
         self.assertAlmostEqual(
-            fin_cont_agent.solution[0].stage_sols["Sha"].ShareFunc_Adj(5, 0.1), 0.14641409
+            fin_cont_agent.solution[0].stage_sols["Sha"].ShareFunc_Adj(5., 0.1), 0.14641409
         )
         self.assertAlmostEqual(
-            fin_cont_agent.solution[0].stage_sols["Cns"].cFunc(3, 4, 0.1), 2.4560881
+            fin_cont_agent.solution[0].stage_sols["Cns"].cFunc(3., 4., 0.1), 2.4560881
         )
 
         # General correlated solver
@@ -86,13 +86,13 @@ def test_finite_disc_share(self):
         fin_disc_agent.solve()
 
         self.assertAlmostEqual(
-            fin_disc_agent.solution[0].stage_sols["Reb"].dfracFunc_Adj(3, 4), -0.8767603
+            fin_disc_agent.solution[0].stage_sols["Reb"].dfracFunc_Adj(3., 4.), -0.8767603
         )
         self.assertAlmostEqual(
-            fin_disc_agent.solution[0].stage_sols["Sha"].ShareFunc_Adj(5, 0.1), 0.1
+            fin_disc_agent.solution[0].stage_sols["Sha"].ShareFunc_Adj(5., 0.1), 0.1
         )
         self.assertAlmostEqual(
-            fin_disc_agent.solution[0].stage_sols["Cns"].cFunc(3, 4, 0.1), 2.45608803
+            fin_disc_agent.solution[0].stage_sols["Cns"].cFunc(3., 4., 0.1), 2.45608803
         )
 
         # General correlated solver