Michael Nedzelsky
107 lines
3.6 KiB
Kotlin
107 lines
3.6 KiB
Kotlin
package foo
|
|
|
|
fun testSize(expectedSize: Int, vararg i: Int): Boolean {
|
|
return (i.size == expectedSize)
|
|
}
|
|
|
|
fun testSum(expectedSum: Int, vararg i: Int): Boolean {
|
|
var sum = 0
|
|
for (j in i) {
|
|
sum += j
|
|
}
|
|
|
|
return (expectedSum == sum)
|
|
}
|
|
|
|
fun testSpreadOperator(vararg args: Int): Boolean {
|
|
var sum = 0;
|
|
for (a in args) sum += a
|
|
|
|
return testSize(args.size, *args) && testSum(sum, *args)
|
|
}
|
|
|
|
class Bar(val size: Int, val sum: Int) {
|
|
fun test(vararg args: Int) = testSize(size, *args) && testSum(sum, *args)
|
|
}
|
|
|
|
object obj {
|
|
fun test(size: Int, sum: Int, vararg args: Int) = testSize(size, *args) && testSum(sum, *args)
|
|
}
|
|
|
|
fun spreadInMethodCall(size: Int, sum: Int, vararg args: Int) = Bar(size, sum).test(*args)
|
|
|
|
fun spreadInObjectMethodCall(size: Int, sum: Int, vararg args: Int) = obj.test(size, sum, *args)
|
|
|
|
fun testVarargWithFunLit(vararg args: Int, f: (a: IntArray) -> Boolean): Boolean = f(args)
|
|
|
|
fun idVarArgs<T>(vararg a: T) = a
|
|
|
|
fun idArrayVarArg<T>(vararg a: Array<T>) = a
|
|
|
|
fun sumFunValuesOnParameters(x: Int, y: Int, vararg a: Int, f: (Int) -> Int): Int {
|
|
var result = f(x) + f(y)
|
|
for(u in a) {
|
|
result += f(u)
|
|
}
|
|
return result
|
|
}
|
|
|
|
fun box(): String {
|
|
if (!testSize(0))
|
|
return "wrong vararg size when call function without args"
|
|
|
|
if (!testSum(0))
|
|
return "wrong vararg sum (arguments) when call function without args"
|
|
|
|
if (!testSize(6, 1, 1, 1, 2, 3, 4))
|
|
return "wrong vararg size when call function with some args (1)"
|
|
|
|
if (!testSum(30, 10, 20, 0))
|
|
return "wrong vararg sum (arguments) when call function with some args (1)"
|
|
|
|
if (!testSpreadOperator(30, 10, 20, 0))
|
|
return "failed when call function using spread operator"
|
|
|
|
if (!Bar(3, 30).test(10, 20, 0))
|
|
return "failed when call method"
|
|
|
|
if (!spreadInMethodCall(2, 3, 1, 2))
|
|
return "failed when call method using spread operator"
|
|
|
|
if (!obj.test(5, 15, 1, 2, 3, 4, 5))
|
|
return "failed when call method of object"
|
|
|
|
if (!spreadInObjectMethodCall(2, 3, 1, 2))
|
|
return "failed when call method of object using spread operator"
|
|
|
|
if (!testVarargWithFunLit(1, 2, 3) { args -> args.size == 3 })
|
|
return "failed when call function with vararg and fun literal"
|
|
|
|
val a = array(1, 2, 3)
|
|
val b = array(4, 5)
|
|
|
|
assertEquals(5, array(*a, *b).size)
|
|
assertEquals(8, array(10, *a, 20, *b, 30).size)
|
|
|
|
assertEquals(5, idVarArgs(*a, *b).size)
|
|
assertEquals(8, idVarArgs(10, *a, 20, *b, 30).size)
|
|
|
|
assertEquals(9, array(1, *a, *a, 1, 2).size)
|
|
assertEquals(9, idVarArgs(1, *a, *a, 1, 2).size)
|
|
|
|
assertEquals(9, array(1, *a, *array(1, 2, 3), 1, 2).size)
|
|
assertEquals(9, idVarArgs(1, *a, *array(1, 2, 3), 1, 2).size)
|
|
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, 3, 4, 5, 6, 7, 8, 9) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, *intArray(3, 4, 5, 6, 7, 8, 9)) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, 3, 4, *intArray(5, 6, 7, 8, 9)) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, *intArray(3, 4, 5, 6, 7), 8, 9) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, *intArray(3, 4, 5), *intArray(6, 7, 8, 9)) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, *intArray(3, 4), 5, 6, *intArray(7, 8, 9)) { 2*it })
|
|
|
|
assertEquals(2, idArrayVarArg(array(1), *array(array(2, 3, 4))).size())
|
|
assertEquals(3, idArrayVarArg(array(1, 2), *array(array(3, 4), array(5, 6))).size())
|
|
assertEquals(6, idArrayVarArg(array(1, 2), *array(array(3, 4), array(5, 6)), array(7), *array(array(8, 9), array(10, 11))).size())
|
|
|
|
return "OK"
|
|
} |