Python Battle._side_endの例

コード例 #1

ファイル: test_battle.py プロジェクト: vs1720/poke-env

def test_battle_side_start_end():
    logger = MagicMock()
    battle = Battle("tag", "username", logger)
    battle._player_role = "p1"

    assert not battle.side_conditions

    condition = "safeguard"
    battle._parse_message(["", "-sidestart", "p1", condition])
    battle._parse_message(["", "-sidestart", "p2", condition])
    assert battle.side_conditions == {SideCondition.SAFEGUARD: 1}
    assert battle.opponent_side_conditions == {SideCondition.SAFEGUARD: 1}
    battle._parse_message(["", "-sidestart", "p1", condition])
    assert battle.side_conditions == {SideCondition.SAFEGUARD: 2}

    battle._parse_message(["", "-sideend", "p1", condition])
    battle._parse_message(["", "-sideend", "p2", condition])
    assert not battle.side_conditions
    assert not battle.opponent_side_conditions

    with pytest.raises(Exception):
        battle._side_end("p1", condition)

    with pytest.raises(Exception):
        battle._side_end("p2", condition)

コード例 #2

def test_battle_side_start_end():
    logger = MagicMock()
    battle = Battle("tag", "username", logger)
    battle._player_role = "p1"

    assert not battle.side_conditions

    condition = "safeguard"
    battle._side_start("p1", condition)

    assert battle.side_conditions == {SideCondition.SAFEGUARD}

    battle._side_end("p1", condition)

    assert not battle.side_conditions