open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-24 17:05:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix unit tests, lower python dependency to 3.9

Browse source
This commit is contained in:
Andreas Koepf 2025-01-26 16:55:17 +01:00
parent ee67374aae
commit ad9f0d265c
11 changed files with 66 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/test_word_sorting.py
View file

@ -2,7 +2,7 @@
import pytest
from reasoning_gym.algorithmic.word_sorting import WordSortingConfig, WordSortingDataset, TextTransformation
from reasoning_gym.algorithmic.word_sorting import TextTransformation, WordSortingConfig, WordSortingDataset
def test_word_sorting_config_validation():
@ -38,7 +38,7 @@ def test_word_sorting_transformations():
"""Test different text transformations"""
seed = 42
size = 5
# Test LOWERCASE
config = WordSortingConfig(transformation=TextTransformation.LOWERCASE, seed=seed, size=size)
dataset = WordSortingDataset(config)
@ -64,14 +64,7 @@ def test_word_sorting_transformations():
def test_word_sorting_dataset_items():
"""Test basic properties of generated items"""
config = WordSortingConfig(
min_words=3,
max_words=6,
min_word_length=3,
max_word_length=8,
size=10,
seed=42
)
config = WordSortingConfig(min_words=3, max_words=6, min_word_length=3, max_word_length=8, size=10, seed=42)
dataset = WordSortingDataset(config)
for i in range(len(dataset)):