libraries/stdlib/test/ranges/ProgressionLastElementTest.kt

/*
 * Copyright 2010-2018 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package test.ranges

import kotlin.test.*

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 && increment != MIN) {
                    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)
        doTest(0, MAX, -3, MAX)

        // end == MIN
        doTest(0, MIN, -MAX, -MAX)
        doTest(0, MIN, MIN / 2, MIN)
        doTest(MAX, MIN, -MAX, -MAX)
        doTest(MIN + 7, MIN, -3, MIN + 1)
        doTest(MIN + 7, MIN, -MAX, MIN + 7)
        doTest(0, MIN, 3, MIN)
    }

    @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 !in minOf(start, end)..maxOf(start, end)) break
                        x = next
                    }

                    doTest(start, end, increment, x)
                }
            }
        }
    }
}