mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-23 16:55:05 +00:00
test: Add test to verify numerators are <= denominators
This commit is contained in:
parent
e1e59d2a2f
commit
b598828e08
1 changed files with 25 additions and 0 deletions
|
|
@ -121,3 +121,28 @@ def test_fraction_iteration():
|
|||
first_items = list(dataset)
|
||||
second_items = list(dataset)
|
||||
assert first_items == second_items
|
||||
|
||||
|
||||
def test_fraction_numerator_smaller():
|
||||
"""Test that numerators are always smaller than denominators"""
|
||||
config = FractionSimplificationConfig(
|
||||
min_value=1,
|
||||
max_value=100,
|
||||
min_factor=2,
|
||||
max_factor=5,
|
||||
size=50,
|
||||
seed=42
|
||||
)
|
||||
dataset = FractionSimplificationDataset(config)
|
||||
|
||||
for i in range(len(dataset)):
|
||||
item = dataset[i]
|
||||
metadata = item["metadata"]
|
||||
|
||||
# Check original fraction
|
||||
assert metadata["numerator"] <= metadata["denominator"], \
|
||||
f"Original numerator {metadata['numerator']} should be <= denominator {metadata['denominator']}"
|
||||
|
||||
# Check simplified fraction
|
||||
assert metadata["simplified_numerator"] <= metadata["simplified_denominator"], \
|
||||
f"Simplified numerator {metadata['simplified_numerator']} should be <= denominator {metadata['simplified_denominator']}"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue