Exemplo n.º 1
0
 def test_get_object(self):
     inventory = Inventory()
     obj1 = Object('Test1')
     obj2 = Object('Test2')
     obj3 = Object('Test3')
     list = [obj1, obj2, obj3]
     
     inventory.add(obj1)
     inventory.add(obj2)
     inventory.add(obj3)
     
     self.assertEquals(inventory.get_object(-1), None) 
     
     for i in range(3):
         self.assertTrue(inventory.get_object(i) in list)
     
     self.assertEquals(inventory.get_object(3), None)