summaryrefslogtreecommitdiff
path: root/01/test.py
diff options
context:
space:
mode:
Diffstat (limited to '01/test.py')
-rwxr-xr-x01/test.py60
1 files changed, 60 insertions, 0 deletions
diff --git a/01/test.py b/01/test.py
new file mode 100755
index 0000000..98cbd7d
--- /dev/null
+++ b/01/test.py
@@ -0,0 +1,60 @@
+#!/usr/bin/python
+
+import unittest
+
+import main
+
+
+
+class ExamplesPart1(unittest.TestCase):
+ def test_example1(self):
+ data = "(())"
+ self.assertEqual(main.solve_partn(1, data), 0)
+
+ def test_example2(self):
+ data = "()()"
+ self.assertEqual(main.solve_partn(1, data), 0)
+
+ def test_example3(self):
+ data = "((("
+ self.assertEqual(main.solve_partn(1, data), 3)
+
+ def test_example4(self):
+ data = "(()(()("
+ self.assertEqual(main.solve_partn(1, data), 3)
+
+ def test_example5(self):
+ data = "))((((("
+ self.assertEqual(main.solve_partn(1, data), 3)
+
+ def test_example6(self):
+ data = "())"
+ self.assertEqual(main.solve_partn(1, data), -1)
+
+ def test_example7(self):
+ data = "))("
+ self.assertEqual(main.solve_partn(1, data), -1)
+
+ def test_example8(self):
+ data = ")))"
+ self.assertEqual(main.solve_partn(1, data), -3)
+
+ def test_example9(self):
+ data = ")())())"
+ self.assertEqual(main.solve_partn(1, data), -3)
+
+
+
+class ExamplesPart2(unittest.TestCase):
+ def test_example1(self):
+ data = ")"
+ self.assertEqual(main.solve_partn(2, data), 1)
+
+ def test_example2(self):
+ data = "()())"
+ self.assertEqual(main.solve_partn(2, data), 5)
+
+
+
+if __name__ == "__main__":
+ unittest.main()