andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-22 22:58:13 +00:00
Code Issues Actions 4 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

10
tests/utilities/evaluators/test_task_evaluator.py
View File

@@ -8,7 +8,7 @@ from crewai.utilities.evaluators.task_evaluator import (
@patch("crewai.utilities.evaluators.task_evaluator.Converter")
def test_evaluate_training_data(converter_mock):
def test_evaluate_training_data(converter_mock) -> None:
training_data = {
"agent_id": {
"data1": {
@@ -21,7 +21,7 @@ def test_evaluate_training_data(converter_mock):
"human_feedback": "Human feedback 2",
"improved_output": "Improved output 2",
},
}
},
}
agent_id = "agent_id"
original_agent = MagicMock()
@@ -30,7 +30,7 @@ def test_evaluate_training_data(converter_mock):
suggestions=[
"The initial output was already good, having a detailed explanation. However, the improved output "
"gave similar information but in a more professional manner using better vocabulary. For future tasks, "
"try to implement more elaborate language and precise terminology from the beginning."
"try to implement more elaborate language and precise terminology from the beginning.",
],
quality=8.0,
final_summary="The agent responded well initially. However, the improved output showed that there is room "
@@ -39,7 +39,7 @@ def test_evaluate_training_data(converter_mock):
)
converter_mock.return_value.to_pydantic.return_value = function_return_value
result = TaskEvaluator(original_agent=original_agent).evaluate_training_data(
training_data, agent_id
training_data, agent_id,
)
assert result == function_return_value
@@ -61,5 +61,5 @@ def test_evaluate_training_data(converter_mock):
"following structure, with the following keys:\n{\n suggestions: List[str],\n quality: float,\n final_summary: str\n}",
),
mock.call().to_pydantic(),
]
],
)