Ilya Ryzhenkov
146 lines
4.1 KiB
Kotlin
146 lines
4.1 KiB
Kotlin
package test.collections
|
|
|
|
import kotlin.test.*
|
|
|
|
import java.util.*
|
|
|
|
import org.junit.Test as test
|
|
|
|
class CollectionJVMTest {
|
|
|
|
test fun flatMap() {
|
|
val data = arrayListOf("", "foo", "bar", "x", "")
|
|
val characters = data.flatMap { it.toCharList() }
|
|
println("Got list of characters ${characters}")
|
|
assertEquals(7, characters.size())
|
|
val text = characters.joinToString("")
|
|
assertEquals("foobarx", text)
|
|
}
|
|
|
|
|
|
test fun filterIntoLinkedList() {
|
|
val data = arrayListOf("foo", "bar")
|
|
val foo = data.filterTo(linkedListOf<String>()) { it.startsWith("f") }
|
|
|
|
assertTrue {
|
|
foo.all { it.startsWith("f") }
|
|
}
|
|
assertEquals(1, foo.size())
|
|
assertEquals(linkedListOf("foo"), foo)
|
|
|
|
assertTrue {
|
|
foo is LinkedList<String>
|
|
}
|
|
}
|
|
|
|
test fun filterNotIntolinkedListOf() {
|
|
val data = arrayListOf("foo", "bar")
|
|
val foo = data.filterNotTo(linkedListOf<String>()) { it.startsWith("f") }
|
|
|
|
assertTrue {
|
|
foo.all { !it.startsWith("f") }
|
|
}
|
|
assertEquals(1, foo.size())
|
|
assertEquals(linkedListOf("bar"), foo)
|
|
|
|
assertTrue {
|
|
foo is LinkedList<String>
|
|
}
|
|
}
|
|
|
|
test fun filterNotNullIntoLinkedListOf() {
|
|
val data = arrayListOf(null, "foo", null, "bar")
|
|
val foo = data.filterNotNullTo(linkedListOf<String>())
|
|
|
|
assertEquals(2, foo.size())
|
|
assertEquals(linkedListOf("foo", "bar"), foo)
|
|
|
|
assertTrue {
|
|
foo is LinkedList<String>
|
|
}
|
|
}
|
|
|
|
test fun filterIntoSortedSet() {
|
|
val data = arrayListOf("foo", "bar")
|
|
val sorted = data.filterTo(sortedSetOf<String>()) { it.length == 3 }
|
|
assertEquals(2, sorted.size())
|
|
assertEquals(sortedSetOf("bar", "foo"), sorted)
|
|
assertTrue {
|
|
sorted is TreeSet<String>
|
|
}
|
|
}
|
|
|
|
test fun first() {
|
|
assertEquals(19, TreeSet(arrayListOf(90, 47, 19)).first())
|
|
}
|
|
|
|
test fun last() {
|
|
val data = arrayListOf("foo", "bar")
|
|
assertEquals("bar", data.last())
|
|
assertEquals(25, arrayListOf(15, 19, 20, 25).last())
|
|
assertEquals('a', linkedListOf('a').last())
|
|
}
|
|
|
|
test fun lastException() {
|
|
fails { linkedListOf<String>().last() }
|
|
}
|
|
|
|
test fun contains() {
|
|
assertTrue(linkedListOf(15, 19, 20).contains(15))
|
|
}
|
|
|
|
test fun toArray() {
|
|
val data = arrayListOf("foo", "bar")
|
|
val arr = data.toArray()
|
|
println("Got array ${arr}")
|
|
assertEquals(2, arr.size())
|
|
todo {
|
|
assertTrue {
|
|
arr is Array<String>
|
|
}
|
|
}
|
|
}
|
|
|
|
test fun takeReturnsFirstNElements() {
|
|
expect(setOf(1, 2)) { sortedSetOf(1, 2, 3, 4, 5).take(2).toSet() }
|
|
}
|
|
|
|
test fun filterIsInstanceList() {
|
|
val src: List<Any> = listOf(1,2,3.toDouble(), "abc", "cde")
|
|
|
|
val ints: List<Int> = src.filterIsInstance<Int>()
|
|
assertEquals(arrayListOf(1,2), ints)
|
|
|
|
val doubles: List<Double> = src.filterIsInstance<Double>()
|
|
assertEquals(arrayListOf(3.0), doubles)
|
|
|
|
val strings: List<String> = src.filterIsInstance<String>()
|
|
assertEquals(arrayListOf("abc", "cde"), strings)
|
|
|
|
val anys: List<Any> = src.filterIsInstance<Any>()
|
|
assertEquals(src.toList(), anys)
|
|
|
|
val chars: List<Char> = src.filterIsInstance<Char>()
|
|
assertEquals(0, chars.size())
|
|
}
|
|
|
|
test fun filterIsInstanceArray() {
|
|
val src: Array<Any> = array(1,2,3.toDouble(), "abc", "cde")
|
|
|
|
val ints: List<Int> = src.filterIsInstance<Int>()
|
|
assertEquals(arrayListOf(1,2), ints)
|
|
|
|
val doubles: List<Double> = src.filterIsInstance<Double>()
|
|
assertEquals(arrayListOf(3.0), doubles)
|
|
|
|
val strings: List<String> = src.filterIsInstance<String>()
|
|
assertEquals(arrayListOf("abc", "cde"), strings)
|
|
|
|
val anys: List<Any> = src.filterIsInstance<Any>()
|
|
assertEquals(src.toList(), anys)
|
|
|
|
val chars: List<Char> = src.filterIsInstance<Char>()
|
|
assertEquals(0, chars.size())
|
|
}
|
|
}
|