def test_part2():
presents = 29000000
n, _ = min_house(presents, part2=True)
assert n == 705600
def test_part1():
presents = 29000000
n, _ = min_house(presents)
assert n == 665280
def test_small_presents():
presents = 150
n, _ = min_house(presents)
assert n == 8
def test_n_presents_numpy_part1_vs_part2(i):
max_presents = 1000
n1, houses1 = min_house(max_presents, part2=False)
n2, houses2 = min_house(max_presents, part2=True)
adjustment = 0 if i <= 50 else 11
assert houses2[i] == (11 * houses1[i] / 10) - adjustment
def test_n_presents_numpy_part2(h, p):
max_presents = max([p for h, p in tests])
n, houses = min_house(max_presents, part2=True)
assert houses[h] == 11 * p / 10
def test_n_presents_numpy(h, p):
max_presents = max([p for h, p in tests])
n, houses = min_house(max_presents)
assert houses[h] == p