def get_kingdom(self, name): """Get the Kingdom Object from the Name of the kingdom Args: name (string): Name of the Kingdom Returns: Kingdom Object: Kingdom Object for the given name Raises: KingdomNotFound: If no Kingdom is found for the kingdom name passed in as a parameter. """ return Kingdom.get_kingdom(name)
def test_get_kingdom(self): name = "ICE" emblem = "MAMMOTH" ice_kingdom = Kingdom(name, emblem) self.assertEqual(ice_kingdom, Kingdom.get_kingdom(name)) Kingdom.remove_all_kingdoms()