fixed failed unit tests

tareknatsheh · Jun 8, 2024 · a004fc4 · a004fc4
1 parent c37bdca
commit a004fc4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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"