add expected layout to tests