fix tests

Author: cameronangliss

unit_tests/player/test_baselines.py

@@ -18,7 +18,7 @@ def test_max_base_power_player():
     assert player.choose_move(battle).message == "/choose default"
 
     battle._available_switches.append(Pokemon(species="ponyta", gen=8))
-    assert player.choose_move(battle).message == "/choose switch ponyta"
+    assert player.choose_move(battle).message == "/choose switch Ponyta"
 
     battle._available_moves.append(Move("protect", gen=8))
     assert player.choose_move(battle).message == "/choose move protect"
@@ -278,32 +278,32 @@ def test_random_player():
     mon._active = True
     battle._team = {"charizard": mon}
     battle._available_switches.append(Pokemon(species="ponyta", gen=8))
-    assert player.choose_move(battle).message == "/choose switch ponyta"
+    assert player.choose_move(battle).message == "/choose switch Ponyta"
 
     battle._available_moves.append(Move("protect", gen=8))
     assert player.choose_move(battle).message in {
         "/choose move protect",
-        "/choose switch ponyta",
+        "/choose switch Ponyta",
     }
 
     battle._available_moves.append(Move("quickattack", gen=8))
     assert player.choose_move(battle).message in {
         "/choose move protect",
-        "/choose switch ponyta",
+        "/choose switch Ponyta",
         "/choose move quickattack",
     }
 
     battle._available_moves.append(Move("flamethrower", gen=8))
     assert player.choose_move(battle).message in {
         "/choose move protect",
-        "/choose switch ponyta",
+        "/choose switch Ponyta",
         "/choose move quickattack",
         "/choose move flamethrower",
     }
 
     assert {player.choose_move(battle).message for _ in range(500)} == {
         "/choose move protect",
-        "/choose switch ponyta",
+        "/choose switch Ponyta",
         "/choose move quickattack",
         "/choose move flamethrower",
     }

unit_tests/player/test_doubles_baselines.py

@@ -21,7 +21,7 @@ def test_doubles_max_damage_player():
 
     # calls player.choose_random_doubles_move(battle)
     battle._available_switches[0].append(Pokemon(species="ponyta", gen=8))
-    assert player.choose_move(battle).message == "/choose switch ponyta, pass"
+    assert player.choose_move(battle).message == "/choose switch Ponyta, pass"
 
     active_raichu = Pokemon(species="raichu", gen=8)
     active_raichu.switch_in()
@@ -30,7 +30,7 @@ def test_doubles_max_damage_player():
     # calls player.choose_random_doubles_move(battle)
     battle._available_switches[1].append(Pokemon(species="rapidash", gen=8))
     assert (
-        player.choose_move(battle).message == "/choose switch ponyta, switch rapidash"
+        player.choose_move(battle).message == "/choose switch Ponyta, switch Rapidash"
     )
 
     active_ducklett = Pokemon(species="ducklett", gen=8)
@@ -42,7 +42,7 @@ def test_doubles_max_damage_player():
 
     # chooses a move for p1a, other slot defaults
     battle._available_moves[0].append(Move("protect", gen=8))
-    assert player.choose_move(battle).message == "/choose move protect, switch rapidash"
+    assert player.choose_move(battle).message == "/choose move protect, switch Rapidash"
 
     # chooses max BP move for both pokemon, targets chosen randomly
     battle._available_moves[0].append(Move("quickattack", gen=8))
@@ -64,13 +64,13 @@ def test_doubles_max_damage_player():
 
     # forced switch
     battle._force_switch = [True, False]
-    assert player.choose_move(battle).message in ["/choose switch ponyta, pass"]
+    assert player.choose_move(battle).message in ["/choose switch Ponyta, pass"]
 
     battle._force_switch = [False, True]
-    assert player.choose_move(battle).message in ["/choose pass, switch rapidash"]
+    assert player.choose_move(battle).message in ["/choose pass, switch Rapidash"]
 
     battle._force_switch = [True, True]
     battle._available_switches[0].append(battle._available_switches[1][0])
     assert player.choose_move(battle).message in [
-        "/choose switch ponyta, switch rapidash"
+        "/choose switch Ponyta, switch Rapidash"
     ]