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

minor formatting changes

Browse source
This commit is contained in:
Andreas Koepf 2025-02-17 18:20:18 +01:00
parent 8989bfca6c
commit 203394314e
2 changed files with 18 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/test_basic_arithmetic.py
View file

@ -1,5 +1,3 @@
from random import Random
import pytest
from reasoning_gym.arithmetic.basic_arithmetic import (
@ -66,6 +64,18 @@ def test_arithmetic_dataset_format_styles():
dataset = BasicArithmeticDataset(config)
assert all(item["question"].strip().endswith(".") for item in dataset)
config = BasicArithmeticDatasetConfig(
size=10,
seed=42,
format_style="natural",
min_terms=2,
max_terms=3, # Keep expressions simple for testing
min_digits=1,
max_digits=2,
)
dataset = BasicArithmeticDataset(config)
assert all(item["question"].strip().endswith(".") for item in dataset)
def test_arithmetic_dataset_iteration():
"""Test that iteration respects dataset size"""