Provide flatMapIndexed operation

- similar to flatMap, but transform function takes index and element

#KT-36894
This commit is contained in:
Ilya Gorbunov
2020-06-19 16:16:55 +03:00
parent db93462bcf
commit 130987fa1e
20 changed files with 1216 additions and 74 deletions
@@ -592,6 +592,14 @@ public class SequenceTest {
assertEquals(expected, result2.toList())
}
@Test fun flatMapIndexed() {
val result1 = sequenceOf(1, 2).flatMapIndexed { index, v -> (0..v + index).asSequence() }
val result2 = sequenceOf(1, 2).flatMapIndexed { index, v -> 0..v + index }
val expected = listOf(0, 1, 0, 1, 2, 3)
assertEquals(expected, result1.toList())
assertEquals(expected, result2.toList())
}
@Test fun flatten() {
val expected = listOf(0, 1, 0, 1, 2)