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)