mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-23 00:44:22 +01:00
acfd966aa4
* feat: pass config to call_ai_functions in coimmands * feat: config for read_audio_from_file * feat: file operations cfg NOTE: we replaced the CFG in the command enable with TRUE b/c not sure how to handle this yet * feat: git command conversion * feat: google search * feat: image generation * feat: extract cfg from browser commands * feat: remove cfg from execute code commands * fix: file operation related tests * fix: linting * fix: tests for read_audio * fix: test error * feat: update cassettes * fix: linting * fix: test typechecking * fix: google_search errors if unexpected kw arg is passed * fix: pass config param to google search test * fix: agent commands were broken + cassettes * fix: agent test * feat: cassettes * feat: enable/disable logic for commands * fix: some commands threw errors * feat: fix tests * Add new cassettes * Add new cassettes * ci: trigger ci * Update autogpt/commands/execute_code.py Co-authored-by: Reinier van der Leer <github@pwuts.nl> * fix prompt * fix prompt + rebase * add config remove useless imports * put back CFG just for download file * lint * The signature should be mandatory in the decorator * black isort * fix: remove the CFG * fix: non typed arg * lint: type some args * lint: add types for libraries * Add new cassettes * fix: windows compatibility * fix: add config access to decorator * fix: remove twitter mention * DDGS search works at 3.0.2 version * ci: linting --------- Co-authored-by: Auto-GPT-Bot <github-bot@agpt.co> Co-authored-by: merwanehamadi <merwanehamadi@gmail.com> Co-authored-by: Reinier van der Leer <github@pwuts.nl> Co-authored-by: kinance <kinance@gmail.com>
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
# Date: 2023-5-13
|
|
# Author: Generated by GoCodeo.
|
|
import pytest
|
|
|
|
from autogpt.commands.analyze_code import analyze_code
|
|
from autogpt.config import Config
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_call_ai_function(mocker):
|
|
return mocker.patch("autogpt.commands.analyze_code.call_ai_function")
|
|
|
|
|
|
class TestAnalyzeCode:
|
|
def test_positive_analyze_code(self, mock_call_ai_function):
|
|
# Positive Test
|
|
mock_call_ai_function.return_value = ["Suggestion 1", "Suggestion 2"]
|
|
code = "def example_function():\n pass"
|
|
config = Config()
|
|
result = analyze_code(code, config)
|
|
assert result == ["Suggestion 1", "Suggestion 2"]
|
|
mock_call_ai_function.assert_called_once_with(
|
|
"def analyze_code(code: str) -> list[str]:",
|
|
[code],
|
|
"Analyzes the given code and returns a list of suggestions for improvements.",
|
|
config=config,
|
|
)
|
|
|
|
def test_negative_analyze_code(
|
|
self,
|
|
mock_call_ai_function,
|
|
config: Config,
|
|
):
|
|
# Negative Test
|
|
mock_call_ai_function.return_value = []
|
|
code = "def example_function():\n pass"
|
|
result = analyze_code(code, config)
|
|
assert result == []
|
|
mock_call_ai_function.assert_called_once_with(
|
|
"def analyze_code(code: str) -> list[str]:",
|
|
[code],
|
|
"Analyzes the given code and returns a list of suggestions for improvements.",
|
|
config=config,
|
|
)
|
|
|
|
def test_error_analyze_code(self, mock_call_ai_function, config: Config):
|
|
# Error Test
|
|
mock_call_ai_function.side_effect = Exception("Error occurred")
|
|
code = "def example_function():\n pass"
|
|
with pytest.raises(Exception):
|
|
result = analyze_code(code, config)
|
|
mock_call_ai_function.assert_called_once_with(
|
|
"def analyze_code(code: str) -> list[str]:",
|
|
[code],
|
|
"Analyzes the given code and returns a list of suggestions for improvements.",
|
|
config=config,
|
|
)
|
|
|
|
def test_edge_analyze_code_empty_code(
|
|
self,
|
|
mock_call_ai_function,
|
|
config: Config,
|
|
):
|
|
# Edge Test
|
|
mock_call_ai_function.return_value = ["Suggestion 1", "Suggestion 2"]
|
|
code = ""
|
|
result = analyze_code(code, config)
|
|
assert result == ["Suggestion 1", "Suggestion 2"]
|
|
mock_call_ai_function.assert_called_once_with(
|
|
"def analyze_code(code: str) -> list[str]:",
|
|
[code],
|
|
"Analyzes the given code and returns a list of suggestions for improvements.",
|
|
config=config,
|
|
)
|