Mercurial > parpg-core
diff tests/test_inventory.py @ 0:1fd2201f5c36
Initial commit of parpg-core.
author | M. George Hansen <technopolitica@gmail.com> |
---|---|
date | Sat, 14 May 2011 01:12:35 -0700 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/test_inventory.py Sat May 14 01:12:35 2011 -0700 @@ -0,0 +1,84 @@ +#!/usr/bin/env python + +# This file is part of PARPG. +# +# PARPG is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# PARPG is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with PARPG. If not, see <http://www.gnu.org/licenses/>. + +import unittest +from parpg.inventory import Inventory +from parpg.objects.base import GameObject, Carryable + +class TestInventory(unittest.TestCase): + class CarryableObject (GameObject, Carryable): + def __init__ (self, ID, **kwargs): + GameObject.__init__(self, ID, **kwargs) + Carryable.__init__(self, **kwargs) + + def setUp(self): + self.item = self.CarryableObject(12, name="TestItem1") + self.item.weight = 12 + self.item2 = self.CarryableObject(13) + self.item2.weight = 13 + self.inventory = Inventory() + + def testPlaceTakeMove(self): + """Test Inventory Place/Take/Move functions""" + self.assertTrue(self.inventory.isSlotEmpty("backpack")) + self.inventory.placeItem(self.item) + self.assertTrue(self.item in self.inventory.getItemsInSlot("backpack").values()) + self.assertEqual(self.inventory.weight, 12) + self.assertEqual(self.inventory.count(), 1) + self.assertFalse(self.inventory.isSlotEmpty("backpack")) + + self.inventory.moveItemToSlot(self.item, "groin") + self.assertFalse(self.item in self.inventory.getItemsInSlot("backpack").values()) + self.assertTrue(self.item in self.inventory.getItemsInSlot("groin").values()) + self.assertEqual(self.inventory.count(), 1) + + self.assertRaises(ValueError, self.inventory.moveItemToSlot, self.item2, "somewhere") + + self.inventory.moveItemToSlot(self.item2, "chest") + self.assertEqual(self.inventory.count(),2) + self.assertEqual(self.inventory.weight, 12+13) + self.assertTrue(self.item2 in self.inventory.getItemsInSlot("chest").values()) + + self.inventory.takeItem(self.item) + self.assertEqual(self.inventory.count(),1) + self.assertEqual(self.inventory.weight, 13) + + def testReplace(self): + """Test Inventory items replace each other in single-item slots""" + self.inventory.placeItem(self.item) + self.inventory.moveItemToSlot(self.item,"neck") + self.assertFalse(self.inventory.isSlotEmpty("neck")) + self.assertTrue(self.item in self.inventory.getItemsInSlot("neck").values()) + + self.inventory.moveItemToSlot(self.item2, "neck") + self.assertFalse(self.inventory.isSlotEmpty("neck")) + self.assertTrue(self.item2 in self.inventory.getItemsInSlot("neck").values()) + self.assertFalse(self.item in self.inventory.getItemsInSlot("neck").values()) + + def testFind(self): + self.inventory.placeItem(self.item) + self.assertEqual(self.inventory.findItemByID(12), self.item) + self.assertEqual(self.inventory.findItemByID(13), None) + self.assertEqual(self.inventory.findItem(name="TestItem1"), self.item) + self.assertEqual(self.inventory.findItem(name="RandomName1"), None) + self.assertEqual(self.inventory.findItem(kind="carryable"), self.item) + self.assertEqual(self.inventory.findItem(kind="weapon"), None) + + +if __name__ == '__main__': + unittest.main() +