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