Add a flatMap test.