2020 Day 15

2020/15/test_code.py

@@ -0,0 +1,26 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2020 Akumatic
+
+from code import part1, part2
+
+def test():
+    assert part1([0, 3, 6]) == 436
+    assert part1([1, 3, 2]) == 1
+    assert part1([2, 1, 3]) == 10
+    assert part1([1, 2, 3]) == 27
+    assert part1([2, 3, 1]) == 78
+    assert part1([3, 2, 1]) == 438
+    assert part1([3, 1, 2]) == 1836
+    print(f"Passed part 1")
+
+    assert part2([0,3,6]) == 175594
+    assert part2([1,3,2]) == 2578
+    assert part2([2,1,3]) == 3544142
+    assert part2([1,2,3]) == 261214
+    assert part2([2,3,1]) == 6895259
+    assert part2([3,2,1]) == 18
+    assert part2([3,1,2]) == 362
+    print(f"Passed part 2")
+
+if __name__ == "__main__":
+    test()