mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-25 17:10:51 +00:00
Add gol curriculum (#354)
* Add gol curriculum * Add difficulty * Make levels of grid size of x and y be valid
This commit is contained in:
parent
fa2b04f4de
commit
adea7a255e
3 changed files with 128 additions and 4 deletions
|
|
@ -2,7 +2,7 @@ import json
|
|||
|
||||
import pytest
|
||||
|
||||
from reasoning_gym.algorithmic.game_of_life import GameOfLifeConfig, GameOfLifeDataset
|
||||
from reasoning_gym.algorithmic.game_of_life import GameOfLifeConfig, GameOfLifeCurriculum, GameOfLifeDataset
|
||||
|
||||
|
||||
def test_game_of_life_config_validation():
|
||||
|
|
@ -93,3 +93,70 @@ def test_game_of_life_iteration():
|
|||
first_items = list(dataset)
|
||||
second_items = list(dataset)
|
||||
assert first_items == second_items, "Multiple iterations should yield same items"
|
||||
|
||||
|
||||
def test_game_of_life_curriculum():
|
||||
"""Test the curriculum for complex arithmetic."""
|
||||
curriculum = GameOfLifeCurriculum()
|
||||
base_value = {"size": 150, "seed": 1}
|
||||
|
||||
base_cfg: GameOfLifeCurriculum = curriculum.generate_configuration(base_value)
|
||||
|
||||
assert base_cfg.seed == 1
|
||||
assert base_cfg.size == 150
|
||||
assert base_cfg.grid_size_x == 10
|
||||
assert base_cfg.grid_size_y == 10
|
||||
assert base_cfg.filled_cells <= base_cfg.grid_size_x * base_cfg.grid_size_y
|
||||
assert base_cfg.simulation_steps == 1
|
||||
|
||||
# Test and validate increase in levels
|
||||
curriculum.increment_attr_level("grid_size_x")
|
||||
curriculum.increment_attr_level("grid_size_y")
|
||||
curriculum.increment_attr_level("filled_cells_weights")
|
||||
curriculum.increment_attr_level("simulation_steps")
|
||||
|
||||
increased_cfg: GameOfLifeCurriculum = curriculum.generate_configuration(base_value)
|
||||
assert increased_cfg.grid_size_x == 100
|
||||
assert increased_cfg.grid_size_y == 100
|
||||
assert increased_cfg.filled_cells_weights == 0.2
|
||||
assert increased_cfg.filled_cells <= increased_cfg.grid_size_x * increased_cfg.grid_size_y
|
||||
assert increased_cfg.simulation_steps == 2
|
||||
|
||||
# Test and validate decrease in levels
|
||||
curriculum.decrement_attr_level("grid_size_x")
|
||||
curriculum.decrement_attr_level("grid_size_y")
|
||||
curriculum.decrement_attr_level("filled_cells_weights")
|
||||
curriculum.decrement_attr_level("simulation_steps")
|
||||
|
||||
decreased_cfg: GameOfLifeCurriculum = curriculum.generate_configuration(base_value)
|
||||
assert decreased_cfg.grid_size_x == 10
|
||||
assert decreased_cfg.grid_size_y == 10
|
||||
assert decreased_cfg.filled_cells_weights == 0.1
|
||||
assert decreased_cfg.filled_cells <= decreased_cfg.grid_size_x * decreased_cfg.grid_size_y
|
||||
assert decreased_cfg.simulation_steps == 1
|
||||
|
||||
# Test upper bound boundary condition
|
||||
for _ in range(10):
|
||||
curriculum.increment_attr_level("grid_size_x")
|
||||
curriculum.increment_attr_level("grid_size_y")
|
||||
curriculum.increment_attr_level("filled_cells_weights")
|
||||
curriculum.increment_attr_level("simulation_steps")
|
||||
upper_bound_cfg: GameOfLifeCurriculum = curriculum.generate_configuration(base_value)
|
||||
assert upper_bound_cfg.grid_size_x == 999
|
||||
assert upper_bound_cfg.grid_size_y == 999
|
||||
assert upper_bound_cfg.filled_cells_weights == 0.8
|
||||
assert upper_bound_cfg.filled_cells <= upper_bound_cfg.grid_size_x * upper_bound_cfg.grid_size_y
|
||||
assert upper_bound_cfg.simulation_steps == 10
|
||||
|
||||
# Test lower bound boundary condition
|
||||
for _ in range(10):
|
||||
curriculum.decrement_attr_level("grid_size_x")
|
||||
curriculum.decrement_attr_level("grid_size_y")
|
||||
curriculum.decrement_attr_level("filled_cells_weights")
|
||||
curriculum.decrement_attr_level("simulation_steps")
|
||||
lower_bound_cfg: GameOfLifeCurriculum = curriculum.generate_configuration(base_value)
|
||||
assert lower_bound_cfg.grid_size_x == 10
|
||||
assert lower_bound_cfg.grid_size_y == 10
|
||||
assert lower_bound_cfg.filled_cells_weights == 0.1
|
||||
assert lower_bound_cfg.filled_cells <= lower_bound_cfg.grid_size_x * lower_bound_cfg.grid_size_y
|
||||
assert lower_bound_cfg.simulation_steps == 1
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue