test: add has_set_count field to PoemState

Co-Authored-By: Joe Moura <joao@crewai.com>
2026-01-08 15:48:29 +00:00 · 2025-01-19 08:30:38 +00:00
parent 4f4f229a0b
commit db626ec99d
1 changed files with 1 additions and 0 deletions
--- a/tests/test_flow_default_override.py
+++ b/tests/test_flow_default_override.py
@@ -15,6 +15,7 @@ class PoemState(FlowState):
    """Test state model with default values that should be overridden."""
    sentence_count: int = 1000  # Default that should be overridden
    poem: str = ""
+    has_set_count: bool = False  # Track whether we've set the count


 def test_default_value_override(tmp_path):