Exemplo n.º 1
0
 def test_remove(self):
     inventory = Inventory()
     obj1 = Object('Test1')
     obj2 = Object('Test2')
     obj3 = Object('Test3')
     
     inventory.add(obj1)
     inventory.add(obj2)
     inventory.add(obj3)
     
     inventory.remove_object(obj1)
     
     self.assertEquals(len(inventory.objects), 2)
     self.assertFalse(obj1 in inventory.objects)
     self.assertTrue(obj2 in inventory.objects)
     self.assertTrue(obj3 in inventory.objects)
     
     inventory.remove_object(obj2)
     
     self.assertEquals(len(inventory.objects), 1)
     self.assertFalse(obj1 in inventory.objects)
     self.assertFalse(obj2 in inventory.objects)
     self.assertTrue(obj3 in inventory.objects)
     
     inventory.remove_object(obj3)
     
     self.assertEquals(len(inventory.objects), 0)
     self.assertFalse(obj1 in inventory.objects)
     self.assertFalse(obj2 in inventory.objects)
     self.assertFalse(obj3 in inventory.objects)
     
     inventory.remove_object(obj3)
     
     self.assertEquals(len(inventory.objects), 0)
     self.assertFalse(obj1 in inventory.objects)
     self.assertFalse(obj2 in inventory.objects)
     self.assertFalse(obj3 in inventory.objects)