open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Correct 1-digit number test ranges in chain sum test

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 12:14:15 +01:00
parent 95fdca5a92
commit b92c278ef7
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/test_chain_sum.py
View file

@ -98,8 +98,16 @@ def test_chain_sum_number_ranges():
assert -999 <= num <= 999, f"Number {num} outside valid range for 3 digits"
else:
assert 100 <= num <= 999, f"Number {num} outside valid range for 3 digits"
# Test 1-digit numbers
config = ChainSumConfig(
min_terms=2,
max_terms=2,
min_digits=1,
max_digits=1,
size=50,
seed=42
)
dataset = ChainSum(config)
for i in range(len(dataset)):
item = dataset[i]