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

feat: fix tests

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-08-08 20:56:25 -03:00
parent 9232ac3e3f
commit 45c16cfa6b
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/cli/cli_test.py
View File

@@ -15,18 +15,18 @@ def runner():
def test_train_default_iterations(train_crew, runner): def test_train_default_iterations(train_crew, runner):
result = runner.invoke(train) result = runner.invoke(train)
train_crew.assert_called_once_with(5) train_crew.assert_called_once_with(5, "trained_agents_data.pkl")
assert result.exit_code == 0 assert result.exit_code == 0
assert "Training the crew for 5 iterations" in result.output assert "Training the Crew for 5 iterations" in result.output
@mock.patch("crewai.cli.cli.train_crew") @mock.patch("crewai.cli.cli.train_crew")
def test_train_custom_iterations(train_crew, runner): def test_train_custom_iterations(train_crew, runner):
result = runner.invoke(train, ["--n_iterations", "10"]) result = runner.invoke(train, ["--n_iterations", "10"])
train_crew.assert_called_once_with(10) train_crew.assert_called_once_with(10, "trained_agents_data.pkl")
assert result.exit_code == 0 assert result.exit_code == 0
assert "Training the crew for 10 iterations" in result.output assert "Training the Crew for 10 iterations" in result.output
@mock.patch("crewai.cli.cli.train_crew") @mock.patch("crewai.cli.cli.train_crew")