mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-06 15:54:22 +01:00
79 lines
2.0 KiB
Python
79 lines
2.0 KiB
Python
import glob
|
|
import importlib
|
|
import json
|
|
import os
|
|
import types
|
|
from pathlib import Path
|
|
from typing import Any, Dict
|
|
|
|
import pytest
|
|
from dotenv import load_dotenv
|
|
|
|
from agbenchmark.challenge import Challenge
|
|
|
|
load_dotenv()
|
|
|
|
IMPROVE = os.getenv("IMPROVE", "False")
|
|
|
|
|
|
json_files = glob.glob("agbenchmark/challenges/**/data.json", recursive=True)
|
|
|
|
|
|
def get_test_path(json_file: str) -> str:
|
|
abs_location = os.path.dirname(os.path.abspath(json_file))
|
|
|
|
path = Path(abs_location)
|
|
|
|
# Find the index of "agbenchmark" in the path parts
|
|
try:
|
|
agbenchmark_index = path.parts.index("agbenchmark")
|
|
except ValueError:
|
|
raise ValueError("Invalid challenge location.")
|
|
|
|
# Create the path from "agbenchmark" onwards
|
|
challenge_location = Path(*path.parts[agbenchmark_index:])
|
|
|
|
return str(challenge_location)
|
|
|
|
|
|
def generate_tests() -> None:
|
|
print("Generating tests...")
|
|
# Dynamic class creation
|
|
for json_file in json_files:
|
|
with open(json_file, "r") as f:
|
|
data = json.load(f)
|
|
|
|
class_name = data.get("name", "")
|
|
|
|
challenge_location = get_test_path(json_file)
|
|
|
|
# Define test class dynamically
|
|
challenge_class = types.new_class(class_name, (Challenge,))
|
|
|
|
setattr(challenge_class, "CHALLENGE_LOCATION", challenge_location)
|
|
|
|
# Define test method within the dynamically created class
|
|
def test_method(self, config: Dict[str, Any]) -> None: # type: ignore
|
|
self.setup_challenge(config)
|
|
|
|
scores = self.get_scores(config)
|
|
assert 1 in scores
|
|
|
|
# Parametrize the method here
|
|
test_method = pytest.mark.parametrize(
|
|
"challenge_data",
|
|
[data],
|
|
indirect=True,
|
|
)(test_method)
|
|
|
|
setattr(challenge_class, "test_method", test_method)
|
|
|
|
# Attach the new class to a module so it can be discovered by pytest
|
|
module = importlib.import_module(__name__)
|
|
setattr(module, class_name, challenge_class)
|
|
|
|
print(f"Generated test for {class_name}.")
|
|
|
|
|
|
generate_tests()
|