mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
parent
e0f8ef061d
commit
fc908d4cf4
3 changed files with 83 additions and 4 deletions
|
|
@ -2,7 +2,7 @@
|
|||
|
||||
import pytest
|
||||
|
||||
from reasoning_gym.algorithmic.caesar_cipher import CaesarCipherConfig, CaesarCipherDataset
|
||||
from reasoning_gym.algorithmic.caesar_cipher import CaesarCipherConfig, CaesarCipherCurriculum, CaesarCipherDataset
|
||||
|
||||
|
||||
def test_caesar_cipher_config_validation():
|
||||
|
|
@ -98,3 +98,48 @@ def test_caesar_cipher_iteration():
|
|||
|
||||
# Test multiple iterations yield same items
|
||||
assert items == list(dataset)
|
||||
|
||||
|
||||
def test_caesar_cipher_curriculum():
|
||||
curriculum = CaesarCipherCurriculum()
|
||||
base_value = {"size": 150, "seed": 1}
|
||||
|
||||
base_cfg: CaesarCipherConfig = curriculum.generate_configuration(base_value)
|
||||
assert base_cfg.seed == 1
|
||||
assert base_cfg.size == 150
|
||||
assert base_cfg.min_rotation == base_cfg.max_rotation == 5
|
||||
assert base_cfg.min_words == base_cfg.max_words == 5
|
||||
|
||||
curriculum.increment_attr_level("rotation")
|
||||
cfg = curriculum.generate_configuration(base_value)
|
||||
assert cfg.min_rotation == 5
|
||||
assert cfg.max_rotation == 10
|
||||
|
||||
curriculum.increment_attr_level("words")
|
||||
cfg = curriculum.generate_configuration(base_value)
|
||||
assert cfg.min_words == 5
|
||||
assert cfg.max_words == 10
|
||||
|
||||
curriculum.increment_attr_level("rotation")
|
||||
curriculum.increment_attr_level("words")
|
||||
cfg = curriculum.generate_configuration(base_value)
|
||||
assert cfg.min_rotation == 5
|
||||
assert cfg.max_rotation == 15
|
||||
assert cfg.min_words == 5
|
||||
assert cfg.max_words == 15
|
||||
|
||||
curriculum.increment_attr_level("rotation")
|
||||
curriculum.increment_attr_level("words")
|
||||
cfg = curriculum.generate_configuration(base_value)
|
||||
assert cfg.min_rotation == 5
|
||||
assert cfg.max_rotation == 25
|
||||
assert cfg.min_words == 5
|
||||
assert cfg.max_words == 25
|
||||
|
||||
curriculum.decrement_attr_level("rotation")
|
||||
curriculum.decrement_attr_level("words")
|
||||
cfg = curriculum.generate_configuration(base_value)
|
||||
assert cfg.min_rotation == 5
|
||||
assert cfg.max_rotation == 15
|
||||
assert cfg.min_words == 5
|
||||
assert cfg.max_words == 15
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue