add tests for flexWrap