Spaces:
Running
Running
| import pytest | |
| from engine.game.enums import Phase | |
| from engine.game.game_state import initialize_game | |
| class TestVerificationBatch13: | |
| 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.fill(False) | |
| 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_look3_pick1(self, game_state, card_no): | |
| """ | |
| Generic test for "Look 3, Add 1, Discard Rest" with Hand Discard Cost. | |
| """ | |
| 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 - use real IDs from the DB to avoid missing card errors | |
| valid_ids = list(gs.member_db.keys())[:5] | |
| p1.main_deck = valid_ids + [fodder_id] * 5 | |
| gs.current_player = 0 | |
| gs.phase = Phase.MAIN | |
| gs.step(1, check_legality=False, in_place=True) # Play Card | |
| # Verify Cost: Discard from Hand | |
| if gs.pending_choices: | |
| choice_type, params = gs.pending_choices[0] | |
| if choice_type == "TARGET_HAND" and params.get("reason") == "cost": | |
| idx = p1.hand.index(fodder_id) | |
| gs.step(500 + idx, check_legality=False, in_place=True) | |
| # Verify Effect: Look 3, Pick 1 | |
| assert gs.pending_choices | |
| choice_type, params = gs.pending_choices[0] | |
| assert "SELECT" in choice_type | |
| assert len(params["cards"]) == 3 | |
| target_pick = params["cards"][1] | |
| gs.step(601, check_legality=False, in_place=True) # Index 1 | |
| assert target_pick in p1.hand | |
| assert len(p1.discard) >= 3 | |
| def _test_draw1_discard1(self, game_state, card_no): | |
| """ | |
| Generic test for "Draw 1, Discard 1". | |
| """ | |
| gs = game_state | |
| card_id = self.get_card_id(gs, card_no) | |
| p1 = gs.players[0] | |
| p1.hand = [card_id] | |
| fodder_id = self.get_card_id(gs, "PL!-sd1-001-SD") | |
| p1.main_deck = [fodder_id] * 5 | |
| gs.current_player = 0 | |
| gs.phase = Phase.MAIN | |
| initial_deck = len(p1.main_deck) | |
| gs.step(1, check_legality=False, in_place=True) # Play | |
| # Verify Draw happened | |
| assert len(p1.main_deck) == initial_deck - 1 | |
| assert len(p1.hand) == 1 | |
| # Verify Discard Choice | |
| assert gs.pending_choices, "Expected Discard Choice" | |
| gs.step(500, check_legality=False, in_place=True) # Discard first | |
| assert len(p1.hand) == 0 | |
| assert len(p1.discard) >= 1 | |
| def test_pln_pr_006_pr_look3_pick1(self, game_state): | |
| self._test_look3_pick1(game_state, "PL!N-PR-006-PR") | |
| def test_pln_pr_013_pr_look3_pick1(self, game_state): | |
| self._test_look3_pick1(game_state, "PL!N-PR-013-PR") | |
| def test_pln_bp1_007_p_look3_pick1(self, game_state): | |
| self._test_look3_pick1(game_state, "PL!N-bp1-007-P") | |
| def test_pln_bp1_007_r_look3_pick1(self, game_state): | |
| self._test_look3_pick1(game_state, "PL!N-bp1-007-R") | |
| def test_pln_bp1_010_p_look3_pick1(self, game_state): | |
| self._test_look3_pick1(game_state, "PL!N-bp1-010-P") | |
| def test_pln_bp1_010_r_look3_pick1(self, game_state): | |
| self._test_look3_pick1(game_state, "PL!N-bp1-010-R") | |
| def test_pln_sd1_002_sd_look3_pick1(self, game_state): | |
| self._test_look3_pick1(game_state, "PL!N-sd1-002-SD") | |
| def test_pln_sd1_003_sd_look3_pick1(self, game_state): | |
| self._test_look3_pick1(game_state, "PL!N-sd1-003-SD") | |
| def test_plhs_bp1_010_n_draw1_discard1(self, game_state): | |
| self._test_draw1_discard1(game_state, "PL!HS-bp1-010-N") | |
| def test_plhs_bp1_014_n_draw1_discard1(self, game_state): | |
| self._test_draw1_discard1(game_state, "PL!HS-bp1-014-N") | |