Anton Bannykh
111 lines
3.8 KiB
Kotlin
Vendored
111 lines
3.8 KiB
Kotlin
Vendored
// EXPECTED_REACHABLE_NODES: 1310
|
|
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 <T> idVarArgs(vararg a: T) = a
|
|
|
|
fun <T> idArrayVarArg(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 = arrayOf(1, 2, 3)
|
|
val b = arrayOf(4, 5)
|
|
|
|
assertEquals(5, arrayOf(*a, *b).size)
|
|
assertEquals(8, arrayOf(10, *a, 20, *b, 30).size)
|
|
|
|
assertEquals(5, idVarArgs(*a, *b).size)
|
|
assertEquals(8, idVarArgs(10, *a, 20, *b, 30).size)
|
|
|
|
assertEquals(9, arrayOf(1, *a, *a, 1, 2).size)
|
|
assertEquals(9, idVarArgs(1, *a, *a, 1, 2).size)
|
|
|
|
assertEquals(9, arrayOf(1, *a, *arrayOf(1, 2, 3), 1, 2).size)
|
|
assertEquals(9, idVarArgs(1, *a, *arrayOf(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, *intArrayOf(3, 4, 5, 6, 7, 8, 9)) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, 3, 4, *intArrayOf(5, 6, 7, 8, 9)) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, *intArrayOf(3, 4, 5, 6, 7), 8, 9) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, *intArrayOf(3, 4, 5), *intArrayOf(6, 7, 8, 9)) { 2*it })
|
|
assertEquals(90, sumFunValuesOnParameters(1, 2, *intArrayOf(3, 4), 5, 6, *intArrayOf(7, 8, 9)) { 2*it })
|
|
|
|
assertEquals(2, idArrayVarArg(arrayOf(1), *arrayOf(arrayOf(2, 3, 4))).size)
|
|
assertEquals(3, idArrayVarArg(arrayOf(1, 2), *arrayOf(arrayOf(3, 4), arrayOf(5, 6))).size)
|
|
assertEquals(6, idArrayVarArg(arrayOf(1, 2), *arrayOf(arrayOf(3, 4), arrayOf(5, 6)), arrayOf(7), *arrayOf(arrayOf(8, 9), arrayOf(10, 11))).size)
|
|
|
|
val c = arrayOf(*a)
|
|
assertFalse(a === c, "Spread operator should copy its argument")
|
|
|
|
return "OK"
|
|
} |