Add files via upload

lib/engine_wrapper.py

@@ -12,7 +12,6 @@
 import time
 import random
 import math
-import sys
 import test_bot.lichess
 from collections import Counter
 from collections.abc import Callable
@@ -588,6 +587,9 @@ def method(*args: Any, **kwargs: Any) -> Any:
         return method
 
 
+test_suffix = "-for-lichess-bot-testing-only"
+
+
 def getHomemadeEngine(name: str) -> type[MinimalEngine]:
     """
     Get the homemade engine with name `name`. e.g. If `name` is `RandomMove` then we will return `strategies.RandomMove`.
@@ -598,8 +600,8 @@ def getHomemadeEngine(name: str) -> type[MinimalEngine]:
     from lib import strategies
     from test_bot import strategies as test_strategies
     engine: type[MinimalEngine]
-    if "pytest" in sys.modules:
-        engine = getattr(test_strategies, name)
+    if name.endswith(test_suffix):  # Test only.
+        engine = getattr(test_strategies, name.removesuffix(test_suffix))
     else:
         engine = getattr(strategies, name)
     return engine

test_bot/test_bot.py

@@ -17,6 +17,7 @@
 from lib import config
 from lib.timer import Timer, to_seconds, seconds
 from typing import Any
+from lib.engine_wrapper import test_suffix
 if "pytest" not in sys.modules:
     sys.exit(f"The script {os.path.basename(__file__)} should only be run by pytest.")
 lichess_bot = importlib.import_module("lichess-bot")
@@ -292,7 +293,7 @@ def test_homemade() -> None:
     with open("./config.yml.default") as file:
         CONFIG = yaml.safe_load(file)
     CONFIG["token"] = ""
-    CONFIG["engine"]["name"] = "Stockfish"
+    CONFIG["engine"]["name"] = f"Stockfish{test_suffix}"
     CONFIG["engine"]["protocol"] = "homemade"
     CONFIG["pgn_directory"] = "TEMP/homemade_game_record"
     win = run_bot(CONFIG, logging_level)