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()
+