Skip to content

Commit 41c2007

Browse files
[Andrew Polyakov] Substitute int32 only for NDArray
1 parent 0423439 commit 41c2007

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

utbot-python/src/main/kotlin/org/utbot/python/fuzzing/provider/IntValueProvider.kt

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ import org.utbot.python.fuzzing.PythonValueProvider
1717
import org.utbot.python.fuzzing.provider.utils.generateSummary
1818
import java.math.BigInteger
1919
import kotlin.random.Random
20+
import org.utpython.types.pythonTypeName
2021

2122
object IntValueProvider : PythonValueProvider {
2223
private val randomStubWithNoUsage = Random(0)
@@ -43,7 +44,12 @@ object IntValueProvider : PythonValueProvider {
4344
}
4445

4546
override fun generate(description: PythonMethodDescription, type: FuzzedUtType) = sequence<Seed<FuzzedUtType, PythonFuzzedValue>> {
46-
val bits = 32
47+
val isNDArray: Boolean = description.type.arguments.any { it.pythonTypeName() == "numpy.ndarray" }
48+
val bits = if (isNDArray) {
49+
32
50+
} else {
51+
128
52+
}
4753
val integerConstants = getIntConstants(description.concreteValues)
4854
val modifiedConstants = integerConstants.flatMap { value ->
4955
listOf(

0 commit comments

Comments
 (0)