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 unit test

Browse source
This commit is contained in:
abdulhakeem 2025-02-17 03:13:06 -06:00
parent 15140d06e7
commit 5f279be306
1 changed files with 1 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_basic_arithmetic.py
View file

@ -64,11 +64,7 @@ def test_arithmetic_dataset_format_styles():
max_digits=2, max_digits=2,
) )
dataset = BasicArithmeticDataset(config) dataset = BasicArithmeticDataset(config)
assert all(item["question"].endswith("=") for item in dataset) assert all(item["question"].strip().endswith(".") for item in dataset)
config.format_style = "natural"
dataset = BasicArithmeticDataset(config)
assert all("=" in item["question"] for item in dataset)
def test_arithmetic_dataset_iteration(): def test_arithmetic_dataset_iteration():