diff --git a/tests/test_evolve_functionality.py b/tests/test_evolve_functionality.py index 7d92c95..8c631a3 100644 --- a/tests/test_evolve_functionality.py +++ b/tests/test_evolve_functionality.py @@ -14,13 +14,13 @@ def test_find_next_evolution_name_and_id_for_bulbasaur(bulbasaur_chain): pokemon_name = "bulbasaur" evo_pokemon_name, evo_pokemon_id = _find_next_evolution_name_and_id(bulbasaur_chain, pokemon_name) assert evo_pokemon_name == "ivysaur" - assert evo_pokemon_id == "2" + assert evo_pokemon_id == 2 def test_find_next_evolution_name_and_id_for_ivysaur(bulbasaur_chain): pokemon_name = "ivysaur" evo_pokemon_name, evo_pokemon_id = _find_next_evolution_name_and_id(bulbasaur_chain, pokemon_name) assert evo_pokemon_name == "venusaur" - assert evo_pokemon_id == "3" + assert evo_pokemon_id == 3 def test_find_next_evolution_name_and_id_for_venusaur(bulbasaur_chain): pokemon_name = "venusaur" @@ -33,13 +33,13 @@ def test_find_next_evolution_name_and_id_for_weedle(weedle_chain): pokemon_name = "weedle" evo_pokemon_name, evo_pokemon_id = _find_next_evolution_name_and_id(weedle_chain, pokemon_name) assert evo_pokemon_name == "kakuna" - assert evo_pokemon_id == "14" + assert evo_pokemon_id == 14 def test_find_next_evolution_name_and_id_for_kakuna(weedle_chain): pokemon_name = "kakuna" evo_pokemon_name, evo_pokemon_id = _find_next_evolution_name_and_id(weedle_chain, pokemon_name) assert evo_pokemon_name == "beedrill" - assert evo_pokemon_id == "15" + assert evo_pokemon_id == 15 def test_find_next_evolution_name_and_id_for_beedrill(weedle_chain): pokemon_name = "beedrill"