Test ACE layout

### What should be done?

We should add better testing for the ACE codegen, especially for functions and methods in the `layout` module.

### How should it be done?

We should add tests to compare the expected layout vs the generated one, for multiple scenarios e.g., w/o buses, w/o public inputs/tables ...

### When is this task done?

When we have the relevant tests

### Additional context

_No response_