Added tests for layout generation