Fix benchmark

primenumber · May 4, 2024 · 23fc861 · 23fc861
1 parent 85131d6
commit 23fc861
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/engine/eval.rs b/src/engine/eval.rs
@@ -1,5 +1,3 @@
-#[cfg(test)]
-mod test;
 use crate::engine::board::*;
 
 pub trait Evaluator: Send + Sync {

diff --git a/src/engine/eval/test.rs → src/engine/pattern_eval/test.rs b/src/engine/eval/test.rs → src/engine/pattern_eval/test.rs
@@ -1,7 +1,7 @@
 extern crate test;
 use super::*;
-use crate::setup::*;
 use std::io::{BufRead, BufReader};
+use std::sync::Arc;
 use test::Bencher;
 
 fn load_stress_test_set() -> Vec<(Board, i8)> {
@@ -26,13 +26,13 @@ fn load_stress_test_set() -> Vec<(Board, i8)> {
 
 #[bench]
 fn bench_eval(b: &mut Bencher) {
-    let solve_obj = setup_default();
+    let evaluator = Arc::new(PatternLinearEvaluator::load(Path::new("table-220710")).unwrap());
     let dataset = load_stress_test_set();
 
     b.iter(|| {
         dataset
             .iter()
-            .map(|(board, _)| solve_obj.evaluator.eval(*board) as i32)
+            .map(|(board, _)| evaluator.eval(*board) as i32)
             .sum::<i32>()
     });
 }