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 deterministic test for RubiksCubeDataset __get_item__()

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-30 00:42:20 +01:00
parent 2653a28903
commit 08ee435d47
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/test_rubiks_cube.py
View file

@ -15,6 +15,16 @@ def test_rubikscube_config_validation():
config.validate()
def test_rubikscube_deterministic():
"""Test that dataset generates same items with same seed"""
config = RubiksCubeConfig(seed=42, size=15)
dataset1 = RubiksCubeDataset(config)
dataset2 = RubiksCubeDataset(config)
for i in range(15): # Only check first 15 entries for speed
assert dataset1[i] == dataset2[i]
def test_rubikscube_items():
"""Test basic properties and solution of generated items"""
config = RubiksCubeConfig(