Python SparseMultiGraph.getNeighboursByEdgeType示例

示例#1

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

    def testGetNeighboursByEdgeType(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(sMultiGraph.getNeighboursByEdgeType(0, 1), [3])
        self.assertEquals(sMultiGraph.getNeighboursByEdgeType(0, 2), [1])
        self.assertEquals(sMultiGraph.getNeighboursByEdgeType(0, 0), [])
        self.assertEquals(set(sMultiGraph.getNeighboursByEdgeType(4, 0)), set([1, 9]))

示例#2

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

    def testGetNeighboursByEdgeType(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(sMultiGraph.getNeighboursByEdgeType(0, 1), [3])
        self.assertEquals(sMultiGraph.getNeighboursByEdgeType(0, 2), [1])
        self.assertEquals(sMultiGraph.getNeighboursByEdgeType(0, 0), [])
        self.assertEquals(set(sMultiGraph.getNeighboursByEdgeType(4, 0)),
                          set([1, 9]))