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

refactor: Update test cases to use 'solutions' instead of 'solution' in metadata

Browse source
This commit is contained in:
Andreas Koepf 2025-02-02 22:15:47 +01:00 • committed by Andreas Koepf (aider)
parent 5889cac5ab
commit 5282d9db31
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/test_n_queens.py
View file

@ -49,17 +49,17 @@ def test_nqueens_dataset_items():
# Check metadata
assert "puzzle" in item["metadata"]
assert "solution" in item["metadata"]
assert "solutions" in item["metadata"]
assert "num_removed" in item["metadata"]
puzzle = item["metadata"]["puzzle"]
solution = item["metadata"]["solution"]
solutions = item["metadata"]["solutions"]
num_removed = item["metadata"]["num_removed"]
# Verify board dimensions
assert len(puzzle) == 8
assert all(len(row) == 8 for row in puzzle)
for board in solution:
for board in solutions:
assert len(board) == 8
assert all(len(row) == 8 for row in board)
@ -69,7 +69,7 @@ def test_nqueens_dataset_items():
assert removed_count == num_removed
# Verify solution validity
for board in solution:
for board in solutions:
assert is_valid_solution(board)
# Verify puzzle matches solution where filled
@ -98,7 +98,7 @@ def test_nqueens_board_generation():
for i in range(len(dataset)):
item = dataset[i]
for board in item["metadata"]["solution"]:
for board in item["metadata"]["solutions"]:
assert is_valid_solution(board)