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:
Adefioye 2025-03-13 15:09:09 -05:00 committed by GitHub
parent fa2b04f4de
commit adea7a255e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 128 additions and 4 deletions

View file

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