mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-29 19:04:28 +01:00
26 lines
687 B
Python
26 lines
687 B
Python
from typing import Any, Dict
|
|
|
|
import pytest
|
|
|
|
from agbenchmark.challenges.memory.memory import MemoryChallenge
|
|
|
|
|
|
class TestBasicMemory(MemoryChallenge):
|
|
"""The first memory challenge"""
|
|
|
|
@pytest.mark.depends(name="test_basic_memory")
|
|
def test_method(self, config: Dict[str, Any]) -> None:
|
|
self.setup_challenge(config)
|
|
|
|
files_contents = self.get_artifacts_out(
|
|
config["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
|