Adjust spiral detection tests

openfisca · Feb 5, 2024 · d6cb459 · d6cb459
1 parent 76a1cb2
commit d6cb459
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/core/test_tracers.py b/tests/core/test_tracers.py
@@ -10,13 +10,15 @@
 from openfisca_country_template.variables.housing import HousingOccupancyStatus
 
 from openfisca_core.simulations import CycleError, Simulation, SpiralError
+from openfisca_core.periods import period
 from openfisca_core.tracers import (
     FullTracer,
     SimpleTracer,
     TraceNode,
     TracingParameterNodeAtInstant,
 )
 
+
 from .parameters_fancy_indexing.test_fancy_indexing import parameters
 
 
@@ -28,6 +30,7 @@ class StubSimulation(Simulation):
     def __init__(self):
         self.exception = None
         self.max_spiral_loops = 1
+        self.max_spiral_lookback_months = 24
 
     def _calculate(self, variable, period):
         if self.exception:
@@ -116,16 +119,17 @@ def test_cycle_error(tracer):
         simulation._check_for_cycle("a", 2017)
 
 
+
 @mark.parametrize("tracer", [SimpleTracer(), FullTracer()])
 def test_spiral_error(tracer):
     simulation = StubSimulation()
     simulation.tracer = tracer
-    tracer.record_calculation_start("a", 2017)
-    tracer.record_calculation_start("a", 2016)
-    tracer.record_calculation_start("a", 2015)
+    tracer.record_calculation_start("a", period(2017))
+    tracer.record_calculation_start("a", period(2016))
+    tracer.record_calculation_start("a", period(2015))
 
     with raises(SpiralError):
-        simulation._check_for_cycle("a", 2015)
+        simulation._check_for_cycle("a", period(2015))
 
 
 def test_full_tracer_one_calculation(tracer):