All class built. Coverage 100%

Author: Paul Hallett

pykemon/models.py

@@ -104,3 +104,63 @@ def __init__(self, bundle):
 
     def __repr__(self):
         return '<Ability - %s>' % self.name.capitalize()
+
+
+class Egg(DateTimeObject):
+    """
+    This class represents a single Egg group resource
+    """
+
+    def __init__(self, bundle):
+        super(Egg, self).__init__(bundle)
+        self.id = bundle['id']
+        self.pokemon = buildr(bundle, 'pokemon')
+
+    def __repr__(self):
+        return '<Egg - %s>' % self.name.capitalize()
+
+
+class Description(DateTimeObject):
+    """
+    This class represents a single Description resource
+    """
+
+    def __init__(self, bundle):
+        super(Description, self).__init__(bundle)
+        self.id = bundle['id']
+        self.description = bundle['description']
+        self.pokemon = bundle['pokemon']
+        self.games = buildr(bundle, 'games')
+
+    def __repr__(self):
+        return '<Description - %s>' % self.name.capitalize()
+
+
+class Sprite(DateTimeObject):
+    """
+    This class represents a single Sprite resource
+    """
+
+    def __init__(self, bundle):
+        super(Sprite, self).__init__(bundle)
+        self.id = bundle['id']
+        self.pokemon = bundle['pokemon']
+        self.image = bundle['image']
+
+    def __repr__(self):
+        return '<Sprite - %s>' % self.name.capitalize()
+
+
+class Game(DateTimeObject):
+    """
+    This class represents a single Game resource
+    """
+
+    def __init__(self, bundle):
+        super(Game, self).__init__(bundle)
+        self.id = bundle['id']
+        self.generation = bundle['generation']
+        self.release_year = bundle['release_year']
+
+    def __repr__(self):
+        return '<Game - %s>' % self.name.capitalize()

pykemon/request.py

@@ -17,14 +17,18 @@
 import requests
 import simplejson
 from simplejson import JSONDecodeError
-from models import Pokemon, Move, Type, Ability
+from models import Pokemon, Move, Type, Ability, Egg, Description, Sprite, Game
 from exceptions import ResourceNotFoundError
 
 CLASSES = {
     'pokemon': Pokemon,
     'move': Move,
     'type': Type,
     'ability': Ability,
+    'egg': Egg,
+    'description': Description,
+    'sprite': Sprite,
+    'game': Game
 }
 
 

tests/test_pykemon.py

@@ -21,24 +21,42 @@ def setUp(self):
         self.move_one = pykemon.get(move_id=15)  # Cut
         self.type_one = pykemon.get(type_id=10)  # Fire
         self.ability_one = pykemon.get(ability_id=1)  # Stench
+        self.egg_one = pykemon.get(egg_id=1)  # Monster
+        self.description_one = pykemon.get(description_id=1)
+        self.sprite_one = pykemon.get(sprite_id=152)  # Mew_auto
+        self.game_one = pykemon.get(game_id=4)  # Red
 
     def test_name_attribute(self):
         self.assertEquals(self.poke_one.name, 'Bulbasaur')
         self.assertEquals(self.move_one.name, 'Cut')
         self.assertEquals(self.type_one.name, 'Fire')
         self.assertEquals(self.ability_one.name, 'Stench')
+        self.assertEquals(self.egg_one.name, 'Monster')
+        self.assertEquals(self.description_one.name, 'Bulbasaur_red_blue')
+        self.assertEquals(self.sprite_one.name, 'Mew_auto')
+        self.assertEquals(self.game_one.name, 'Red')
 
     def test_repr(self):
         self.assertEquals(str(self.poke_one), '<Pokemon - Bulbasaur>')
         self.assertEquals(str(self.move_one), '<Move - Cut>')
         self.assertEquals(str(self.type_one), '<Type - Fire>')
         self.assertEquals(str(self.ability_one), '<Ability - Stench>')
+        self.assertEquals(str(self.egg_one), '<Egg - Monster>')
+        self.assertEquals(
+            str(self.description_one), '<Description - Bulbasaur_red_blue>')
+        self.assertEquals(str(self.sprite_one), '<Sprite - Mew_auto>')
+        self.assertEquals(str(self.game_one), '<Game - Red>')
 
     def test_resource_uri_attribute(self):
         self.assertEquals(self.poke_one.resource_uri, '/api/v1/pokemon/1/')
         self.assertEquals(self.move_one.resource_uri, '/api/v1/move/15/')
         self.assertEquals(self.type_one.resource_uri, '/api/v1/type/10/')
         self.assertEquals(self.ability_one.resource_uri, '/api/v1/ability/1/')
+        self.assertEquals(self.egg_one.resource_uri, '/api/v1/egg/1/')
+        self.assertEquals(
+            self.description_one.resource_uri, '/api/v1/description/1/')
+        self.assertEquals(self.sprite_one.resource_uri, '/api/v1/sprite/152/')
+        self.assertEquals(self.game_one.resource_uri, '/api/v1/game/4/')
 
     def test_pokemon_complex_attribs(self):
         self.assertIn('Ivysaur', self.poke_one.evolutions)
@@ -53,6 +71,12 @@ def test_type_complex_attribs(self):
         self.assertIn('fire', self.type_one.ineffective)
         self.assertNotIn('Test', self.type_one.resistance)
 
+    def test_egg_complex_attribs(self):
+        self.assertIn('Blastoise', self.egg_one.pokemon)
+
+    def test_description_complex_attribs(self):
+        self.assertIn('blue', self.description_one.games)
+
 
 class Testexceptions(unittest.TestCase):
     " Test error handling "