coolguy
12b59211b2
맥락: - Python 코드만 대상으로 Ruff 기반 format/lint와 pre-commit hook을 도입한다. - Cython game.pyx는 Ruff 대상에서 제외해 포맷터 충돌을 피한다. 변경: - ruff와 pre-commit dev dependency 및 Ruff 설정을 추가했다. - pre-commit config와 Python format/check 스크립트를 추가했다. - Ruff format/check --fix 결과로 Python import 정렬과 포맷을 적용했다. 확인: - uv sync --extra gui - uv run pre-commit run --all-files - scripts/check-python.sh - uv run lost-cities-classic
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
import numpy as np
|
|
from coolrl_lost_cities.games.classic.game import Card, GameState, LostCitiesConfig
|
|
|
|
from coolrl_lost_cities.games.classic.env import LostCitiesEnv
|
|
|
|
|
|
def test_env_observation_uses_fixed_unified_mask() -> None:
|
|
config = LostCitiesConfig()
|
|
env = LostCitiesEnv(config)
|
|
|
|
obs = env.reset()
|
|
assert obs["legal_mask"].shape == (config.action_size,)
|
|
assert env.phase == "card"
|
|
|
|
card_action = int(np.nonzero(obs["legal_mask"])[0][0])
|
|
obs, _, _, _ = env.step(card_action)
|
|
|
|
assert env.phase == "draw"
|
|
assert obs["legal_mask"].shape == (config.action_size,)
|
|
assert np.all(obs["legal_mask"][: config.card_action_size] == 0)
|
|
assert np.any(obs["legal_mask"][config.card_action_size :])
|
|
|
|
|
|
def test_env_step_accepts_legacy_draw_action_ids() -> None:
|
|
config = LostCitiesConfig()
|
|
env = LostCitiesEnv(config)
|
|
env.state = GameState.empty(config)
|
|
env.state.hands[0] = [Card(0, 1)]
|
|
env.state.hands[1] = [Card(1, 1)]
|
|
env.state.deck = [Card(2, 1), Card(2, 2)]
|
|
env.state.phase = "draw"
|
|
|
|
obs, reward, done, _ = env.step(0)
|
|
|
|
assert obs["legal_mask"].shape == (config.action_size,)
|
|
assert reward == 0.0
|
|
assert done is False
|
|
assert env.current_player == 1
|
|
assert env.phase == "card"
|
|
|
|
|
|
def test_terminal_reward_is_relative_to_actor() -> None:
|
|
config = LostCitiesConfig(
|
|
n_colors=2,
|
|
n_ranks=1,
|
|
min_rank=1,
|
|
n_handshakes=0,
|
|
hand_size=1,
|
|
expedition_penalty=0,
|
|
bonus_threshold=99,
|
|
)
|
|
env = LostCitiesEnv(config)
|
|
env.state = GameState.empty(config)
|
|
env.state.current_player = 1
|
|
env.state.phase = "draw"
|
|
env.state.deck = [Card(1, 1)]
|
|
env.state.expeditions[1][0] = [Card(0, 1)]
|
|
|
|
_, reward, done, _ = env.step(config.card_action_size)
|
|
|
|
assert done is True
|
|
assert reward == 1.0
|