andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 08:12:39 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #1597 from crewAIInc/tm-fix-crew-train-test

Browse Source 
Fix crew_train_success test
This commit is contained in:
Thiago Moretto
2024-11-13 10:52:49 -03:00
committed by GitHub
parent 7fb1289205 c57cbd8591
commit 4ca785eb15
1 changed files with 16 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tests/crew_test.py
View File

@@ -1779,26 +1779,22 @@ def test_crew_train_success(
] ]
) )
crew_training_handler.assert_has_calls( crew_training_handler.assert_any_call("training_data.pkl")
[ crew_training_handler().load.assert_called()
mock.call("training_data.pkl"),
mock.call().load(), crew_training_handler.assert_any_call("trained_agents_data.pkl")
mock.call("trained_agents_data.pkl"), crew_training_handler().load.assert_called()
mock.call().save_trained_data(
crew_training_handler().save_trained_data.assert_has_calls([
mock.call(
agent_id="Researcher", agent_id="Researcher",
trained_data=task_evaluator().evaluate_training_data().model_dump(), trained_data=task_evaluator().evaluate_training_data().model_dump(),
), ),
mock.call("trained_agents_data.pkl"), mock.call(
mock.call().save_trained_data(
agent_id="Senior Writer", agent_id="Senior Writer",
trained_data=task_evaluator().evaluate_training_data().model_dump(), trained_data=task_evaluator().evaluate_training_data().model_dump(),
),
mock.call(),
mock.call().load(),
mock.call(),
mock.call().load(),
]
) )
])
def test_crew_train_error(): def test_crew_train_error():