mining/knn: replace package "math/rand" with "crypto/rand"

1 files changed, 15 insertions, 5 deletions

diff --git a/lib/mining/knn/neighbor_test.go b/lib/mining/knn/neighbor_test.go
index 7d5139db..521fe2e4 100644
--- a/lib/mining/knn/neighbor_test.go
+++ b/lib/mining/knn/neighbor_test.go
@@ -5,10 +5,10 @@
 package knn
 
 import (
-	"math/rand"
+	"crypto/rand"
+	"math/big"
 	"sort"
 	"testing"
-	"time"
 
 	"github.com/shuLhan/share/lib/tabula"
 	"github.com/shuLhan/share/lib/test"
@@ -55,12 +55,22 @@ func createNeigboursByIdx(indices []int) (neighbors Neighbors) {
 }
 
 func TestContain(t *testing.T) {
-	rand.Seed(time.Now().UnixNano())
-
 	neighbors := createNeigbours()
 
 	// pick random sample from neighbors
-	pickIdx := rand.Intn(neighbors.Len())
+	var (
+		randMax = big.NewInt(int64(neighbors.Len()))
+		randv   *big.Int
+		err     error
+	)
+
+	randv, err = rand.Int(rand.Reader, randMax)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	var pickIdx = int(randv.Int64())
+
 	randSample := neighbors.Row(pickIdx).Clone()
 
 	isin, idx := neighbors.Contain(randSample)