rand: update module

vlib/rand/rand_test.v

@@ -1,58 +1,58 @@
 import rand
 
+const (
+	rnd_count = 20
+	seeds = [42, 256]
+)
+
+fn test_rand_reproducibility() {
+	for seed in seeds {
+		mut randoms1 := gen_randoms(seed)
+		mut randoms2 := gen_randoms(seed)
+
+		assert_randoms_equal(randoms1, randoms2)
+	}
+}
+
+fn test_rand_r_reproducibility() {
+	for seed in seeds {
+		mut randoms1 := gen_randoms_r(seed)
+		mut randoms2 := gen_randoms_r(seed)
+
+		assert_randoms_equal(randoms1, randoms2)
+	}
+}
+
+fn test_rand_r_seed_update() {
+	seed := 10
+
+	for _ in 0..rnd_count {
+		prev_seed := seed
+		res := rand.rand_r(&seed)
+
+		assert prev_seed != seed
+	}
+}
+
 fn gen_randoms(seed int) []int {
-	mut randoms := [0].repeat(20)
+	mut randoms := [0].repeat(rnd_count)
 	rand.seed(seed)
-	for i in 0..20 {
+	for i in 0..rnd_count {
 		randoms[i] = rand.next(100)
 	}
 	return randoms
 }
 
-fn test_rand_reproducibility() {
-	mut randoms1 := gen_randoms(42)
-	mut randoms2 := gen_randoms(42)
-	assert randoms1.len == randoms2.len
-
-	mut len := randoms1.len
-	for i in 0..len {
-		assert randoms1[i] == randoms2[i]
-	}
-
-	randoms1 = gen_randoms(256)
-	randoms2 = gen_randoms(256)
-	assert randoms1.len == randoms2.len
-
-	len = randoms1.len
-	for i in 0..len {
-		assert randoms1[i] == randoms2[i]
-	}
-}
-
 fn gen_randoms_r(seed int) []int {
-	mut randoms := [0].repeat(20)
-	for i in 0..20 {
+	mut randoms := [0].repeat(rnd_count)
+	for i in 0..rnd_count {
 		randoms[i] = rand.rand_r(&seed)
 	}
 	return randoms
 }
 
-fn test_rand_r_reproducibility() {
-	mut randoms1 := gen_randoms_r(42)
-	mut randoms2 := gen_randoms_r(42)
-	assert randoms1.len == randoms2.len
-
-	mut len := randoms1.len
-	for i in 0..len {
-		assert randoms1[i] == randoms2[i]
-	}
-
-	randoms1 = gen_randoms_r(256)
-	randoms2 = gen_randoms_r(256)
-	assert randoms1.len == randoms2.len
-
-	len = randoms1.len
-	for i in 0..len {
-		assert randoms1[i] == randoms2[i]
+fn assert_randoms_equal(r1, r2 []int) {
+	for i in 0..rnd_count {
+		assert r1[i] == r2[i]
 	}
 }