def testAddVertices(self):
numFeatures = 5
vList = VertexList(10, numFeatures)
vList.setVertex(1, numpy.ones(numFeatures)*0.1)
self.assertEquals(vList.getNumVertices(), 10)
nptst.assert_array_equal(vList[1], numpy.ones(numFeatures)*0.1)
vList.addVertices(5)
self.assertEquals(vList.getNumVertices(), 15)
vList.setVertex(11, numpy.ones(numFeatures)*2)
nptst.assert_array_equal(vList[1], numpy.ones(numFeatures)*0.1)
nptst.assert_array_equal(vList[11], numpy.ones(numFeatures)*2)
def testAddVertices(self):
numFeatures = 5
vList = VertexList(10, numFeatures)
vList.setVertex(1, numpy.ones(numFeatures) * 0.1)
self.assertEquals(vList.getNumVertices(), 10)
nptst.assert_array_equal(vList[1], numpy.ones(numFeatures) * 0.1)
vList.addVertices(5)
self.assertEquals(vList.getNumVertices(), 15)
vList.setVertex(11, numpy.ones(numFeatures) * 2)
nptst.assert_array_equal(vList[1], numpy.ones(numFeatures) * 0.1)
nptst.assert_array_equal(vList[11], numpy.ones(numFeatures) * 2)
class VertexListTest(unittest.TestCase, AbstractVertexListTest):
def setUp(self):
self.numVertices = 10
self.numFeatures = 3
self.vList = VertexList(self.numVertices, self.numFeatures)
self.emptyVertex = numpy.zeros(self.numFeatures)
self.initialise()
def testConstructor(self):
self.assertEquals(self.vList.getNumFeatures(), self.numFeatures)
self.assertEquals(self.vList.getNumVertices(), self.numVertices)
def testSaveLoad(self):
try:
vList = VertexList(self.numVertices, self.numFeatures)
vList.setVertex(0, numpy.array([1, 2, 3]))
vList.setVertex(1, numpy.array([4, 5, 6]))
vList.setVertex(2, numpy.array([7, 8, 9]))
tempDir = PathDefaults.getTempDir()
fileName = tempDir + "vList"
vList.save(fileName)
vList2 = VertexList.load(fileName)
self.assertTrue(
(vList.getVertices() == vList2.getVertices()).all())
except IOError as e:
logging.warn(e)
pass
def testGetItem2(self):
V = numpy.random.rand(self.numVertices, self.numFeatures)
self.vList.setVertices(V)
for i in range(self.numVertices):
nptst.assert_array_equal(self.vList[i, :], V[i, :])
def testSetItem2(self):
V = numpy.random.rand(self.numVertices, self.numFeatures)
for i in range(self.numVertices):
self.vList[i, :] = V[i, :]
nptst.assert_array_equal(self.vList[i, :], V[i, :])
def testAddVertices(self):
numFeatures = 5
vList = VertexList(10, numFeatures)
vList.setVertex(1, numpy.ones(numFeatures) * 0.1)
self.assertEquals(vList.getNumVertices(), 10)
nptst.assert_array_equal(vList[1], numpy.ones(numFeatures) * 0.1)
vList.addVertices(5)
self.assertEquals(vList.getNumVertices(), 15)
vList.setVertex(11, numpy.ones(numFeatures) * 2)
nptst.assert_array_equal(vList[1], numpy.ones(numFeatures) * 0.1)
nptst.assert_array_equal(vList[11], numpy.ones(numFeatures) * 2)
class VertexListTest(unittest.TestCase, AbstractVertexListTest):
def setUp(self):
self.numVertices = 10
self.numFeatures = 3
self.vList = VertexList(self.numVertices, self.numFeatures)
self.emptyVertex = numpy.zeros(self.numFeatures)
self.initialise()
def testConstructor(self):
self.assertEquals(self.vList.getNumFeatures(), self.numFeatures)
self.assertEquals(self.vList.getNumVertices(), self.numVertices)
def testSaveLoad(self):
try:
vList = VertexList(self.numVertices, self.numFeatures)
vList.setVertex(0, numpy.array([1, 2, 3]))
vList.setVertex(1, numpy.array([4, 5, 6]))
vList.setVertex(2, numpy.array([7, 8, 9]))
tempDir = PathDefaults.getTempDir()
fileName = tempDir + "vList"
vList.save(fileName)
vList2 = VertexList.load(fileName)
self.assertTrue((vList.getVertices() == vList2.getVertices()).all())
except IOError as e:
logging.warn(e)
pass
def testGetItem2(self):
V = numpy.random.rand(self.numVertices, self.numFeatures)
self.vList.setVertices(V)
for i in range(self.numVertices):
nptst.assert_array_equal(self.vList[i, :], V[i, :])
def testSetItem2(self):
V = numpy.random.rand(self.numVertices, self.numFeatures)
for i in range(self.numVertices):
self.vList[i, :] = V[i, :]
nptst.assert_array_equal(self.vList[i, :], V[i, :])
def testAddVertices(self):
numFeatures = 5
vList = VertexList(10, numFeatures)
vList.setVertex(1, numpy.ones(numFeatures)*0.1)
self.assertEquals(vList.getNumVertices(), 10)
nptst.assert_array_equal(vList[1], numpy.ones(numFeatures)*0.1)
vList.addVertices(5)
self.assertEquals(vList.getNumVertices(), 15)
vList.setVertex(11, numpy.ones(numFeatures)*2)
nptst.assert_array_equal(vList[1], numpy.ones(numFeatures)*0.1)
nptst.assert_array_equal(vList[11], numpy.ones(numFeatures)*2)
