diff --git a/compiler/tests/org/jetbrains/kotlin/runtime/ProgressionUtilTest.java b/compiler/tests/org/jetbrains/kotlin/runtime/ProgressionUtilTest.java deleted file mode 100644 index b6b1e0e3b4f..00000000000 --- a/compiler/tests/org/jetbrains/kotlin/runtime/ProgressionUtilTest.java +++ /dev/null @@ -1,113 +0,0 @@ -/* - * Copyright 2010-2015 JetBrains s.r.o. - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -package org.jetbrains.kotlin.runtime; - -import com.intellij.testFramework.UsefulTestCase; -import org.jetbrains.kotlin.codegen.forTestCompile.ForTestCompileRuntime; - -import java.lang.reflect.Method; - -public class ProgressionUtilTest extends UsefulTestCase { - private static final int MAX = Integer.MAX_VALUE; - private static final int MIN = Integer.MIN_VALUE; - - private Method intMethod; - private Method longMethod; - - @Override - protected void setUp() throws Exception { - super.setUp(); - Class progressionUtil = ForTestCompileRuntime.runtimeAndReflectJarClassLoader().loadClass("kotlin.internal.ProgressionUtilKt"); - this.intMethod = progressionUtil.getMethod("getProgressionFinalElement", int.class, int.class, int.class); - this.longMethod = progressionUtil.getMethod("getProgressionFinalElement", long.class, long.class, long.class); - } - - @Override - protected void tearDown() throws Exception { - this.intMethod = null; - this.longMethod = null; - super.tearDown(); - } - - private void doTest(int start, int end, int increment, int expected) throws Exception { - Object actualInt = intMethod.invoke(null, start, end, increment); - assertEquals(expected, actualInt); - - Object actualLong = longMethod.invoke(null, (long) start, (long) end, (long) increment); - assertEquals((long) expected, actualLong); - } - - private static final int[] INTERESTING = new int[]{ MIN, MIN / 2, -239, -23, -1, 0, 1, 42, 239, MAX / 2, MAX }; - - public void testGetFinalElement() throws Exception { - // start == end - for (int x : INTERESTING) { - for (int increment : INTERESTING) if (increment != 0) { - doTest(x, x, increment, x); - } - } - - // increment == 1 - for (int start = 0; start < INTERESTING.length; start++) { - for (int end = start; end < INTERESTING.length; end++) { - doTest(INTERESTING[start], INTERESTING[end], 1, INTERESTING[end]); - } - } - - // increment == -1 - for (int end = 0; end < INTERESTING.length; end++) { - for (int start = end; start < INTERESTING.length; start++) { - doTest(INTERESTING[start], INTERESTING[end], -1, INTERESTING[end]); - } - } - - // end == MAX - doTest(0, MAX, MAX, MAX); - doTest(0, MAX, MAX / 2, MAX - 1); - doTest(MIN + 1, MAX, MAX, MAX); - doTest(MAX - 7, MAX, 3, MAX - 1); - doTest(MAX - 7, MAX, MAX, MAX - 7); - - // end == MIN - doTest(0, MIN, MIN, MIN); - doTest(0, MIN, MIN / 2, MIN); - doTest(MAX, MIN, MIN, -1); - doTest(MIN + 7, MIN, -3, MIN + 1); - doTest(MIN + 7, MIN, MIN, MIN + 7); - - // Small tests - for (int start = -10; start < 10; start++) { - for (int end = -10; end < 10; end++) { - for (int increment = -20; increment < 20; increment++) { - // Cut down incorrect test data - if (increment == 0) continue; - if ((increment > 0) != (start <= end)) continue; - - // Iterate over the progression and obtain the expected result - int x = start; - while (true) { - int next = x + increment; - if (next < Math.min(start, end) || next > Math.max(start, end)) break; - x = next; - } - - doTest(start, end, increment, x); - } - } - } - } -} diff --git a/libraries/stdlib/test/language/ProgressionLastElementTest.kt b/libraries/stdlib/test/language/ProgressionLastElementTest.kt new file mode 100644 index 00000000000..e40befc992d --- /dev/null +++ b/libraries/stdlib/test/language/ProgressionLastElementTest.kt @@ -0,0 +1,84 @@ +package language + +import org.junit.Test +import kotlin.test.assertEquals + + +class ProgressionLastElementTest { + + private val MAX = Int.MAX_VALUE + private val MIN = Int.MIN_VALUE + + private val INTERESTING = intArrayOf(MIN, MIN / 2, -239, -23, -1, 0, 1, 42, 239, MAX / 2, MAX) + + private fun doTest(start: Int, end: Int, increment: Int, expected: Int) { + + val actualInt = IntProgression.fromClosedRange(start, end, increment).last + assertEquals(expected, actualInt) + + val actualLong = LongProgression.fromClosedRange(start.toLong(), end.toLong(), increment.toLong()).last + assertEquals(expected.toLong(), actualLong) + } + + @Test fun calculateFinalElement() { + // start == end + for (x in INTERESTING) { + for (increment in INTERESTING) + if (increment != 0) { + doTest(x, x, increment, x) + } + } + + // increment == 1 + for (start in INTERESTING.indices) { + for (end in start..INTERESTING.size - 1) { + doTest(INTERESTING[start], INTERESTING[end], 1, INTERESTING[end]) + } + } + + // increment == -1 + for (end in INTERESTING.indices) { + for (start in end..INTERESTING.size - 1) { + doTest(INTERESTING[start], INTERESTING[end], -1, INTERESTING[end]) + } + } + + // end == MAX + doTest(0, MAX, MAX, MAX) + doTest(0, MAX, MAX / 2, MAX - 1) + doTest(MIN + 1, MAX, MAX, MAX) + doTest(MAX - 7, MAX, 3, MAX - 1) + doTest(MAX - 7, MAX, MAX, MAX - 7) + + // end == MIN + doTest(0, MIN, MIN, MIN) + doTest(0, MIN, MIN / 2, MIN) + doTest(MAX, MIN, MIN, -1) + doTest(MIN + 7, MIN, -3, MIN + 1) + doTest(MIN + 7, MIN, MIN, MIN + 7) + } + + @Test fun iterateToFinalElement() { + // Small tests + for (start in -5..4) { + for (end in -5..4) { + for (increment in -10..9) { + // Cut down incorrect test data + if (increment == 0) continue + if (increment > 0 != start <= end) continue + + // Iterate over the progression and obtain the expected result + // println("$start,$end,$increment") + var x = start + while (true) { + val next = x + increment + if (next < Math.min(start, end) || next > Math.max(start, end)) break + x = next + } + + doTest(start, end, increment, x) + } + } + } + } +}