mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 22:14:28 +01:00
29 lines
910 B
Python
29 lines
910 B
Python
import os
|
|
from pathlib import Path
|
|
from typing import Any, Dict
|
|
import pytest
|
|
|
|
from agbenchmark.tests.basic_abilities.basic_challenge import BasicChallenge
|
|
|
|
|
|
class TestWriteFile(BasicChallenge):
|
|
"""Testing if LLM can write to a file"""
|
|
|
|
def get_file_path(self) -> str: # all tests must implement this method
|
|
return os.path.join(os.path.dirname(__file__), "w_file_data.json")
|
|
|
|
@pytest.mark.depends(name="basic_write_file")
|
|
def test_method(self, config: Dict[str, Any]) -> None:
|
|
self.setup_challenge(config)
|
|
|
|
workspace = Path(os.getcwd()) / config["workspace"]
|
|
files_contents = self.open_files(workspace, self.data.ground.files)
|
|
|
|
scores = []
|
|
for file_content in files_contents:
|
|
score = self.scoring(file_content, self.data.ground)
|
|
print("Your score is:", score)
|
|
scores.append(score)
|
|
|
|
assert 1 in scores
|