Fixes SpeedTest

fathzer-games · Jan 10, 2024 · ae34962 · ae34962
1 parent 230f641
commit ae34962
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/main/java/com/fathzer/jchess/uci/extended/SpeedTest.java b/src/main/java/com/fathzer/jchess/uci/extended/SpeedTest.java
@@ -71,12 +71,14 @@ public long run() {
 		final DeepeningPolicy policy = uciEngine.getEngine().getDeepeningPolicy();
 		int size = policy.getSize();
 		int accuracy = policy.getAccuracy();
+		int depth = policy.getDepth();
 		String fen = (uciEngine.isPositionSet() && uciEngine instanceof Displayable) ? ((Displayable)uciEngine).getFEN() : null;
 		try {
 			final long start = System.currentTimeMillis();
 			doSpeedTest();
 			return System.currentTimeMillis()-start;
 		} finally {
+			policy.setDepth(depth);
 			policy.setSize(size);
 			policy.setAccuracy(accuracy);
 			if (fen!=null) {
@@ -89,6 +91,7 @@ public long run() {
 	private void doSpeedTest() {
 		final DeepeningPolicy policy = uciEngine.getEngine().getDeepeningPolicy();
 		policy.setSize(Integer.MAX_VALUE);
+		policy.setDepth(8);
 
 		// 3 possible Mats in 1 with whites
 		Result<M> mv = fill("7k/5p2/5PQN/5PPK/6PP/8/8/8 w - - 6 5");