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