def populateGraph(self, text):
from textProcessor.core.tokenCore import Token
from textProcessor.core.graphCore import Graph
tokens = Token().createTokens(text.lower())
graph = Graph()
"""
Create the Graph
"""
for count in range(0, len(tokens)):
nFrom = tokens[count]
graph.addVertex(Token().removePontuation(nFrom))
graph.addWord(Token().removePontuation(nFrom))
for c in nFrom:
graph.addCharacter(c)
if count + 1 < len(tokens):
nTo = tokens[count + 1]
if nFrom[-1] not in (self.__endPontuation) and nFrom[0] not in (self.__endPontuation):
graph.createEdge(Token().removePontuation(nFrom), Token().removePontuation(nTo))
"""
3-grams
"""
for count in range(0, len(tokens)):
if count + 2 < len(tokens):
graph.createNGram((tokens[count], tokens[count + 1], tokens[count + 2]))
return graph
def testGraphCreateEdge(self):
from textProcessor.core.graphCore import Graph
graph = Graph()
graph.addVertex("um")
graph.addVertex("dois")
graph.createEdge(nodeFrom="um", nodeTo="dois")
self.assertEqual(graph.getEdges(),{("um","dois"):1}, "An error ocurred during the creation of an edge")