def testSolve1(self):
sInput = StringIO.StringIO("<tag1><tag2></tag2></tag1><tag3></tag3>")
sOutput = StringIO.StringIO()
XML.solve(sInput, sOutput)
self.assert_(sOutput.getvalue() == "0\n")
def testSolve2(self):
sInput = StringIO.StringIO("<xml><tag1><tag2></tag2></tag1><tag3></tag3></xml>")
sOutput = StringIO.StringIO()
XML.solve(sInput, sOutput)
self.assert_(sOutput.getvalue() == "Unexpected input encountered. Could not pharse XML.")
def testSolve3(self):
sInput = StringIO.StringIO("<THU><Team><ACRush></ACRush><Jelly></Jelly><Cooly></Cooly></Team><JiaJia><Team><Ahyangyi></Ahyangyi><Dragon></Dragon><Cooly><Amber></Amber></Cooly></Team></JiaJia></THU><Team><Cooly></Cooly></Team>")
sOutput = StringIO.StringIO()
XML.solve(sInput, sOutput)
self.assert_(sOutput.getvalue() == "2\n2\n7\n")
