Initial commit

1 files changed, 32 insertions, 0 deletions

diff --git a/05/test.py b/05/test.py
new file mode 100755
index 0000000..66fc927
--- /dev/null
+++ b/05/test.py
@@ -0,0 +1,32 @@
+#!/usr/bin/python
+
+import unittest
+
+import main
+
+
+
+class ExamplesPart1(unittest.TestCase):
+    def test_example1(self):
+        lines = ["ugknbfddgicrmopn", "aaa"]
+        self.assertEqual(main.solve_part1(lines), 2)
+
+    def test_example2(self):
+        lines = ["jchzalrnumimnmhp", "haegwjzuvuyypxyu", "dvszwmarrgswjxmb"]
+        self.assertEqual(main.solve_part1(lines), 0)
+
+
+
+class ExamplesPart2(unittest.TestCase):
+    def test_example1(self):
+        lines = ["qjhvhtzxzqqjkmpb", "xxyxx"]
+        self.assertEqual(main.solve_part2(lines), 2)
+
+    def test_example2(self):
+        lines = ["uurcxstgmygtbstg", "ieodomkazucvgmuy"]
+        self.assertEqual(main.solve_part2(lines), 0)
+
+
+
+if __name__ == "__main__":
+    unittest.main()