34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
|
# SPDX-License-Identifier: MIT
|
||
|
|
# Copyright (c) 2021 Akumatic
|
||
|
|
|
||
|
|
from code import *
|
||
|
|
|
||
|
|
def test():
|
||
|
|
scanner = read_file("test_input_1.txt")[0]
|
||
|
|
tranformations = [
|
||
|
|
[[-1, -1, 1], [-2, -2, 2], [-3, -3, 3], [-2, -3, 1], [5, 6, -4], [8, 0, 7]],
|
||
|
|
[[1, -1, 1], [2, -2, 2], [3, -3, 3], [2, -1, 3], [-5, 4, -6], [-8, -7, 0]],
|
||
|
|
[[-1, -1, -1], [-2, -2, -2], [-3, -3, -3], [-1, -3, -2], [4, 6, 5], [-7, 0, 8]],
|
||
|
|
[[1, 1, -1], [2, 2, -2], [3, 3, -3], [1, 3, -2], [-4, -6, 5], [7, 0, 8]],
|
||
|
|
[[1, 1, 1], [2, 2, 2], [3, 3, 3], [3, 1, 2], [-6, -4, -5], [0, 7, -8]]
|
||
|
|
]
|
||
|
|
for transformation in tranformations:
|
||
|
|
assert transformation in scanner.transformations.values()
|
||
|
|
print("Transformation passed")
|
||
|
|
|
||
|
|
scanners = read_file("test_input_2.txt")
|
||
|
|
find_scanner_positions(scanners)
|
||
|
|
assert scanners[0].get_position() == (0, 0, 0)
|
||
|
|
assert scanners[1].get_position() == (68,-1246,-43)
|
||
|
|
assert scanners[2].get_position() == (1105,-1205,1229)
|
||
|
|
assert scanners[3].get_position() == (-92,-2380,-20)
|
||
|
|
print("Passed position finding")
|
||
|
|
|
||
|
|
assert part1(scanners) == 79
|
||
|
|
print("Passed Part 1")
|
||
|
|
assert part2(scanners) == 3621
|
||
|
|
print("Passed Part 2")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test()
|