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

test: Add isalpha check for case transformation tests

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-26 16:53:21 +01:00
parent 2593699080
commit ee67374aae
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_word_sorting.py
View file

@ -44,14 +44,16 @@ def test_word_sorting_transformations():
dataset = WordSortingDataset(config)
for item in dataset:
for word in item["metadata"]["transformed_words"]:
assert word.islower()
if word.isalpha(): # Only test alphabetic strings
assert word.islower()
# Test UPPERCASE
config = WordSortingConfig(transformation=TextTransformation.UPPERCASE, seed=seed, size=size)
dataset = WordSortingDataset(config)
for item in dataset:
for word in item["metadata"]["transformed_words"]:
assert word.isupper()
if word.isalpha(): # Only test alphabetic strings
assert word.isupper()
# Test ORIGINAL
config = WordSortingConfig(transformation=TextTransformation.ORIGINAL, seed=seed, size=size)