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

Apply automatic linting fixes to tests directory

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-12 13:31:07 +00:00
parent ad1ea46bbb
commit 46621113af
62 changed files with 1738 additions and 1821 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tests/test_flow_default_override.py
View File

@@ -6,12 +6,13 @@ from crewai.flow.persistence import persist
class PoemState(FlowState):
"""Test state model with default values that should be overridden."""
sentence_count: int = 1000 # Default that should be overridden
has_set_count: bool = False # Track whether we've set the count
poem_type: str = ""
def test_default_value_override():
def test_default_value_override() -> None:
"""Test that persisted state values override class defaults."""
@persist()
@@ -19,7 +20,7 @@ def test_default_value_override():
initial_state = PoemState
@start()
def set_sentence_count(self):
def set_sentence_count(self) -> None:
if self.state.has_set_count and self.state.sentence_count == 2:
self.state.sentence_count = 3
@@ -59,7 +60,7 @@ def test_default_value_override():
assert flow4.state.sentence_count == 1000 # Should load 1000, not 2
def test_multi_step_default_override():
def test_multi_step_default_override() -> None:
"""Test default value override with multiple start methods."""
@persist()
@@ -67,15 +68,13 @@ def test_multi_step_default_override():
initial_state = PoemState
@start()
def set_sentence_count(self):
print("Setting sentence count")
def set_sentence_count(self) -> None:
if not self.state.has_set_count:
self.state.sentence_count = 3
self.state.has_set_count = True
@listen(set_sentence_count)
def set_poem_type(self):
print("Setting poem type")
def set_poem_type(self) -> None:
if self.state.sentence_count == 3:
self.state.poem_type = "haiku"
elif self.state.sentence_count == 5:
@@ -84,8 +83,8 @@ def test_multi_step_default_override():
self.state.poem_type = "free_verse"
@listen(set_poem_type)
def finished(self):
print("finished")
def finished(self) -> None:
pass
# First run - should set both sentence count and poem type
flow1 = MultiStepPoemFlow()
@@ -98,7 +97,7 @@ def test_multi_step_default_override():
flow2 = MultiStepPoemFlow()
flow2.kickoff(inputs={
"id": original_uuid,
"sentence_count": 5
"sentence_count": 5,
})
assert flow2.state.sentence_count == 5
assert flow2.state.poem_type == "limerick"
@@ -106,7 +105,7 @@ def test_multi_step_default_override():
# Third run - new flow without persisted state should use defaults
flow3 = MultiStepPoemFlow()
flow3.kickoff(inputs={
"id": original_uuid
"id": original_uuid,
})
assert flow3.state.sentence_count == 5
assert flow3.state.poem_type == "limerick"
assert flow3.state.poem_type == "limerick"