From 4f4f229a0b0516dab425d501053f7ef55273bf39 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sun, 19 Jan 2025 08:28:58 +0000 Subject: [PATCH] test: improve state restoration verification with has_set_count flag Co-Authored-By: Joe Moura --- tests/test_flow_default_override.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_flow_default_override.py b/tests/test_flow_default_override.py index 495a3edf9..099fc5bee 100644 --- a/tests/test_flow_default_override.py +++ b/tests/test_flow_default_override.py @@ -28,9 +28,10 @@ def test_default_value_override(tmp_path): @start() def set_sentence_count(self): - # First run will set a custom value - if not self.state.sentence_count or self.state.sentence_count == 1000: + # Only set sentence_count on first run, not when loading from persistence + if not self.state.has_set_count: self.state.sentence_count = 2 + self.state.has_set_count = True # First run - should set sentence_count to 2 # First run - should set sentence_count to 2