mirror of
https://github.com/LukeHagar/pokemonAPI-Amari.git
synced 2025-12-06 04:20:57 +00:00
9e67a44a31
Adding IDE Project files Created an interactive_test_poke_api.py file that takes user input as an example.
55 lines
2.3 KiB
Python
55 lines
2.3 KiB
Python
import unittest
|
|
from poke_sdk.PokeAPI import PokeAPI
|
|
from poke_sdk.exceptions import APIError
|
|
|
|
|
|
class InteractiveTestPokeAPI(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
"""Set up the API client for testing."""
|
|
cls.api = PokeAPI()
|
|
|
|
def test_get_pokemon_with_user_input(self):
|
|
"""Test retrieving a Pokémon based on user input. Will return Pokémon Type and generation"""
|
|
pokemon_name = input("Enter the Pokémon name (e.g., pikachu, charizard, blastoise): ")
|
|
try:
|
|
response = self.api.get_pokemon(pokemon_name)
|
|
print(f"Retrieved Pokémon: {response.name}, ID: {response.id}, Types: {response.types}")
|
|
self.assertEqual(response.name, pokemon_name)
|
|
except APIError as e:
|
|
self.fail(f"APIError occurred: {e}")
|
|
except Exception as e:
|
|
print(f"An unexpected error occurred: {e}")
|
|
|
|
def test_get_pokemon_in_generation(self):
|
|
"""Test retrieving Pokémon in a specified generation based on user input. Returns all Pokémon in generation"""
|
|
generation_id = input("Enter the generation ID (e.g., 1): ")
|
|
try:
|
|
response = self.api.get_generation(generation_id)
|
|
print(f"Retrieved Generation: {response.name}, ID: {response.id}")
|
|
print("Pokémon in this generation:")
|
|
for pokemon in response.pokemon_species:
|
|
print(f"- {pokemon}")
|
|
except APIError as e:
|
|
self.fail(f"APIError occurred: {e}")
|
|
except Exception as e:
|
|
print(f"An unexpected error occurred: {e}")
|
|
|
|
#Test generation is a valid Pokémon Generation based on user input
|
|
def test_get_generation_with_user_input(self):
|
|
"""Test retrieving a generation based on user input. Will validate Pokémon generation is a valid Gen"""
|
|
generation_id = input("Enter the generation ID, will validate if gen is available(e.g., 1): ")
|
|
try:
|
|
response = self.api.get_generation(generation_id)
|
|
print(f"Retrieved Generation: {response.name}, ID: {response.id}")
|
|
self.assertEqual(response.id, int(generation_id))
|
|
except APIError as e:
|
|
self.fail(f"APIError occurred: {e}")
|
|
except Exception as e:
|
|
print(f"An unexpected error occurred: {e}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|