python-tdd/vending_machine.py at master · Huckcity/python-tdd · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from byotest import *

#usd_coins = [100,50,25,10,5,1]
#eur_coins = [100, 50, 20, 10, 5, 2, 1]

eur_coins =	{
  100: 20,
  50: 20,
  20: 20,
  10: 20,
  5: 20,
  2: 2,
  1: 20
}

usd_coins =	{
  100: 20,
  50: 20,
  25: 20,
  10: 20,
  5: 20,
  1: 20
}

def get_change(amount, coins=eur_coins):

    change = []
    for denom in sorted(coins.keys(), reverse=True):
        while denom <= amount and coins[denom] > 0:
            amount -= denom
            change.append(denom)
            coins[denom] -= 1

        if amount != 0:
            raise Exception "Not enough change to tender!"


    return change

test_are_equal(get_change(0), [])
test_are_equal(get_change(1), [1])
test_are_equal(get_change(2), [2])
test_are_equal(get_change(5), [5])
test_are_equal(get_change(10), [10])
test_are_equal(get_change(20), [20])
test_are_equal(get_change(50), [50])
test_are_equal(get_change(100), [100])
test_are_equal(get_change(3), [2,1])
test_are_equal(get_change(7), [5,2])
test_are_equal(get_change(9), [5,2,2])
test_are_equal(get_change(35, usd_coins), [25,10])

test_are_equal(get_change(9), [5,2,2])

print("All tests pass!")