File size: 665 Bytes
72a7241 1435892 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | import pytest
from src.pytorch_debug_env.bug_library import BUG_TEMPLATES
from src.pytorch_debug_env.scenario_generator import ScenarioGenerator
def test_generate_invalid_difficulty_raises():
generator = ScenarioGenerator(BUG_TEMPLATES)
with pytest.raises(ValueError):
generator.generate("unknown")
def test_generate_seed_reproducibility():
generator = ScenarioGenerator(BUG_TEMPLATES)
first = generator.generate("easy", seed=123)
second = generator.generate("easy", seed=123)
assert first.ground_truth == second.ground_truth
assert first.repo_files == second.repo_files
assert first.training_log == second.training_log
|