update unit tests

open-spaced-repetition · Dec 9, 2024 · 69d772b · 69d772b
1 parent c4074e8
commit 69d772b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/model_test.py b/tests/model_test.py
@@ -65,7 +65,7 @@ def test_forward(self):
         difficulty = state[:, 1]
         assert torch.allclose(
             stability,
-            torch.tensor([0.2619, 1.7073, 5.8691, 25.0123, 0.3403, 2.1482]),
+            torch.tensor([0.2619, 1.7074, 5.8691, 25.0124, 0.2859, 2.1482]),
             atol=1e-4,
         )
         assert torch.allclose(

diff --git a/tests/simulator_test.py b/tests/simulator_test.py
@@ -11,7 +11,7 @@ def test_simulate(self):
             cost_per_day,
             revlogs,
         ) = simulate(w=DEFAULT_PARAMETER, request_retention=0.9)
-        assert memorized_cnt_per_day[-1] == 5875.025236206539
+        assert memorized_cnt_per_day[-1] == 5880.482440745369
 
     def test_optimal_retention(self):
         default_params = {