Franck Rasolo
69 lines
2.0 KiB
Kotlin
69 lines
2.0 KiB
Kotlin
package test.sequences
|
|
|
|
import kotlin.sequences.grouped
|
|
import kotlin.sequences.sequence
|
|
import kotlin.sequences.sliding
|
|
import kotlin.sequences.times
|
|
import kotlin.util.arrayList
|
|
|
|
import kotlin.test.assertEquals
|
|
import org.junit.Test
|
|
|
|
class SequencesTest {
|
|
|
|
Test fun timesComputesTheCartesianSquareOfAnIterable() {
|
|
val actual = (1..3) * (1..3)
|
|
val expected = arrayList(
|
|
#(1, 1), #(1, 2), #(1, 3),
|
|
#(2, 1), #(2, 2), #(2, 3),
|
|
#(3, 1), #(3, 2), #(3, 3)
|
|
)
|
|
assertEquals(expected, actual.toList())
|
|
}
|
|
|
|
Test fun timesComputesTheCartesianProductOfTwoRanges() {
|
|
val actual = (1..3) * (4..7)
|
|
val expected = arrayList(
|
|
#(1, 4), #(1, 5), #(1, 6), #(1, 7),
|
|
#(2, 4), #(2, 5), #(2, 6), #(2, 7),
|
|
#(3, 4), #(3, 5), #(3, 6), #(3, 7)
|
|
)
|
|
assertEquals(expected, actual.toList())
|
|
}
|
|
|
|
Test fun timesComputesTheCartesianProductOfTwoIterables() {
|
|
val actual = sequence(1, 3, 5, 7) * (4..6)
|
|
val expected = arrayList(
|
|
#(1, 4), #(1, 5), #(1, 6),
|
|
#(3, 4), #(3, 5), #(3, 6),
|
|
#(5, 4), #(5, 5), #(5, 6),
|
|
#(7, 4), #(7, 5), #(7, 6)
|
|
)
|
|
assertEquals(expected, actual.toList())
|
|
}
|
|
|
|
Test fun groupedReturnsFixedSizeGroupsOfStrings() {
|
|
val digits = """
|
|
62229893423380308135336276614282806444486645238749
|
|
30358907296290491560440772390713810515859307960866
|
|
70172427121883998797908792274921901699720888093776
|
|
65727333001053367881220235421809751254540594752243
|
|
""".trim().replaceAll("\\n", "")
|
|
|
|
val actual = digits.grouped(50).toList().get(2)
|
|
assertEquals("70172427121883998797908792274921901699720888093776", actual)
|
|
}
|
|
|
|
Test fun slidingOverALongStringOfDigits() {
|
|
val digits = """
|
|
62229893423380308135336276614282806444486645238749
|
|
30358907296290491560440772390713810515859307960866
|
|
70172427121883998797908792274921901699720888093776
|
|
65727333001053367881220235421809751254540594752243
|
|
""".trim().replaceAll("\\n", "")
|
|
|
|
val actual = digits.sliding(3).filter { Integer.parseInt(it) > 990 }
|
|
assertEquals(arrayList("998", "997"), actual.toList())
|
|
}
|
|
}
|