open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: Add size parameter and improve test case for QuantumLockDataset

Browse source
This commit is contained in:
Andreas Koepf 2025-01-30 01:21:13 +01:00 • committed by Andreas Koepf (aider)
parent 8a5cecb939
commit 01dfae6fe7
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/test_quantum_lock.py
View file

@ -6,6 +6,7 @@ def test_quantumlock_items():
"""Test basic properties and solution of generated items"""
config = QuantumLockConfig(
difficulty=10,
size=25,
)
dataset = QuantumLockDataset(config)
@ -19,5 +20,5 @@ def test_quantumlock_items():
assert "solution_path" in item["metadata"]
assert "difficulty" in item["metadata"]
assert dataset.score_answer(answer=item['metadata']['solution_path'], entry=item) == 1.0
assert dataset.score_answer(answer=item["metadata"]["solution_path"], entry=item) == 1.0
assert dataset.score_answer(answer=None, entry=item) == 0.0