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

feat: Add task augmentation functions mirror, inverse, and identity to arc_1d.py

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-02-02 22:42:21 +01:00
parent 6ecd25c283
commit 017148d78d
2 changed files with 43 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/test_arc_1d.py
View file

@ -46,6 +46,24 @@ def test_all_arc_1d_tasks():
# Fixed move_pix value for testing
move_pix = 2
# Test task augmentation functions
base_task = task_move_n_pix(rng, size, move_pix, True)
assert base_task is not None
mirrored = task_mirror(base_task)
assert mirrored is not None
assert mirrored["input"] == list(reversed(base_task["input"]))
assert mirrored["output"] == list(reversed(base_task["output"]))
inversed = task_inverse(base_task)
assert inversed is not None
assert inversed["input"] == base_task["output"]
assert inversed["output"] == base_task["input"]
identical = task_identity(base_task)
assert identical is not None
assert identical == base_task
tasks = [
(task_move_n_pix, {"move_pix": move_pix, "solid": True}),
(task_move_n_pix_wrapped, {"move_pix": move_pix, "solid": True}),