open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-23 16:55:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add rule composition support to PatternRule

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 14:04:36 +01:00
parent 3381669f98
commit a2686341a3
2 changed files with 20 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_sequences.py
View file

@ -23,11 +23,17 @@ def test_pattern_rule():
# Test simple addition
rule = PatternRule([Operation.ADD], [2])
assert rule.apply([1, 3], 1) == 5
# Test composition
rule = PatternRule([Operation.DOUBLE, Operation.ADD], [0, 3])
assert rule.apply([1, 4], 1) == 11 # (4 * 2) + 3
# Test rule composition
rule1 = PatternRule([Operation.DOUBLE], [0]) # Double the number
rule2 = PatternRule([Operation.ADD], [3]) # Add 3
composed = PatternRule.compose([rule1, rule2])
assert composed.apply([1, 4], 1) == 11 # (4 * 2) + 3
def test_sequence_dataset_deterministic():
"""Test that dataset generates same items with same seed"""