def test_edge_markovian_empty(self):
n = 10
p = 0.
q = 1.
g = 0.
em = contact.edge_markovian(n, p, q, g)
for i in range(5):
contacts = next(em)
self.assertEqual(len(contacts), 0)
for i,j in contacts:
self.assertNotEqual(i,j)
def test_edge_markovian_empty(self):
n = 10
p = 0.
q = 1.
g = 0.
em = contact.edge_markovian(n, p, q, g)
for i in range(5):
contacts = next(em)
self.assertEqual(len(contacts), 0)
for i, j in contacts:
self.assertNotEqual(i, j)
def test_edge_markovian_alternating(self):
n = 10
p = 1.
q = 1.
g = 0.
em = contact.edge_markovian(n, p, q, g)
for i in range(5):
contacts = next(em)
if i%2 == 0:
self.assertEqual(len(contacts), (n*(n-1))/2)
else:
self.assertEqual(len(contacts), 0)
for i,j in contacts:
self.assertNotEqual(i,j)
def test_edge_markovian_alternating(self):
n = 10
p = 1.
q = 1.
g = 0.
em = contact.edge_markovian(n, p, q, g)
for i in range(5):
contacts = next(em)
if i % 2 == 0:
self.assertEqual(len(contacts), (n * (n - 1)) / 2)
else:
self.assertEqual(len(contacts), 0)
for i, j in contacts:
self.assertNotEqual(i, j)
