def test_one_product_can_take_many_slots(self):
machine = Machine(slots=2, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=20,
price=Decimal(2.1)),
}
machine.load_products(products)
available_products = machine.get_available_products().keys()
self.assertIn(ProductName("coca-cola"), available_products)
self.assertNotIn(ProductName("mars"), available_products)
def test_two_products_cant_share_one_slot(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=15,
price=Decimal(2.1)),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=15, price=Decimal(1.9)),
}
with self.assertRaises(MachineOverloadedException):
machine.load_products(products)
def test_loading_products(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=7, price=Decimal('2.1')),
ProductName("mars"): Product(name=ProductName("mars"), quantity=5, price=Decimal('1.9')),
ProductName("orbit"): Product(name=ProductName("orbit"), quantity=6, price=Decimal('2.3')),
}
machine.load_products(products)
available_products = machine.get_available_products().keys()
self.assertIn(ProductName("coca-cola"), available_products)
self.assertIn(ProductName("mars"), available_products)
self.assertIn(ProductName("orbit"), available_products)
self.assertNotIn(ProductName("lays"), available_products)
def test_can_give_smart_change(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal('2.1')),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5,
price=Decimal('1.9')),
ProductName("orbit"):
Product(name=ProductName("orbit"),
quantity=6,
price=Decimal('1.4')),
}
machine.load_products(products)
money = Coins({Decimal('0.2'): 3, Decimal('0.5'): 1})
machine.load_coins(money)
slot_code, _ = machine.get_available_products()[ProductName("orbit")]
product, change = machine.choose_product(slot_code,
Coins({Decimal(2): 1}))
self.assertIsNotNone(product)
self.assertIsNotNone(change)
self.assertEqual(product.name, ProductName("orbit"))
self.assertEqual(change, Coins({Decimal('0.2'): 3}))
def test_too_many_product_kinds(self):
machine = Machine(slots=2, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal(2.1)),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5, price=Decimal(1.9)),
ProductName("orbit"):
Product(name=ProductName("orbit"), quantity=6, price=Decimal(2.3)),
}
with self.assertRaises(MachineOverloadedException):
machine.load_products(products)
def test_cant_buy_unavailable_product(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=1,
price=Decimal('2.1')),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5,
price=Decimal('1.9')),
ProductName("orbit"):
Product(name=ProductName("orbit"),
quantity=6,
price=Decimal('2.3')),
}
machine.load_products(products)
money = Coins({Decimal(5): 1})
nonexistent_slot_code = SlotCode("doesn't exist")
self.assertNotIn(
nonexistent_slot_code,
[code for code, _ in machine.get_available_products().items()])
product, change = machine.choose_product(nonexistent_slot_code, money)
self.assertIsNone(product)
self.assertEqual(money, change)
self.assertEqual(machine.get_balance(), Decimal(0))
def test_second_delivery_to_big(self):
machine = Machine(slots=5, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal('2.1')),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5,
price=Decimal('1.9')),
ProductName("orbit"):
Product(name=ProductName("orbit"),
quantity=12,
price=Decimal('2.3')),
}
machine.load_products(products)
products_2 = {
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5,
price=Decimal('1.9')),
ProductName("lays"):
Product(name=ProductName("lays"), quantity=8,
price=Decimal('2.3')),
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal('2.1')),
}
with self.assertRaises(MachineOverloadedException):
machine.load_products(products_2)
def test_buy_a_product_if_cant_give_a_change(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=1, price=Decimal('2.1')),
ProductName("mars"): Product(name=ProductName("mars"), quantity=5, price=Decimal('1.9')),
ProductName("orbit"): Product(name=ProductName("orbit"), quantity=6, price=Decimal('2.3')),
}
machine.load_products(products)
money = Coins({Decimal(5): 1})
slot_code, _ = machine.get_available_products()[ProductName("coca-cola")]
product, change = machine.choose_product(slot_code, money)
self.assertIsNone(product)
self.assertEqual(money, change)
self.assertEqual(machine.get_balance(), Decimal(0))
def test_machine_keeps_track_of_available_products(self):
machine = Machine(slots=1, slot_depth=7)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=7, price=Decimal('2.1'))
}
machine.load_products(products)
slot_code, _ = machine.get_available_products()[ProductName('coca-cola')]
for _ in range(7):
product, change = machine.choose_product(slot_code, Coins({Decimal('2.1'): 1}))
self.assertIsNotNone(product)
self.assertIsNone(change)
new_cola, change = machine.choose_product(slot_code, Coins({Decimal('2.1'): 1}))
self.assertIsNone(new_cola)
self.assertIsNotNone(change)
self.assertEqual(change, Coins({Decimal('2.1'): 1}))
def test_cant_get_product_for_too_little_money(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=7, price=Decimal('2.1')),
ProductName("mars"): Product(name=ProductName("mars"), quantity=5, price=Decimal('1.9')),
ProductName("orbit"): Product(name=ProductName("orbit"), quantity=6, price=Decimal('2.3')),
}
too_little_money_to_buy_coke = Coins({Decimal('0.2'): 2})
machine.load_products(products)
slot_code, _ = machine.get_available_products()[ProductName("coca-cola")]
product, change = machine.choose_product(slot_code, too_little_money_to_buy_coke)
self.assertIsNone(product)
self.assertIsNotNone(change)
self.assertEqual(change, too_little_money_to_buy_coke)
def test_can_buy_available_product_and_different_change(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal(2.1)),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5, price=Decimal(1.9)),
ProductName("orbit"):
Product(name=ProductName("orbit"), quantity=6, price=Decimal(2.3)),
}
money = Coins({Decimal(2): 1, Decimal(0.5): 1})
machine.load_products(products)
machine.load_coins(Coins({Decimal(0.2): 2}))
slot_code, _ = machine.get_available_products()[ProductName(
"coca-cola")]
product, change = machine.choose_product(slot_code, money)
self.assertIsNotNone(product)
self.assertIsNotNone(change)
self.assertEqual(product.name, ProductName("coca-cola"))
self.assertEqual(change, Coins({Decimal(0.2): 2}))
from vending import Machine
from time import sleep
if __name__ == '__main__':
vm = Machine()
while True:
pass
def test_machine_can_rearrange(self):
machine = Machine(slots=4, slot_depth=10)
first_delivery = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=11,
price=Decimal('2.1')),
ProductName("mars"):
Product(name=ProductName("mars"),
quantity=15,
price=Decimal('1.9')),
}
machine.load_products(first_delivery)
cola_slot_code, _ = machine.get_available_products()[ProductName(
'coca-cola')]
_, _ = machine.choose_product(cola_slot_code,
Coins({Decimal('2.1'): 1}))
for _ in range(6):
mars_slot_code, _ = machine.get_available_products()[ProductName(
'mars')]
_, _ = machine.choose_product(mars_slot_code,
Coins({Decimal('1.9'): 1}))
second_delivery = {
ProductName("orbit"):
Product(name=ProductName("orbit"),
quantity=15,
price=Decimal('2.3')),
}
machine.load_products(second_delivery)
self.assertIn(ProductName("orbit"),
machine.get_available_products().keys())
def test_loading_coins(self):
machine = Machine(slots=2, slot_depth=10)
money = Coins({(Decimal(1) / Decimal(5)): 5, Decimal(1): 1})
machine.load_coins(money)
self.assertEqual(machine.get_balance(), Decimal(2))
from vending import Machine
from time import sleep
if __name__ == '__main__':
vm = Machine()
vm.display.print_rus_string(
(((0, 2), 'Тест компонентов'), ((2, 4), 'Любая кнопка'),
((3, 2), 'для продолжения')))
vm.wait_any_key()
vm.buttons.turn_on_light()
#vm.buttons.run_key_test()
vm.display.clear()
t1t2 = vm.t.get_temperature()
vm.display.print_rus_string(
(((0, 4), 'Температура'), ((1, 2),
'Датчик 1 Т={:.2f}'.format(t1t2[1])),
((2, 2), 'Датчик 2 Т={:.2f}'.format(t1t2[2]))))
vm.wait_any_key()
vm.display.clear()
h1h2 = vm.t.get_humidity()
vm.display.print_rus_string(
(((0, 5), 'Влажность'), ((1, 1), 'Датчик 1 Н={:.2f}%'.format(h1h2[1])),
((2, 1), 'Датчик 2 Н={:.2f}%'.format(h1h2[2]))))
vm.wait_any_key()
vm.display.clear()
vm.display.print_rus_string(
(((1, 4), 'Тестирование'), ((2, 6), 'окончено')))
sleep(2)
vm.buttons.turn_off_light()
vm.display.turn_off()
