import unittest from sources.check_bracket import check_bracket class MyTestCase(unittest.TestCase): def test1(self): exp = "[]" self.assertEqual("Success", check_bracket(exp)) def test2(self): exp = "{}[]" self.assertEqual("Success", check_bracket(exp)) def test3(self): exp = "[()]" self.assertEqual("Success", check_bracket(exp)) def test4(self): exp = "(())" self.assertEqual("Success", check_bracket(exp)) def test5(self): exp = "{[]}()" self.assertEqual("Success", check_bracket(exp)) def test6(self): exp = "{" self.assertEqual(1, check_bracket(exp)) def test7(self): exp = "{[}" self.assertEqual(3, check_bracket(exp)) def test8(self): exp = "foo(bar)" self.assertEqual("Success", check_bracket(exp)) def test9(self): exp = "foo(bar[i);" self.assertEqual(10, check_bracket(exp)) def test10(self): exp = "[](()" self.assertEqual(3, check_bracket(exp)) def test11(self): exp = "{[}haha" self.assertEqual(3, check_bracket(exp)) def test12(self): exp = "haha{[}" self.assertEqual(7, check_bracket(exp)) if __name__ == '__main__': unittest.main()