mirror of
https://github.com/LukeHagar/pokemonAPI-Amari.git
synced 2025-12-06 04:20:57 +00:00
32 lines
884 B
Python
Executable File
32 lines
884 B
Python
Executable File
import unittest
|
|
from poke_sdk.PokeAPI import PokeAPI
|
|
from poke_sdk.exceptions import APIError
|
|
|
|
|
|
class TestPokeAPI(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.api = PokeAPI()
|
|
|
|
def test_get_pokemon_valid(self):
|
|
response = self.api.get_pokemon("pikachu")
|
|
self.assertEqual(response.name, 'pikachu')
|
|
self.assertEqual(response.id, 25)
|
|
|
|
def test_get_pokemon_invalid(self):
|
|
with self.assertRaises(APIError):
|
|
self.api.get_pokemon("invalid_name")
|
|
|
|
def test_get_generation_valid(self):
|
|
response = self.api.get_generation("1")
|
|
self.assertEqual(response.id, 1)
|
|
self.assertEqual(response.name, 'generation-i')
|
|
|
|
def test_get_generation_invalid(self):
|
|
with self.assertRaises(APIError):
|
|
self.api.get_generation("invalid_id")
|
|
|
|
#adding main
|
|
if __name__ == '__main__':
|
|
unittest.main()
|