Adds tests for flatMap and flatMapValues.