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

formatting

Browse source
This commit is contained in:
Andreas Koepf 2025-01-24 10:34:07 +01:00
parent 98988c8481
commit 20069b2a7d
37 changed files with 504 additions and 666 deletions
Download patch file Download diff file Expand all files Collapse all files

43
tests/test_base_conversion.py
View file

@ -1,10 +1,8 @@
"""Tests for base conversion task generation"""
import pytest
from reasoning_gym.algorithmic.base_conversion import (
BaseConversionConfig,
BaseConversionDataset,
)
from reasoning_gym.algorithmic.base_conversion import BaseConversionConfig, BaseConversionDataset
def test_base_conversion_config_validation():
@ -38,14 +36,7 @@ def test_base_conversion_dataset_deterministic():
def test_base_conversion_dataset_items():
"""Test basic properties of generated items"""
config = BaseConversionConfig(
min_base=2,
max_base=16,
min_value=0,
max_value=1000,
size=10,
seed=42
)
config = BaseConversionConfig(min_base=2, max_base=16, min_value=0, max_value=1000, size=10, seed=42)
dataset = BaseConversionDataset(config)
for i in range(len(dataset)):
@ -55,28 +46,28 @@ def test_base_conversion_dataset_items():
assert "question" in item
assert "answer" in item
assert "metadata" in item
# Check metadata
assert "decimal_value" in item["metadata"]
assert "source_base" in item["metadata"]
assert "target_base" in item["metadata"]
assert "source_repr" in item["metadata"]
assert "target_repr" in item["metadata"]
# Verify value range
assert config.min_value <= item["metadata"]["decimal_value"] <= config.max_value
# Verify base range
assert config.min_base <= item["metadata"]["source_base"] <= config.max_base
assert config.min_base <= item["metadata"]["target_base"] <= config.max_base
assert item["metadata"]["source_base"] != item["metadata"]["target_base"]
# Verify conversion correctness
decimal_value = item["metadata"]["decimal_value"]
target_base = item["metadata"]["target_base"]
expected = format(decimal_value, 'x' if target_base == 16 else 'b' if target_base == 2 else '').strip()
expected = format(decimal_value, "x" if target_base == 16 else "b" if target_base == 2 else "").strip()
if target_base not in (2, 16):
expected = format(decimal_value, f'{target_base}x').lower().strip()
expected = format(decimal_value, f"{target_base}x").lower().strip()
assert item["answer"] == expected
@ -100,24 +91,24 @@ def test_base_conversion_special_bases():
min_value=0,
max_value=255, # Use small range for predictable results
size=100,
seed=42
seed=42,
)
dataset = BaseConversionDataset(config)
binary_found = False
hex_found = False
for i in range(len(dataset)):
item = dataset[i]
if item["metadata"]["target_base"] == 2:
binary_found = True
# Verify binary format
assert all(c in '01' for c in item["answer"])
assert all(c in "01" for c in item["answer"])
elif item["metadata"]["target_base"] == 16:
hex_found = True
# Verify hex format
assert all(c in '0123456789abcdef' for c in item["answer"])
assert all(c in "0123456789abcdef" for c in item["answer"])
assert binary_found, "No binary conversion tasks generated"
assert hex_found, "No hexadecimal conversion tasks generated"
@ -130,10 +121,10 @@ def test_base_conversion_formatting():
min_value=10, # Ensure multi-digit numbers
max_value=1000,
size=10,
seed=42
seed=42,
)
dataset = BaseConversionDataset(config)
for i in range(len(dataset)):
item = dataset[i]
# Verify lowercase letters are used