def testAddVertices(self): numVertices = 10 vertices = HIVVertices(numVertices) v1 = vertices[4] vertices.addVertices(5) print(vertices.V.shape) self.assertEquals(numVertices + 5, vertices.getNumVertices()) nptst.assert_array_equal(v1, vertices[4])
def testContructor(self): numVertices = 10 vertices = HIVVertices(numVertices) self.assertEquals(numVertices, vertices.getNumVertices()) self.assertEquals(8, vertices.getNumFeatures())