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

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 # Verify CrewEvaluator interactions
crew_evaluator_mock.assert_has_calls( # We don't check the exact LLM object since it's created internally
[ assert len(crew_evaluator_mock.mock_calls) == 4
mock.call(crew, "gpt-4o-mini"), assert crew_evaluator_mock.mock_calls[1] == mock.call().set_iteration(1)
mock.call().set_iteration(1), assert crew_evaluator_mock.mock_calls[2] == mock.call().set_iteration(2)
mock.call().set_iteration(2), assert crew_evaluator_mock.mock_calls[3] == mock.call().print_crew_evaluation_result()
mock.call().print_crew_evaluation_result(),
]
)
@pytest.mark.vcr(filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])