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

post merge lint

Browse source
This commit is contained in:
Andreas Koepf 2025-02-02 10:04:18 +01:00
parent 02cfa9556a
commit f396d3df60
6 changed files with 197 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/test_aiw.py
View file

@ -14,8 +14,7 @@ def test_aiw_config_validation():
config.validate()
with pytest.raises(AssertionError):
config = AliceInWonderlandConfig(
female_names=["Mary", "Jane"]) # No Alice
config = AliceInWonderlandConfig(female_names=["Mary", "Jane"]) # No Alice
config.validate()
with pytest.raises(AssertionError):
@ -56,8 +55,7 @@ def test_aiw_items():
# Verify question task type characteristics
task_type = item["metadata"]["task_type"]
if task_type == TaskType.SIBLINGS.value:
assert any(phrase in item["question"]
for phrase in ["brothers", "sisters"])
assert any(phrase in item["question"] for phrase in ["brothers", "sisters"])
elif task_type == TaskType.FRIENDS.value:
assert "friends" in item["question"]
elif task_type == TaskType.COLLEAGUES:
@ -95,5 +93,4 @@ def test_aiw_random_ranges():
numbers = [int(n) for n in question.split() if n.isdigit()]
# Check all numbers are in reasonable range (1-6 as per implementation)
assert all(
1 <= n <= 12 for n in numbers), f"Numbers out of range: {numbers}"
assert all(1 <= n <= 12 for n in numbers), f"Numbers out of range: {numbers}"