def testUpperDetectionRates(self): """ See if the upper bound on detection rates is correct """ undirected = True numVertices = 10 graph = HIVGraph(numVertices, undirected) hiddenDegSeq = self.gen.rvs(size=graph.getNumVertices()) rates = HIVRates(graph, hiddenDegSeq) t = 0.1 graph.getVertexList().setInfected(0, t) graph.getVertexList().setInfected(1, t) graph.getVertexList().setInfected(8, t) t = 0.2 rates.removeEvent(8, HIVVertices.randomDetect, t) rates.infectionProbability = 1.0 infectedList = graph.infectedIndsAt(t) removedList = graph.removedIndsAt(t) n = graph.size-removedList self.assertEquals(rates.upperDetectionRates(infectedList, n), rates.randomDetectionRates(infectedList, n, seed=21).sum()) t = 0.3 rates.contactEvent(0, 2, t) graph.vlist.setInfected(2, t) t = 0.4 rates.removeEvent(0, HIVVertices.randomDetect, t) infectedList = graph.infectedIndsAt(t) removedSet = graph.removedIndsAt(t) removedSet = set(removedSet.tolist()) nptst.assert_array_almost_equal(rates.contactTracingRates(infectedList, removedSet, t + rates.ctStartTime + 1), numpy.array([0, rates.ctRatePerPerson])) upperDetectionRates = rates.ctRatePerPerson + rates.randomDetectionRates(infectedList, n, seed=21).sum() self.assertEquals(rates.upperDetectionRates(infectedList, n), upperDetectionRates)
def testRandomDetectionRates(self): undirected = True numVertices = 10 graph = HIVGraph(numVertices, undirected) t = 0.1 graph.getVertexList().setInfected(0, t) hiddenDegSeq = self.gen.rvs(size=graph.getNumVertices()) rates = HIVRates(graph, hiddenDegSeq) infectedList = [0, 2, 9] rdRates = rates.randomDetectionRates(infectedList, float(graph.size - len(graph.getRemovedSet()))) nptst.assert_array_almost_equal(rdRates, numpy.ones(len(infectedList))*rates.randDetectRate*len(infectedList)/float(graph.size - len(graph.getRemovedSet())))