Exemple #1
0
    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
Exemple #2
0
 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")