[Layout] Add tests