def test_choose_random_move_doubles(pseudo_random, example_doubles_request):
logger = MagicMock()
battle = DoubleBattle("tag", "username", logger)
player = RandomPlayer()
battle._parse_request(example_doubles_request)
battle._switch("p2a: Tyranitar", "Tyranitar, L50, M", "48/48")
pseudo_random.side_effect = lambda: 0
choice = player.choose_random_move(battle)
assert choice.message == "/choose move psychic -2, move geargrind -1"
pseudo_random.side_effect = lambda: 0.5
choice = player.choose_random_move(battle)
assert (choice.message ==
"/choose switch zamazentacrowned, move geargrind dynamax -1")
pseudo_random.side_effect = lambda: 0.999
choice = player.choose_random_move(battle)
assert choice.message == "/choose move slackoff dynamax, switch thundurus"
battle._switch("p2b: Excadrill", "Excadrill, L50, M", "48/48")
pseudo_random.side_effect = lambda: 0
choice = player.choose_random_move(battle)
assert choice.message == "/choose move psychic -2, move geargrind -1"
pseudo_random.side_effect = lambda: 0.5
choice = player.choose_random_move(battle)
assert choice.message == "/choose move slackoff, move wildcharge dynamax 2"
pseudo_random.side_effect = lambda: 0.999
choice = player.choose_random_move(battle)
assert choice.message == "/choose move slackoff dynamax, switch thundurus"
def test_choose_random_move_doubles(example_doubles_request):
logger = MagicMock()
battle = DoubleBattle("tag", "username", logger)
battle._parse_request(example_doubles_request)
battle._switch("p2a: Tyranitar", "Tyranitar, L50, M", "48/48")
player = RandomPlayer()
active_pokemon_1, active_pokemon_2 = battle.active_pokemon
choice_1 = player.choose_random_move(battle)
assert (any([move in choice_1 for move in active_pokemon_1.moves])
or "/choose switch" in choice_1)
assert (any([move in choice_1 for move in active_pokemon_2.moves])
or ",switch" in choice_1)
choices_100 = [player.choose_random_move(battle) for _ in range(100)]
assert not any([" 2" in choice for choice in choices_100])
assert any(["1" in choice for choice in choices_100])
assert any(["-1" in choice for choice in choices_100])
assert any(["-2" in choice for choice in choices_100])
assert any(["dynamax" in choice for choice in choices_100])
assert not any([choice.count("dynamax") == 2 for choice in choices_100])
assert any([choice.count("switch") == 2 for choice in choices_100])
battle._switch("p2b: Charizard", "Charizard, L50, M", "48/48")
choices_100 = [player.choose_random_move(battle) for _ in range(100)]
assert any([" 2" in choice for choice in choices_100])
def test_choose_random_move_doubles(pseudo_random, example_doubles_request):
# possible_choices_memo = (
# []
# ) # this needs to be reset at each start of Player.choose_random_move
# def count_substrings(substring: str, in_: List[str]) -> int:
# return sum(map(lambda el: substring in el, in_))
# def choose_non_dynamax(first_orders,) -> str:
# joined = DoubleBattleOrder.join_orders(possible_choices)
# possible_choices_memo.append(possible_choices.copy())
# for possible_choice in possible_choices:
# if " dynamax" not in possible_choice:
# return possible_choice
# raise ValueError(f"Only max moves are available in {possible_choices}")
logger = MagicMock()
battle = DoubleBattle("tag", "username", logger)
player = RandomPlayer()
battle._parse_request(example_doubles_request)
battle._switch("p2a: Tyranitar", "Tyranitar, L50, M", "48/48")
pseudo_random.side_effect = lambda: 0
choice = player.choose_random_move(battle)
assert choice.message == "/choose move psychic -2, move geargrind -1"
pseudo_random.side_effect = lambda: 0.5
choice = player.choose_random_move(battle)
assert (choice.message ==
"/choose switch zamazentacrowned, move geargrind dynamax -1")
pseudo_random.side_effect = lambda: 0.999
choice = player.choose_random_move(battle)
assert choice.message == "/choose move slackoff dynamax, switch thundurus"
battle._switch("p2b: Excadrill", "Excadrill, L50, M", "48/48")
pseudo_random.side_effect = lambda: 0
choice = player.choose_random_move(battle)
assert choice.message == "/choose move psychic -2, move geargrind -1"
pseudo_random.side_effect = lambda: 0.5
choice = player.choose_random_move(battle)
assert choice.message == "/choose move slackoff, move wildcharge dynamax 2"
pseudo_random.side_effect = lambda: 0.999
choice = player.choose_random_move(battle)
assert choice.message == "/choose move slackoff dynamax, switch thundurus"
def test_get_possible_showdown_targets(example_doubles_request):
logger = MagicMock()
battle = DoubleBattle("tag", "username", logger)
battle._parse_request(example_doubles_request)
mr_rime, klinklang = battle.active_pokemon
psychic = mr_rime.moves["psychic"]
slackoff = mr_rime.moves["slackoff"]
battle._switch("p2b: Tyranitar", "Tyranitar, L50, M", "48/48")
assert battle.get_possible_showdown_targets(psychic, mr_rime) == [-2, 2]
battle._switch("p2a: Milotic", "Milotic, L50, F", "48/48")
assert battle.get_possible_showdown_targets(psychic, mr_rime) == [-2, 1, 2]
assert battle.get_possible_showdown_targets(slackoff, mr_rime) == [0]
assert battle.get_possible_showdown_targets(psychic, mr_rime, dynamax=True) == [
1,
2,
]
assert battle.get_possible_showdown_targets(slackoff, mr_rime, dynamax=True) == [0]
def test_battle_request_parsing(example_doubles_request):
logger = MagicMock()
battle = DoubleBattle("tag", "username", logger)
battle._parse_request(example_doubles_request)
assert len(battle.team) == 6
pokemon_names = set(
map(lambda pokemon: pokemon.species, battle.team.values()))
assert "thundurus" in pokemon_names
assert "raichualola" in pokemon_names
assert "maractus" in pokemon_names
assert "zamazentacrowned" in pokemon_names
zamazenta = battle.get_pokemon("p1: Zamazenta")
zamazenta_moves = zamazenta.moves
assert (len(zamazenta_moves) == 4 and "closecombat" in zamazenta_moves
and "crunch" in zamazenta_moves
and "psychicfangs" in zamazenta_moves
and "behemothbash" in zamazenta_moves)
def test_battle_request_parsing_and_interactions(example_doubles_request):
logger = MagicMock()
battle = DoubleBattle("tag", "username", logger)
battle._parse_request(example_doubles_request)
mr_rime, klinklang = battle.active_pokemon
assert isinstance(mr_rime, Pokemon)
assert isinstance(klinklang, Pokemon)
assert battle.get_pokemon("p1: Mr. Rime") == mr_rime
assert battle.get_pokemon("p1: Klinklang") == klinklang
assert set(battle.available_moves[0]) == set(
battle.active_pokemon[0].moves.values()
)
assert set(battle.available_moves[1]) == set(
battle.active_pokemon[1].moves.values()
)
assert len(battle.available_switches) == 2
assert all(battle.can_dynamax)
assert not any(battle.can_z_move)
assert not any(battle.can_mega_evolve)
assert not any(battle.trapped)
assert not any(battle.force_switch)
assert not any(battle.maybe_trapped)
mr_rime._boosts = {
"accuracy": -2,
"atk": 1,
"def": -6,
"evasion": 4,
"spa": -4,
"spd": 2,
"spe": 3,
}
klinklang._boosts = {
"accuracy": -6,
"atk": 6,
"def": -1,
"evasion": 1,
"spa": 4,
"spd": -3,
"spe": 2,
}
battle._clear_all_boosts()
cleared_boosts = {
"accuracy": 0,
"atk": 0,
"def": 0,
"evasion": 0,
"spa": 0,
"spd": 0,
"spe": 0,
}
assert mr_rime.boosts == cleared_boosts
assert klinklang.boosts == cleared_boosts
assert battle.active_pokemon == [mr_rime, klinklang]
battle._parse_message(["", "swap", "p1b: Klinklang", ""])
assert battle.active_pokemon == [klinklang, mr_rime]
battle._switch("p2a: Milotic", "Milotic, L50, F", "48/48")
battle._switch("p2b: Tyranitar", "Tyranitar, L50, M", "48/48")
milotic, tyranitar = battle.opponent_active_pokemon
assert milotic.species == "milotic"
assert tyranitar.species == "tyranitar"
assert all(battle.opponent_can_dynamax)