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

fix: Remove duplicate 1-digit number test section in test_chain_sum.py

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-23 12:16:09 +01:00
parent b92c278ef7
commit 4e7d69e39f
1 changed files with 0 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/test_chain_sum.py
View file

@ -74,30 +74,6 @@ def test_chain_sum_number_ranges():
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]
expression = item["metadata"]["expression"]
# Extract numbers from expression
numbers = [int(n) for n in expression.split() if n.isdigit()]
# Verify each number is in the correct range
for num in numbers:
if config.allow_negation:
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(