Add tests for assert and layout modules