def testVectorStatistics(self):
numFeatures = 1
numVertices = 10
vList = VertexList(numVertices, numFeatures)
graph = SparseGraph(vList)
graph.addEdge(0, 2)
graph.addEdge(0, 1)
growthStatistics = GraphStatistics()
statsDict = growthStatistics.vectorStatistics(graph)
self.assertTrue((statsDict["outDegreeDist"] == numpy.array([7,2,1])).all())
self.assertTrue((statsDict["inDegreeDist"] == numpy.array([7,2,1])).all())
self.assertTrue((statsDict["hopCount"] == numpy.array([10,14,16])).all())
self.assertTrue((statsDict["triangleDist"] == numpy.array([10])).all())
W = graph.getWeightMatrix()
W = (W + W.T)/2
lmbda, V = numpy.linalg.eig(W)
maxEigVector = V[:, numpy.argmax(lmbda)]
lmbda = numpy.flipud(numpy.sort(lmbda[lmbda>0]))
self.assertTrue((statsDict["maxEigVector"] == maxEigVector).all())
self.assertTrue((statsDict["eigenDist"] == lmbda).all())
self.assertTrue((statsDict["componentsDist"] == numpy.array([0, 7, 0, 1])).all())
graph.addEdge(0, 3)
graph.addEdge(0, 4)
graph.addEdge(1, 4)
growthStatistics = GraphStatistics()
statsDict = growthStatistics.vectorStatistics(graph)
self.assertTrue((statsDict["outDegreeDist"] == numpy.array([5,2,2,0,1])).all())
self.assertTrue((statsDict["inDegreeDist"] == numpy.array([5,2,2,0,1])).all())
self.assertTrue((statsDict["hopCount"] == numpy.array([10,20,30])).all())
self.assertTrue((statsDict["triangleDist"] == numpy.array([7, 0, 3])).all())
W = graph.getWeightMatrix()
W = (W + W.T)/2
lmbda, V = numpy.linalg.eig(W)
maxEigVector = V[:, numpy.argmax(lmbda)]
lmbda = numpy.flipud(numpy.sort(lmbda[lmbda>0]))
self.assertTrue((statsDict["maxEigVector"] == maxEigVector).all())
self.assertTrue((statsDict["eigenDist"] == lmbda).all())
self.assertTrue((statsDict["componentsDist"] == numpy.array([0, 5, 0, 0, 0, 1])).all())
#Test on a directed graph and generating tree statistics
vList = VertexList(numVertices, numFeatures)
graph = SparseGraph(vList, False)
graph.addEdge(0, 1)
graph.addEdge(0, 2)
graph.addEdge(2, 3)
graph.addEdge(4, 5)
statsDict = growthStatistics.vectorStatistics(graph, treeStats=True)
self.assertTrue(( statsDict["inDegreeDist"] == numpy.array([6, 4]) ).all())
self.assertTrue(( statsDict["outDegreeDist"] == numpy.array([7, 2, 1]) ).all())
self.assertTrue(( statsDict["triangleDist"] == numpy.array([10]) ).all())
self.assertTrue(( statsDict["treeSizesDist"] == numpy.array([0, 4, 1, 0, 1]) ).all())
self.assertTrue(( statsDict["treeDepthsDist"] == numpy.array([4, 1, 1]) ).all())
