def test_part2():
with open('input.txt') as f:
pixels = f.read()
image = build_image(pixels, x=25, y=6)
render = flatten(image)
assert image_to_string(render) == part2_output
def test_part1():
with open('input.txt') as f:
pixels = f.read()
image = build_image(pixels, x=25, y=6)
layer = find_densest_layer(image)
assert layer == 7
checksum = calc_checksum(image, layer)
assert checksum == 1677
def test_1():
image = build_image(test_input, x=3, y=2)
expected_0 = np.array([[1, 2, 3], [4, 5, 6]])
expected_1 = np.array([[7, 8, 9], [0, 1, 2]])
assert np.all(expected_0 == image[0, :, :])
assert np.all(expected_1 == image[1, :, :])
def test_flatten():
image = build_image('0222112222120000', x=2, y=2)
render = flatten(image)
expected = np.array([[0, 1], [1, 0]])
assert np.all(render == expected)
def test_3():
image = build_image(test_input, x=3, y=2)
layer = find_densest_layer(image)
checksum = calc_checksum(image, layer)
assert checksum == 1
def test_2():
image = build_image(test_input, x=3, y=2)
layer = find_densest_layer(image)
assert layer == 0