def compareTrees(self, origFile, newFileFH): origXML = project.readTiger(origFile).next() newXML = project.readTiger(newFileFH).next() origTree = project.tigerToTree(origXML) newTree = project.tigerToTree(newXML) # get sentences from xml for (origNode, newNode) in itertools.izip_longest( shared.ma_util.walkTree(origTree), shared.ma_util.walkTree(newTree), fillvalue="LIST_LENGTH_NOT_EQUAL"): print "comparing: %s, %s" % (origNode, newNode) print "comparing values: %s, %s" % (origNode.node, newNode.node) self.assertEqual(origNode, newNode) self.assertEqual(origNode.node, newNode.node)
def test2(self): inputFile = "tests/1.xml" inputFile = th.collapseToTmp(inputFile) print "Unary-collapsed tree at %s" % inputFile self.maxDiff = None penn = shared.ma_util.readPenn("tests/1-binarization-yv.ptb").next() tiger = project.readTiger(inputFile).next() res = project.getMappingFromNodeIDToSentiment(tiger, penn) expected = yaml.load(open("tests/1.newcollapse.expected.yml")) self.assertEqual(self._mapToString(res), self._mapToString(expected))
def testReadTiger(self): count = 0 for sentence in project.readTiger(open("tests/1.xml")): count += 1 self.assertEqual(count, 1)