Spaces:
Running
Running
| import numpy as np | |
| import pytest | |
| from engine.game.enums import Phase | |
| from engine.game.game_state import initialize_game | |
| class TestVerificationBatch11: | |
| 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_pl_bp4_003_p_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!-bp4-003-P") | |
| def test_pl_bp4_003_r_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!-bp4-003-R") | |
| def test_pl_pb1_024_n_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!-pb1-024-N") | |
| def test_plhs_bp2_004_p_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!HS-bp2-004-P") | |
| def test_plhs_bp2_004_r_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!HS-bp2-004-R") | |
| def test_pln_pr_009_pr_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!N-PR-009-PR") | |
| def test_pln_pr_012_pr_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!N-PR-012-PR") | |
| def test_pln_pr_014_pr_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!N-PR-014-PR") | |
| def test_pln_sd1_011_sd_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!N-sd1-011-SD") | |
| def test_pls_pr_026_pr_live_retrieve(self, game_state): | |
| self._test_live_retrieve(game_state, "PL!S-PR-026-PR") | |