2022 Day 06

2022/06/test_code.py

@@ -0,0 +1,18 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2022 Akumatic
+
+from code import part1, part2
+
+def test():
+    inputs = ["mjqjpqmgbljsphdztnvjfqwrcgsmlb", "bvwbjplbgvbhsrlpgdmjqwftvncz", 
+    "nppdvjthqldpwncqszvftbrmjlhg", "nznrnfrfntjfmvfwmzdfjlvtqnbhcprsg", "zcfzfwzzqfrljwzlrfnpqdbhtmscgvjw"]
+    expected_values = [(7, 19), (5, 23), (6, 23), (10, 29), (11, 26)]
+    for i in range(len(inputs)):
+        assert part1(inputs[i]) == expected_values[i][0]
+    print("Passed Part 1")
+    for i in range(len(inputs)):
+        assert part2(inputs[i]) == expected_values[i][1]
+    print("Passed Part 2")
+
+if __name__ == "__main__":
+    test()