Revise tests to use new de-seralisation functions instead of deprecat…

…ed one.
gambitproject · Jan 9, 2025 · 1e40753 · 1e40753
1 parent ac2ab5d
commit 1e40753
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/games.py b/tests/games.py
@@ -5,7 +5,12 @@
 
 
 def read_from_file(fn: str) -> gbt.Game:
-    return gbt.Game.read_game(pathlib.Path("tests/test_games")/fn)
+    if fn.endswith(".efg"):
+        return gbt.read_efg(pathlib.Path("tests/test_games")/fn)
+    elif fn.endswith(".nfg"):
+        return gbt.read_nfg(pathlib.Path("tests/test_games")/fn)
+    else:
+        raise ValueError(f"Unknown file extension in {fn}")
 
 
 ################################################################################################