#!/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()