mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
lint
This commit is contained in:
parent
21c47db6c1
commit
ebb88e6c6a
24 changed files with 1215 additions and 814 deletions
|
|
@ -1,20 +1,14 @@
|
|||
import pytest
|
||||
|
||||
from reasoning_gym.games.game_of_life import GameOfLifeConfig, GameOfLifeConfigDataset
|
||||
from reasoning_gym.games.game_of_life import GameOfLifeConfig, GameOfLifeDataset
|
||||
|
||||
|
||||
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)
|
||||
config = GameOfLifeConfig(seed=42, size=1, grid_size_x=20, grid_size_y=20, filled_cells=10, simulation_steps=1)
|
||||
dataset = GameOfLifeDataset(config)
|
||||
|
||||
for item in dataset:
|
||||
assert isinstance(item, dict)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue