tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

test: updated tests to allow padding

Browse source
This commit is contained in:
Robin Meersman 2026-05-10 16:47:47 +02:00
parent 16c93ea9d5
commit 699f435cf5
4 changed files with 24 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_network_shapes.py
View file

@ -36,9 +36,9 @@ def test_centralized_forward_pass_with_padding():
)
global_state = obs_processor(amputated_obs)
# 40 + 40 + 20 = 100 dimensions
assert global_state.shape == (batch_size, 1, 100), (
f"Expected global state shape (2, 100), got {global_state.shape}"
# 40 + 40 + 20 + padding = 145 dimensions
assert global_state.shape == (batch_size, 1, 145), (
f"Expected global state shape (2, 1, 145), got {global_state.shape}"
)
# 4. Initialize dummy networks (40 actuators for the max morphology output)