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

lint

Browse source
This commit is contained in:
Andreas Koepf 2025-01-30 22:55:04 +01:00
parent 21c47db6c1
commit ebb88e6c6a
24 changed files with 1215 additions and 814 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/test_bf.py
View file

@ -2,6 +2,7 @@ import pytest
from reasoning_gym.code.bf import BFConfig, BFDataset
def test_bf():
"""Test basic properties and solution of generated items"""
@ -34,4 +35,4 @@ def test_bf():
config = BFConfig(seed=44, size=20, difficulty=3)
dataset = BFDataset(config)
for item in dataset:
assert dataset.score_answer(answer=item["answer"], entry=item) == 1.0
assert dataset.score_answer(answer=item["answer"], entry=item) == 1.0

14
tests/test_game_of_life.py
View file

@ -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)