andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: update tests logic

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-07-04 17:18:32 -03:00
parent 2fb56f1f9f
commit 5247123e37
3 changed files with 6634 additions and 11584 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8420
tests/cassettes/test_increment_delegations_for_hierarchical_process.yaml
View File

File diff suppressed because it is too large Load Diff

9786
tests/cassettes/test_increment_tool_errors.yaml
View File

File diff suppressed because it is too large Load Diff

12
tests/task_test.py
View File

@@ -413,19 +413,17 @@ def test_increment_delegations_for_hierarchical_process():
agents=[scorer], agents=[scorer],
tasks=[task], tasks=[task],
process=Process.hierarchical, process=Process.hierarchical,
manager_llm=ChatOpenAI(model="gpt-4-0125-preview"), manager_llm=ChatOpenAI(model="gpt-4o"),
) )
with patch.object(Task, "increment_delegations") as increment_delegations: with patch.object(Task, "increment_delegations") as increment_delegations:
increment_delegations.return_value = None increment_delegations.return_value = None
crew.kickoff() crew.kickoff()
increment_delegations.assert_called_once increment_delegations.assert_called_once()
@pytest.mark.vcr(filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])
def test_increment_delegations_for_sequential_process(): def test_increment_delegations_for_sequential_process():
pass
manager = Agent( manager = Agent(
role="Manager", role="Manager",
goal="Coordinate scoring processes", goal="Coordinate scoring processes",
@@ -455,7 +453,7 @@ def test_increment_delegations_for_sequential_process():
with patch.object(Task, "increment_delegations") as increment_delegations: with patch.object(Task, "increment_delegations") as increment_delegations:
increment_delegations.return_value = None increment_delegations.return_value = None
crew.kickoff() crew.kickoff()
increment_delegations.assert_called_once increment_delegations.assert_called_once()
@pytest.mark.vcr(filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])
@@ -484,13 +482,13 @@ def test_increment_tool_errors():
agents=[scorer], agents=[scorer],
tasks=[task], tasks=[task],
process=Process.hierarchical, process=Process.hierarchical,
manager_llm=ChatOpenAI(model="gpt-4-0125-preview"), manager_llm=ChatOpenAI(model="gpt-4o"),
) )
with patch.object(Task, "increment_tools_errors") as increment_tools_errors: with patch.object(Task, "increment_tools_errors") as increment_tools_errors:
increment_tools_errors.return_value = None increment_tools_errors.return_value = None
crew.kickoff() crew.kickoff()
increment_tools_errors.assert_called_once increment_tools_errors.assert_called_once()
def test_task_definition_based_on_dict(): def test_task_definition_based_on_dict():