Spaces:
Running
Running
File size: 2,966 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 |
import numpy as np
from pytest_bdd import given, parsers, scenarios, then, when
from engine.game.game_state import MemberCard
from engine.models.ability import Ability, AbilityCostType, Cost, Effect, EffectType, TargetType, TriggerType
scenarios("../features/triggers.feature")
@given('the player has a member "Shizuku" with an optional discard cost ability')
def setup_shizuku(context, game_state):
p = game_state.players[0]
# Create Shizuku member
mid = 777
shizuku = MemberCard(
card_id=mid,
card_no="SZ-01",
name="Shizuku",
cost=1,
groups=[],
hearts=np.zeros(6),
blade_hearts=np.zeros(7),
blades=1,
)
# Add ability: Discard 1 (Optional) -> Recover Live
abi = Ability(
raw_text="Test Shizuku",
trigger=TriggerType.ACTIVATED, # Using ACTIVATED for manual trigger in test
costs=[Cost(AbilityCostType.DISCARD_HAND, 1, is_optional=True)],
effects=[Effect(EffectType.RECOVER_LIVE, 1, TargetType.CARD_DISCARD, {"group": "Niji"})],
)
shizuku.abilities.append(abi)
game_state.member_db[mid] = shizuku
p.stage[0] = mid
# Needs discard for RECOVER_LIVE not to fail logic?
# Actually if cost skipped, effect won't run.
context["shizuku_id"] = mid
@when('the player activates the ability of "Shizuku"')
def activate_shizuku(context, game_state):
# Action 200 = Activate ability of Member at Slot 0
# (200 + area)
game_state = game_state.step(200)
context["game_state"] = game_state
@then("the choice should be optional")
def check_optional_choice(context, game_state):
current_state = context["game_state"]
choice = current_state.pending_choices[0]
params = choice[1]
# Check if params has is_optional=True
# Currently engine mismatch: engine doesn't set it yet. Expect failure.
assert params.get("is_optional") is True, f"Choice params {params} should have is_optional=True"
@when("the player chooses to skip the optional cost")
def skip_optional_cost(context, game_state):
current_state = context["game_state"]
# Verify action 0 is legal?
actions = current_state.get_legal_actions()
assert actions[0], "Action 0 (SKIP) should be legal"
current_state = current_state.step(0)
context["game_state"] = current_state
@then(parsers.parse("the player's hand size should remain {count:d}"))
def check_hand_remain(context, game_state, count):
current_state = context["game_state"]
p = current_state.players[0]
assert len(p.hand) == count
@then("the ability effects should not execute")
def check_no_effect(context, game_state):
current_state = context["game_state"]
# Check pending effects are empty
assert len(current_state.pending_effects) == 0
# Check pending choices are empty
assert len(current_state.pending_choices) == 0
|