Initial commit

1 files changed, 39 insertions, 0 deletions

diff --git a/19/test.py b/19/test.py
new file mode 100755
index 0000000..f2fab39
--- /dev/null
+++ b/19/test.py
@@ -0,0 +1,39 @@
+#!/usr/bin/python
+
+import unittest
+
+import main
+
+
+
+class Examples(unittest.TestCase):
+    def test_example1(self):
+        lines = [
+            "e => H",
+            "e => O",
+            "H => HO",
+            "H => OH",
+            "O => HH",
+            "",
+            "HOH"
+        ]
+        self.assertEqual(main.solve_part1(lines), 4)
+        self.assertEqual(main.solve_part2(lines), 3)
+
+    def test_example2(self):
+        lines = [
+            "e => H",
+            "e => O",
+            "H => HO",
+            "H => OH",
+            "O => HH",
+            "",
+            "HOHOHO"
+        ]
+        self.assertEqual(main.solve_part1(lines), 7)
+        self.assertEqual(main.solve_part2(lines), 6)
+
+
+
+if __name__ == "__main__":
+    unittest.main()