Skip to content

Commit 54cb096

Browse files
fix test
1 parent ea0e4da commit 54cb096

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

unit_tests/environment/test_battle.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -545,7 +545,7 @@ def test_battle_request_and_interactions(example_request):
545545
necrozma = battle.active_pokemon
546546
groudon = battle.opponent_active_pokemon
547547

548-
necrozma.switch_out()
548+
necrozma.switch_out(battle.fields)
549549
groudon.switch_in()
550550
groudon._ability = None
551551

@@ -595,7 +595,7 @@ def test_battle_request_and_interactions(example_request):
595595
["", "-ability", "p1a: Groudon", "Insomnia", "[from] move: Worry Seed"]
596596
)
597597
assert groudon.ability == "insomnia"
598-
groudon.switch_out()
598+
groudon.switch_out(battle.fields)
599599
groudon.switch_in()
600600
assert groudon.ability == "desolateland"
601601

@@ -614,7 +614,7 @@ def test_battle_request_and_interactions(example_request):
614614
assert groudon.ability == "prismarmor"
615615
assert necrozma.ability == "desolateland"
616616
groudon.switch_in()
617-
groudon.switch_out()
617+
groudon.switch_out(battle.fields)
618618
assert groudon.ability == "desolateland"
619619

620620
battle.parse_message(["", "switch", "p1a: Ho-oh", "Ho-oh, L82", "100/100"])
@@ -631,7 +631,7 @@ def test_battle_request_and_interactions(example_request):
631631
assert hooh.type_1 == PokemonType.THREE_QUESTION_MARKS
632632
assert hooh.type_2 == PokemonType.FLYING
633633
assert hooh.types == [PokemonType.THREE_QUESTION_MARKS, PokemonType.FLYING]
634-
hooh.switch_out()
634+
hooh.switch_out(battle.fields)
635635
hooh.switch_in()
636636
assert hooh.type_1 == PokemonType.FIRE
637637
assert hooh.type_2 == PokemonType.FLYING
@@ -642,7 +642,7 @@ def test_battle_request_and_interactions(example_request):
642642
assert hooh.type_1 == PokemonType.WATER
643643
assert hooh.type_2 is None
644644
assert hooh.types == [PokemonType.WATER]
645-
hooh.switch_out()
645+
hooh.switch_out(battle.fields)
646646
hooh.switch_in()
647647
assert hooh.type_1 == PokemonType.FIRE
648648
assert hooh.type_2 == PokemonType.FLYING

0 commit comments

Comments
 (0)