Spaces:
Sleeping
Sleeping
File size: 5,290 Bytes
bb3fbf9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 |
import numpy as np
import pytest
from engine.game.enums import Phase
from engine.game.game_state import initialize_game
class TestVerificationBatch12:
@pytest.fixture
def game_state(self):
gs = initialize_game(use_real_data=True)
if not gs.member_db:
pytest.skip("Card data not loaded")
gs.active_player.tapped_energy = np.zeros(20, dtype=bool)
return gs
def get_card_id(self, gs, card_no):
for cid, card in gs.member_db.items():
if card.card_no == card_no:
return cid
for cid, card in gs.live_db.items():
if card.card_no == card_no:
return cid
pytest.fail(f"Card {card_no} not found")
def _test_live_retrieve(self, game_state, card_no):
"""
Generic test for "Retire self -> Retrieve Live from Discard"
Text: 自分の控え室からライブカードを1枚手札に加える (Add 1 Live from WR to Hand)
Cost: Type 5 (Sacrifice Self)
"""
gs = game_state
card_id = self.get_card_id(gs, card_no)
p1 = gs.players[0]
p1.stage[0] = card_id
# Setup discard with a valid Live card
live_id = 999
if live_id not in gs.live_db:
from engine.models.card import LiveCard
gs.live_db[live_id] = LiveCard(live_id, "D", "Dummy", 0, np.zeros(7, dtype=np.int32))
p1.discard = [live_id]
gs.current_player = 0
gs.phase = Phase.MAIN
# Activate (Action 200 for slot 0)
gs._execute_action(200)
# Cost: Retire
assert p1.stage[0] == -1, f"Card {card_no} should be retired (Cost)"
# Effect: Select Live from Discard
assert gs.pending_choices, "Should have pending choice"
choice_type, params = gs.pending_choices[0]
assert choice_type == "SELECT_FROM_DISCARD"
assert live_id in params["cards"]
gs._handle_choice(660) # Select index 0
assert live_id in p1.hand, "Live card should be in hand"
def _test_look3_pick1(self, game_state, card_no):
"""
Generic test for "Look 3, Add 1, Discard Rest" with Hand Discard Cost.
Text: 自分のデッキの上からカードを3枚見る... (Look 3 cards...)
Cost: Discard 1 card from hand (Type 3)
"""
gs = game_state
card_id = self.get_card_id(gs, card_no)
p1 = gs.players[0]
fodder_id = self.get_card_id(gs, "PL!-sd1-001-SD")
p1.hand = [card_id, fodder_id]
# Setup Deck
deck_ids = [901, 902, 903] # 3 Dummy IDs
p1.main_deck = deck_ids + [888] * 5
gs.current_player = 0
gs.phase = Phase.MAIN
gs._execute_action(1) # Play Card
# Verify Cost: Discard from Hand
assert gs.pending_choices, "Expected pending choice (Cost)"
choice_type, params = gs.pending_choices[0]
# We assume strict compliance: Cost Type 3 = TARGET_HAND
if choice_type == "TARGET_HAND" and params.get("reason") == "cost":
assert fodder_id in p1.hand
# Select fodder (index 0 if card_id played, or index 0 of remaining?)
# Played card is removed. Hand has [fodder]. Index 0.
gs._handle_choice(500)
else:
pytest.fail(f"Expected Discard Cost (TARGET_HAND), got {choice_type}")
# Verify Effect: Look 3, Pick 1
assert gs.pending_choices
choice_type, params = gs.pending_choices[0]
assert "SELECT" in choice_type # SELECT_FROM_LIST usually for Look & Choose
assert len(params["cards"]) == 3
# Pick 2nd card (902)
target_pick = params["cards"][1]
gs._handle_choice(601) # Index 1
assert target_pick in p1.hand
# Discard should contain: 1 (Cost) + 2 (Unpicked) = 3
assert len(p1.discard) == 3
def test_pls_bp2_009_p_live_retrieve(self, game_state):
self._test_live_retrieve(game_state, "PL!S-bp2-009-P")
def test_pls_bp2_009_r_live_retrieve(self, game_state):
self._test_live_retrieve(game_state, "PL!S-bp2-009-R")
def test_pls_pb1_004_p_plus_live_retrieve(self, game_state):
self._test_live_retrieve(game_state, "PL!S-pb1-004-P＋")
def test_pls_pb1_004_r_live_retrieve(self, game_state):
self._test_live_retrieve(game_state, "PL!S-pb1-004-R")
def test_plsp_bp1_011_p_live_retrieve(self, game_state):
self._test_live_retrieve(game_state, "PL!SP-bp1-011-P")
def test_plsp_bp1_011_r_live_retrieve(self, game_state):
self._test_live_retrieve(game_state, "PL!SP-bp1-011-R")
def test_plsp_pb1_018_n_live_retrieve(self, game_state):
self._test_live_retrieve(game_state, "PL!SP-pb1-018-N")
def test_plsp_sd1_006_sd_live_retrieve(self, game_state):
self._test_live_retrieve(game_state, "PL!SP-sd1-006-SD")
def test_pl_sd1_016_sd_look3_pick1(self, game_state):
self._test_look3_pick1(game_state, "PL!-sd1-016-SD")
def test_pln_pr_004_pr_look3_pick1(self, game_state):
self._test_look3_pick1(game_state, "PL!N-PR-004-PR")
|