andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

test: improve state restoration verification with has_set_count flag

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-01-19 08:28:58 +00:00
parent 44b92ef12f
commit 4f4f229a0b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_flow_default_override.py
View File

@@ -28,9 +28,10 @@ def test_default_value_override(tmp_path):
@start() @start()
def set_sentence_count(self): def set_sentence_count(self):
# First run will set a custom value # Only set sentence_count on first run, not when loading from persistence
if not self.state.sentence_count or self.state.sentence_count == 1000: if not self.state.has_set_count:
self.state.sentence_count = 2 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
# First run - should set sentence_count to 2 # First run - should set sentence_count to 2