1 files changed, 57 insertions, 0 deletions
diff --git a/07/test.py b/07/test.py
new file mode 100755
index 0000000..a171571
--- /dev/null
+++ b/07/test.py
@@ -0,0 +1,57 @@
+#!/usr/bin/python
+
+import unittest
+
+import main
+
+
+
+class ExamplesPart1(unittest.TestCase):
+    def setUp(self):
+        self.lines = [
+            "123 -> x",
+            "456 -> y",
+            "x AND y -> d",
+            "x OR y -> e",
+            "x LSHIFT 2 -> f",
+            "y RSHIFT 2 -> g",
+            "NOT x -> h",
+            "NOT y -> i"
+        ]
+
+    def test_example1(self):
+        lines = "\n".join(self.lines).replace("d", "a").splitlines()
+        self.assertEqual(main.solve_partn(1, lines),    72)
+
+    def test_example2(self):
+        lines = "\n".join(self.lines).replace("e", "a").splitlines()
+        self.assertEqual(main.solve_partn(1, lines),   507)
+
+    def test_example3(self):
+        lines = "\n".join(self.lines).replace("f", "a").splitlines()
+        self.assertEqual(main.solve_partn(1, lines),   492)
+
+    def test_example4(self):
+        lines = "\n".join(self.lines).replace("g", "a").splitlines()
+        self.assertEqual(main.solve_partn(1, lines),   114)
+
+    def test_example5(self):
+        lines = "\n".join(self.lines).replace("h", "a").splitlines()
+        self.assertEqual(main.solve_partn(1, lines), 65412)
+
+    def test_example6(self):
+        lines = "\n".join(self.lines).replace("i", "a").splitlines()
+        self.assertEqual(main.solve_partn(1, lines), 65079)
+
+    def test_example7(self):
+        lines = "\n".join(self.lines).replace("x", "a").splitlines()
+        self.assertEqual(main.solve_partn(1, lines),   123)
+
+    def test_example8(self):
+        lines = "\n".join(self.lines).replace("y", "a").splitlines()
+        self.assertEqual(main.solve_partn(1, lines),   456)
+
+
+
+if __name__ == "__main__":
+    unittest.main()