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)
