Adding unit tests for the layout builder