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

fix: update test assertions and sort imports

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-09 23:05:31 +00:00
parent dad121a692
commit e343017414
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/crew_test.py
View File

@@ -3335,14 +3335,11 @@ def test_crew_testing_function(copy_mock, crew_evaluator_mock):
)
# Verify CrewEvaluator interactions
crew_evaluator_mock.assert_has_calls(
[
mock.call(crew, "gpt-4o-mini"),
mock.call().set_iteration(1),
mock.call().set_iteration(2),
mock.call().print_crew_evaluation_result(),
]
)
# We don't check the exact LLM object since it's created internally
assert len(crew_evaluator_mock.mock_calls) == 4
assert crew_evaluator_mock.mock_calls[1] == mock.call().set_iteration(1)
assert crew_evaluator_mock.mock_calls[2] == mock.call().set_iteration(2)
assert crew_evaluator_mock.mock_calls[3] == mock.call().print_crew_evaluation_result()
@pytest.mark.vcr(filter_headers=["authorization"])