Adequate tests for the flatmap exercise