From 08a8b93ff53b837315f1e634b30122f91359fe7e Mon Sep 17 00:00:00 2001
From: Ziyang Hu <hu.ziyang@cantab.net>
Date: Sun, 4 Dec 2022 22:54:23 +0800
Subject: [PATCH] fix test

---
 cozo-core/benches/pokec.rs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/cozo-core/benches/pokec.rs b/cozo-core/benches/pokec.rs
index 7cf3d27d..820d2999 100644
--- a/cozo-core/benches/pokec.rs
+++ b/cozo-core/benches/pokec.rs
@@ -226,8 +226,8 @@ fn single_vertex() {
 }
 
 fn single_vertex_write() {
+    let i = rand::thread_rng().gen_range(1..SIZES.0 * 10);
     for _ in 0..10 {
-        let i = rand::thread_rng().gen_range(1..SIZES.0 * 10);
         if TEST_DB
             .run_script(
                 "?[uid, cmpl_pct, gender, age] <- [[$id, 0, null, null]] :put user {uid => cmpl_pct, gender, age}",
@@ -241,12 +241,12 @@ fn single_vertex_write() {
 }
 
 fn single_edge_write() {
+    let i = rand::thread_rng().gen_range(1..SIZES.0);
+    let mut j = rand::thread_rng().gen_range(1..SIZES.0);
+    while j == i {
+        j = rand::thread_rng().gen_range(1..SIZES.0);
+    }
     for _ in 0..10 {
-        let i = rand::thread_rng().gen_range(1..SIZES.0);
-        let mut j = rand::thread_rng().gen_range(1..SIZES.0);
-        while j == i {
-            j = rand::thread_rng().gen_range(1..SIZES.0);
-        }
         if TEST_DB
             .run_script(
                 r#"
@@ -264,8 +264,8 @@ fn single_edge_write() {
 }
 
 fn single_vertex_update() {
+    let i = rand::thread_rng().gen_range(1..SIZES.0);
     for _ in 0..10 {
-        let i = rand::thread_rng().gen_range(1..SIZES.0);
         if TEST_DB
             .run_script(
                 r#"