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

18
tests/cli/test_crew_test.py
View File

@@ -7,7 +7,7 @@ from crewai.cli import evaluate_crew
@pytest.mark.parametrize(
"n_iterations,model",
("n_iterations", "model"),
[
(1, "gpt-4o"),
(5, "gpt-3.5-turbo"),
@@ -15,10 +15,10 @@ from crewai.cli import evaluate_crew
],
)
@mock.patch("crewai.cli.evaluate_crew.subprocess.run")
def test_crew_success(mock_subprocess_run, n_iterations, model):
def test_crew_success(mock_subprocess_run, n_iterations, model) -> None:
"""Test the crew function for successful execution."""
mock_subprocess_run.return_value = subprocess.CompletedProcess(
args=f"uv run test {n_iterations} {model}", returncode=0
args=f"uv run test {n_iterations} {model}", returncode=0,
)
result = evaluate_crew.evaluate_crew(n_iterations, model)
@@ -32,7 +32,7 @@ def test_crew_success(mock_subprocess_run, n_iterations, model):
@mock.patch("crewai.cli.evaluate_crew.click")
def test_test_crew_zero_iterations(click):
def test_test_crew_zero_iterations(click) -> None:
evaluate_crew.evaluate_crew(0, "gpt-4o")
click.echo.assert_called_once_with(
"An unexpected error occurred: The number of iterations must be a positive integer.",
@@ -41,7 +41,7 @@ def test_test_crew_zero_iterations(click):
@mock.patch("crewai.cli.evaluate_crew.click")
def test_test_crew_negative_iterations(click):
def test_test_crew_negative_iterations(click) -> None:
evaluate_crew.evaluate_crew(-2, "gpt-4o")
click.echo.assert_called_once_with(
"An unexpected error occurred: The number of iterations must be a positive integer.",
@@ -51,7 +51,7 @@ def test_test_crew_negative_iterations(click):
@mock.patch("crewai.cli.evaluate_crew.click")
@mock.patch("crewai.cli.evaluate_crew.subprocess.run")
def test_test_crew_called_process_error(mock_subprocess_run, click):
def test_test_crew_called_process_error(mock_subprocess_run, click) -> None:
n_iterations = 5
mock_subprocess_run.side_effect = subprocess.CalledProcessError(
returncode=1,
@@ -74,13 +74,13 @@ def test_test_crew_called_process_error(mock_subprocess_run, click):
err=True,
),
mock.call.echo("Error", err=True),
]
],
)
@mock.patch("crewai.cli.evaluate_crew.click")
@mock.patch("crewai.cli.evaluate_crew.subprocess.run")
def test_test_crew_unexpected_exception(mock_subprocess_run, click):
def test_test_crew_unexpected_exception(mock_subprocess_run, click) -> None:
# Arrange
n_iterations = 5
mock_subprocess_run.side_effect = Exception("Unexpected error")
@@ -93,5 +93,5 @@ def test_test_crew_unexpected_exception(mock_subprocess_run, click):
check=True,
)
click.echo.assert_called_once_with(
"An unexpected error occurred: Unexpected error", err=True
"An unexpected error occurred: Unexpected error", err=True,
)