Andrey Breslav
125 lines
3.6 KiB
Kotlin
125 lines
3.6 KiB
Kotlin
package test.collections
|
|
|
|
import kotlin.test.*
|
|
|
|
import java.util.*
|
|
|
|
import org.junit.Test as test
|
|
|
|
class CollectionJVMTest {
|
|
|
|
test fun flatMap() {
|
|
val data = arrayList("", "foo", "bar", "x", "")
|
|
val characters = data.flatMap<String,Char>{ it.toCharList() }
|
|
println("Got list of characters ${characters}")
|
|
assertEquals(7, characters.size())
|
|
val text = characters.makeString("")
|
|
assertEquals("foobarx", text)
|
|
}
|
|
|
|
|
|
// TODO would be nice to avoid the <String>
|
|
test fun filterIntoLinkedList() {
|
|
val data = arrayList("foo", "bar")
|
|
val foo = data.filterTo(linkedList<String>()){it.startsWith("f")}
|
|
|
|
assertTrue {
|
|
foo.all{it.startsWith("f")}
|
|
}
|
|
assertEquals(1, foo.size)
|
|
assertEquals(linkedList("foo"), foo)
|
|
|
|
assertTrue {
|
|
foo is LinkedList<String>
|
|
}
|
|
}
|
|
|
|
// TODO would be nice to avoid the <String>
|
|
test fun filterNotIntoLinkedList() {
|
|
val data = arrayList("foo", "bar")
|
|
val foo = data.filterNotTo(linkedList<String>()){it.startsWith("f")}
|
|
|
|
assertTrue {
|
|
foo.all{!it.startsWith("f")}
|
|
}
|
|
assertEquals(1, foo.size)
|
|
assertEquals(linkedList("bar"), foo)
|
|
|
|
assertTrue {
|
|
foo is LinkedList<String>
|
|
}
|
|
}
|
|
|
|
// TODO would be nice to avoid the <String>
|
|
test fun filterNotNullIntoLinkedList() {
|
|
val data = arrayList(null, "foo", null, "bar")
|
|
val foo = data.filterNotNullTo(linkedList<String>())
|
|
|
|
assertEquals(2, foo.size)
|
|
assertEquals(linkedList("foo", "bar"), foo)
|
|
|
|
assertTrue {
|
|
foo is LinkedList<String>
|
|
}
|
|
}
|
|
|
|
|
|
// TODO would be nice to avoid the <String>
|
|
test fun filterIntoSortedSet() {
|
|
val data = arrayList("foo", "bar")
|
|
val sorted = data.filterTo(sortedSet<String>()){it.length == 3}
|
|
assertEquals(2, sorted.size)
|
|
assertEquals(sortedSet("bar", "foo"), sorted)
|
|
assertTrue {
|
|
sorted is TreeSet<String>
|
|
}
|
|
}
|
|
|
|
//todo after KT-1873 the name might be returned to 'last'
|
|
test fun lastElement() {
|
|
val data = arrayList("foo", "bar")
|
|
assertEquals("bar", data.last())
|
|
assertEquals(25, arrayList(15, 19, 20, 25).last())
|
|
assertEquals('a', linkedList('a').last())
|
|
}
|
|
|
|
test fun lastException() {
|
|
fails { linkedList<String>().last() }
|
|
}
|
|
|
|
test fun contains() {
|
|
assertTrue(linkedList(15, 19, 20).contains(15))
|
|
}
|
|
|
|
test fun sortBy() {
|
|
expect(arrayList("two" to 2, "three" to 3)) {
|
|
arrayList("three" to 3, "two" to 2).sortBy { it.second }
|
|
}
|
|
expect(arrayList("three" to 3, "two" to 2)) {
|
|
arrayList("three" to 3, "two" to 2).sortBy { it.first }
|
|
}
|
|
expect(arrayList("two" to 2, "three" to 3)) {
|
|
arrayList("three" to 3, "two" to 2).sortBy { it.first.length }
|
|
}
|
|
}
|
|
|
|
|
|
test fun sortFunctionShouldReturnSortedCopyForList() {
|
|
// TODO fixme Some sort of in/out variance thing - or an issue with Java interop?
|
|
todo {
|
|
// val list : List<Int> = arrayList<Int>(2, 3, 1)
|
|
// expect(arrayList(1, 2, 3)) { list.sort() }
|
|
// expect(arrayList(2, 3, 1)) { list }
|
|
}
|
|
}
|
|
|
|
test fun sortFunctionShouldReturnSortedCopyForIterable() {
|
|
// TODO fixme Some sort of in/out variance thing - or an issue with Java interop?
|
|
todo {
|
|
// val list : Iterable<Int> = arrayList(2, 3, 1)
|
|
// expect(arrayList(1, 2, 3)) { list.sort() }
|
|
// expect(arrayList(2, 3, 1)) { list }
|
|
}
|
|
}
|
|
}
|