mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-28 17:29:39 +00:00
Merge pull request #30 from Miserlou/miserlou/gol
Add Conway's Game of Life Simulations
This commit is contained in:
commit
21c47db6c1
4 changed files with 134 additions and 0 deletions
33
tests/test_game_of_life.py
Normal file
33
tests/test_game_of_life.py
Normal file
|
|
@ -0,0 +1,33 @@
|
|||
import pytest
|
||||
|
||||
from reasoning_gym.games.game_of_life import GameOfLifeConfig, GameOfLifeConfigDataset
|
||||
|
||||
def test_game_of_life():
|
||||
"""Test basic properties and solution of generated items"""
|
||||
|
||||
# Easy
|
||||
config = GameOfLifeConfig(
|
||||
seed=42,
|
||||
size=1,
|
||||
grid_size_x=20,
|
||||
grid_size_y=20,
|
||||
filled_cells=10,
|
||||
simulation_steps=1
|
||||
)
|
||||
dataset = GameOfLifeConfigDataset(config)
|
||||
|
||||
for item in dataset:
|
||||
assert isinstance(item, dict)
|
||||
assert "question" in item
|
||||
assert "answer" in item
|
||||
assert "metadata" in item
|
||||
|
||||
# # Check metadata contains required fields
|
||||
assert "grid_size_x" in item["metadata"]
|
||||
assert "grid_size_y" in item["metadata"]
|
||||
assert "filled_cells" in item["metadata"]
|
||||
assert "simulation_steps" in item["metadata"]
|
||||
|
||||
# # Test the scoring
|
||||
assert dataset.score_answer(answer=item["answer"], entry=item) == 1.0
|
||||
assert dataset.score_answer(answer=None, entry=item) == 0.0
|
||||
Loading…
Add table
Add a link
Reference in a new issue