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

fix: Update test_arc_1d.py to handle task function argument order

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-02-02 16:42:46 +01:00
parent d56e8c3a03
commit 2d3012d5ae
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/test_arc_1d.py
View file

@ -20,14 +20,17 @@ def test_all_arc_1d_tasks():
size = 20 # Reasonable size for testing
# Test all task functions
# Fixed move_pix value for testing
move_pix = 2
tasks = [
(task_move_n_pix, {"solid": True}),
(task_move_n_pix_wrapped, {"solid": True}),
(lambda s, r, **k: task_move_n_pix(s, move_pix, **k, r=r), {"solid": True}),
(lambda s, r, **k: task_move_n_pix_wrapped(s, move_pix, **k, r=r), {"solid": True}),
(task_gravity, {}),
(task_gravity_counting, {}),
(task_gravity_antigravity, {}),
(task_block_touch_dot, {}),
(task_block_touch_dot_n_pix, {}),
(lambda s, r, **k: task_block_touch_dot_n_pix(s, move_pix, r), {}),
(task_block_scale_to_dot, {}),
(task_two_points_and_fill, {}),
(task_reflect_block_with_border_pixel, {}),