[perf] Add single-module layout to the tests