fix: RandomSubSlice

milkyway-labs · Dec 11, 2024 · 30da3d5 · 30da3d5
1 parent 5e805a1
commit 30da3d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/testutils/simtesting/utils.go b/testutils/simtesting/utils.go
@@ -100,8 +100,8 @@ func RandomSubSlice[T any](r *rand.Rand, items []T) []T {
 
 	var elements []T
 	// Randomly select how many items to pick
-	count := r.Intn(len(items))
-	for i := 0; i < count; i++ {
+	count := r.Intn(len(items) + 1)
+	for len(pickedIndexes) < count {
 		// Get a random index
 		index := r.Intn(len(items))
 		_, found := pickedIndexes[index]