commonMain.io.kotest.properties.shrinking.IntShrinker.kt Maven / Gradle / Ivy

Go to download
package io.kotest.properties.shrinking

import kotlin.math.abs

object IntShrinker : Shrinker {
   override fun shrink(failure: Int): List =
      when (failure) {
         0 -> emptyList()
         1, -1 -> listOf(0)
         else -> {
            val a = listOf(0, 1, -1, abs(failure), failure / 3, failure / 2, failure * 2 / 3)
            val b = (1..5).map { failure - it }.reversed().filter { it > 0 }
            (a + b).distinct().filterNot { it == failure }
         }
      }
}

object LongShrinker : Shrinker {
   override fun shrink(failure: Long): List =
      when (failure) {
         0L -> emptyList()
         1L, -1L -> listOf(0)
         else -> {
            val a = listOf(0L, 1L, -1L, abs(failure), failure / 3, failure / 2, failure * 2 / 3)
            val b = (1L..5L).map { failure - it }.reversed().filter { it > 0 }
            (a + b).distinct().filterNot { it == failure }
         }
      }
}