Python SparseMultiGraph.neighbours示例

示例#1

文件： SparseMultiGraphTest.py 项目： awj223/Insight-Data-Engineering-Code-Challenge

    def testGetNeighbours(self):
        sMultiGraph = SparseMultiGraph(self.vList, self.maxEdgeTypes, True)

        sMultiGraph.addEdge(0, 1, 2)
        sMultiGraph.addEdge(0, 3, 1)
        sMultiGraph.addEdge(2, 1, 0)
        sMultiGraph.addEdge(1, 4, 0)
        sMultiGraph.addEdge(9, 4, 0)
        sMultiGraph.addEdge(9, 4, 1)

        self.assertEquals(set(sMultiGraph.neighbours(0)), set([1, 3]))
        self.assertEquals(set(sMultiGraph.neighbours(1)), set([0, 2, 4]))
        self.assertEquals(sMultiGraph.neighbours(3), [0])
        self.assertEquals(set(sMultiGraph.neighbours(4)), set([9, 1]))
        self.assertEquals(sMultiGraph.neighbours(9), [4])

示例#2

文件： SparseMultiGraphTest.py 项目： spencer-ortega/DynamicGraph-AnomalyDetection

    def testGetNeighbours(self):
        sMultiGraph = SparseMultiGraph(self.vList, self.maxEdgeTypes, True)

        sMultiGraph.addEdge(0, 1, 2)
        sMultiGraph.addEdge(0, 3, 1)
        sMultiGraph.addEdge(2, 1, 0)
        sMultiGraph.addEdge(1, 4, 0)
        sMultiGraph.addEdge(9, 4, 0)
        sMultiGraph.addEdge(9, 4, 1)

        self.assertEquals(set(sMultiGraph.neighbours(0)), set([1, 3]))
        self.assertEquals(set(sMultiGraph.neighbours(1)), set([0, 2, 4]))
        self.assertEquals(sMultiGraph.neighbours(3), [0])
        self.assertEquals(set(sMultiGraph.neighbours(4)), set([9, 1]))
        self.assertEquals(sMultiGraph.neighbours(9), [4])