def test_remove_item_from_slot_no_items(self):
vending_machine = VendingMachine()
removed = vending_machine.remove_item_from_slot(1)
self.assertFalse(removed)
self.assertEqual(9, vending_machine.available_slots)
self.assertIsNone(vending_machine.slot_items[1])
def test_remove_item_from_slot_invalid_slot(self):
vending_machine = VendingMachine()
removed = vending_machine.remove_item_from_slot(11)
self.assertFalse(removed)
self.assertEqual(9, vending_machine.available_slots)
self.assertEqual('Invalid slot',
vending_machine.slot_items.get(11, 'Invalid slot'))
def test_remove_item_from_slot_success(self):
sparkling_water = Item('Sparkling Water', 1.25, 6)
vending_machine = VendingMachine()
vending_machine.add_item_to_slot(1, sparkling_water)
removed = vending_machine.remove_item_from_slot(1)
self.assertTrue(removed)
self.assertEqual(9, vending_machine.available_slots)
self.assertIsNone(vending_machine.slot_items[1])
