def testGenerateReceivers(self):
eCsvReader = EgoCsvReader()
numAlters = 10
numPartialFields = 11
alterFields = numPartialFields + 2
alterFieldIndices = list(range(0, numPartialFields))
egoAlterArray = numpy.zeros((2, 45))
alterArray = numpy.zeros((numAlters, alterFields))
egoAlterArray[0, 0:11] = [2, 12, 1, 0, 0, 0, 0, 0, 0, 0, 1]
egoAlterArray[0, 11:22] = [1, 11, 1, 0, 0, 0, 0, 0, 0, 0, 1]
egoAlterArray[1, 0:11] = [5, 6, 1, 0, 0, 0, 0, 0, 0, 0, 3]
alterArray[0, :] = [2, 12, 1, 0, 0, 0, 0, 0, 0, 0, 1, 12, 13]
alterArray[1, :] = [1, 11, 1, 0, 0, 0, 0, 0, 0, 0, 1, 84, 12]
alterArray[2, :] = [2, 12, 1, 0, 0, 0, 1, 0, 0, 0, 0, 4, 9]
alterArray[4, :] = [5, 6, 1, 0, 0, 0, 0, 0, 0, 0, 3, 34, 12]
generatedAltersArray2 = numpy.zeros((3, alterFields))
generatedAltersArray2[0, :] = alterArray[0, :]
generatedAltersArray2[1, :] = alterArray[1, :]
generatedAltersArray2[2, :] = alterArray[4, :]
(generatedAltersArray, egoIndices, alterIndices) = eCsvReader.generateReceivers(egoAlterArray, alterArray, alterFieldIndices)
self.assertTrue((egoIndices == numpy.array([0,0,1])).all())
self.assertTrue((alterIndices == numpy.array([0,1,4])).all())
self.assertTrue((generatedAltersArray == generatedAltersArray2).all())
#2nd test
egoAlterArray[1, 11:22] = [2, 12, 1, 0, 0, 0, 1, 0, 0, 0, 0]
(generatedAltersArray, egoIndices, alterIndices) = eCsvReader.generateReceivers(egoAlterArray, alterArray, alterFieldIndices)
generatedAltersArray2 = numpy.zeros((4, alterFields))
generatedAltersArray2[0, :] = alterArray[0, :]
generatedAltersArray2[1, :] = alterArray[1, :]
generatedAltersArray2[2, :] = alterArray[4, :]
generatedAltersArray2[3, :] = alterArray[2, :]
self.assertTrue((egoIndices == numpy.array([0,0,1,1])).all())
self.assertTrue((alterIndices == numpy.array([0,1,4,2])).all())
self.assertTrue((generatedAltersArray == generatedAltersArray2).all())
