Skip to content

Commit 06508a5

Browse files
committed
test: Add test coverage for steady-state warmup step increment
Add an additional forward() call in test_warmup_step_counter to cover line 207 in _stch.py (the else branch that increments step during steady-state operation after warmup completes).
1 parent 850b370 commit 06508a5

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

tests/unit/aggregation/test_stch.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,9 +219,12 @@ def test_warmup_step_counter():
219219
weighting(gramian)
220220
assert weighting.step == 3
221221

222-
weighting(gramian)
222+
weighting(gramian) # First step after warmup: nadir_vector gets computed
223223
assert weighting.step == 4
224224

225+
weighting(gramian) # Steady-state: nadir_vector is already set
226+
assert weighting.step == 5
227+
225228

226229
def test_warmup_with_varying_gramians():
227230
"""Test warmup with different gramians to verify accumulation."""

0 commit comments

Comments
 (0)