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)