def testSimulateCircularPool(self):
model = Model()
model.setParameters(100, 0.1)
model.setStartingState(0, 0, 2)
model.simulate(100)
assert model.eventQueue.time == 100
assert model.pool.size() >= 1 and model.pool.size() <= 102
def testSimulateMixedPool(self):
model = Model()
model.setParameters(10000, 0.000001)
model.setStartingState(0, 100, 0)
model.simulate(100000)
assert model.eventQueue.time == 100000
assert model.pool.size() >= 1 and model.pool.size() <= 1000
def testSimulateAllParams(self):
model = Model()
model.setParameters(10000, 0.00001, 0.00001, 0.00001, 0.1, 0.1, 0.5)
model.setStartingState(100, 30, 30)
model.simulate(100000)
def testModelInit(self):
model = Model()
assert model.pool.size() == 0
model.setParameters(100, 0.1)
model.setStartingState(0, 21, 3)
assert model.pool.size() == 24
assert model.pool.weight() == 100
linCount = 0
cirCount = 0
for contig in model.pool.dataElements():
if type(contig) == LinearContig:
linCount += 1
elif type(contig) == CircularContig:
cirCount += 1
else:
assert False
assert linCount == 21
assert cirCount == 3
model.setStartingState(1, 3, 0)
assert model.pool.size() == 4
assert model.pool.weight() == 100
model.setParameters(55, 0.1)
model.setStartingState(11, 0, 4)
assert model.pool.size() == 5
assert model.pool.weight() == 55
