고속 게임 엔진 골격 추가

2026-05-06 21:40:33 +09:00
parent 771c0ec821
commit ec0e468fa1
8 changed files with 862 additions and 2 deletions
@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+import random
+
+from coolrl_lost_cities.games.classic.game import Card, GameState, LostCitiesConfig
+
+from coolrl_lost_cities.games.classic.engines import FastGameState
+
+
+def _small_config() -> LostCitiesConfig:
+    return LostCitiesConfig(
+        n_colors=2,
+        n_ranks=2,
+        min_rank=1,
+        n_handshakes=0,
+        hand_size=1,
+        expedition_penalty=0,
+        bonus_threshold=99,
+        bonus_amount=0,
+    )
+
+
+def test_fast_new_game_from_deck_matches_game_state_snapshot() -> None:
+    config = _small_config()
+    deck = [
+        Card(0, 1),
+        Card(0, 2),
+        Card(1, 1),
+        Card(1, 2),
+    ]
+
+    classic = GameState.new_game_from_deck(deck, config)
+    fast = FastGameState.new_game_from_deck(deck, config)
+
+    assert fast.to_snapshot() == classic.to_snapshot()
+    fast.validate_invariants()
+
+
+def test_fast_snapshot_roundtrip_preserves_snapshot() -> None:
+    config = LostCitiesConfig(seed=11)
+    classic = GameState.new_game(config)
+    fast = FastGameState.from_snapshot(classic.to_snapshot())
+
+    assert fast.to_snapshot() == classic.to_snapshot()
+    restored = FastGameState.from_snapshot(fast.to_snapshot())
+    assert restored.to_snapshot() == fast.to_snapshot()
+
+
+def test_fast_random_action_sequence_matches_game_state() -> None:
+    config = LostCitiesConfig(
+        n_colors=3,
+        n_ranks=5,
+        min_rank=2,
+        n_handshakes=1,
+        hand_size=5,
+    )
+    for seed in range(48):
+        classic = GameState.new_game(config, seed=seed)
+        fast = FastGameState.new_game(config, seed=seed)
+        rng = random.Random(seed ^ 0xF457)
+        steps = 0
+
+        while True:
+            assert fast.to_snapshot() == classic.to_snapshot()
+            assert fast.unified_legal_mask() == classic.unified_legal_mask()
+            assert fast.score_diff(0) == classic.score_diff(0)
+            if classic.terminal:
+                break
+
+            legal = [
+                index for index, is_legal in enumerate(classic.unified_legal_mask()) if is_legal
+            ]
+            action = rng.choice(legal)
+            classic.apply_unified_action(action)
+            fast.apply_unified_action(action)
+            steps += 1
+            assert steps < 1000
+
+
+def test_fast_apply_undo_restores_every_legal_action() -> None:
+    config = LostCitiesConfig(
+        n_colors=3,
+        n_ranks=5,
+        min_rank=2,
+        n_handshakes=1,
+        hand_size=5,
+    )
+    for seed in range(32):
+        state = FastGameState.new_game(config, seed=seed)
+        rng = random.Random(seed ^ 0xFA57A11)
+        steps = 0
+
+        while not state.terminal:
+            legal = [index for index, is_legal in enumerate(state.unified_legal_mask()) if is_legal]
+            for action in legal:
+                candidate = state.clone()
+                before = candidate.to_snapshot()
+                undo = candidate.apply_unified_action_with_undo(action)
+                candidate.undo_action(undo)
+                assert candidate.to_snapshot() == before
+                candidate.validate_invariants()
+
+            state.apply_unified_action(rng.choice(legal))
+            steps += 1
+            assert steps < 1000