Python SparseMultiGraph.getAllEdges示例

示例#1

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

    def testGetAllEdges(self):
        sMultiGraph = SparseMultiGraph(self.vList, self.maxEdgeTypes, False)

        self.assertEquals(sMultiGraph.getAllEdges().shape[0], 0)
        self.assertEquals(sMultiGraph.getAllEdges().shape[1], 3)

        sMultiGraph.addEdge(0, 1, 0)
        sMultiGraph.addEdge(0, 1, 1)
        sMultiGraph.addEdge(0, 2, 0)

        allEdges = numpy.array([[0,1,0], [0,2,0], [0,1,1]])

        self.assertTrue((sMultiGraph.getAllEdges() == allEdges).all())

示例#2

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

    def testGetAllEdges(self):
        sMultiGraph = SparseMultiGraph(self.vList, self.maxEdgeTypes, False)

        self.assertEquals(sMultiGraph.getAllEdges().shape[0], 0)
        self.assertEquals(sMultiGraph.getAllEdges().shape[1], 3)

        sMultiGraph.addEdge(0, 1, 0)
        sMultiGraph.addEdge(0, 1, 1)
        sMultiGraph.addEdge(0, 2, 0)

        allEdges = numpy.array([[0, 1, 0], [0, 2, 0], [0, 1, 1]])

        self.assertTrue((sMultiGraph.getAllEdges() == allEdges).all())