def test_exact_infection1(self):
a = User('A')
b = User('B')
c = User('C')
d = User('D')
e = User('E')
f = User('F')
user_lst = [a, b, c, d]
e.add_student(f)
a.add_coach(b)
b.add_student(c)
d.add_student(a)
self.assertEqual(a.get_status(), 'Clean')
self.assertEqual(b.get_status(), 'Clean')
self.assertEqual(c.get_status(), 'Clean')
self.assertEqual(d.get_status(), 'Clean')
self.assertEqual(e.get_status(), 'Clean')
self.assertEqual(f.get_status(), 'Clean')
result = infections.exact_infection(a, 4)
self.assertTrue(result)
infected_count = 0
for user in user_lst:
infected_count += user.get_status_num()
self.assertEqual(infected_count, 4)
result = infections.exact_infection(f, 10)
self.assertFalse(result)
self.assertEqual(e.get_status(), 'Clean')
self.assertEqual(f.get_status(), 'Clean')
def test_user_creation(self):
u = User('A')
self.assertEqual(u.get_name(), 'A')
self.assertEqual(u.get_status(), 'Clean')
v = User('B')
u.add_student(v)
self.assertTrue(v in u.get_students())
def test_total_infection_rand(self):
users = []
user_num = 50
for i in xrange(user_num):
user = User(i)
users.append(user)
coaching = random.randint(0, 1)
if i:
related_user = users[random.randint(0, i - 1)]
if coaching:
user.add_student(related_user)
else:
user.add_coach(related_user)
for user in users:
self.assertEqual(user.get_status(), 'Clean')
user = users[random.randint(0, user_num - 1)]
infections.total_infection(user)
for user in users:
self.assertEqual(user.get_status(), 'Infected')
def test_limited_infection1(self):
a = User('A')
b = User('B')
c = User('C')
d = User('D')
e = User('E')
f = User('F')
e.add_student(f)
a.add_coach(b)
b.add_student(c)
d.add_student(a)
self.assertEqual(a.get_status(), 'Clean')
self.assertEqual(b.get_status(), 'Clean')
self.assertEqual(c.get_status(), 'Clean')
self.assertEqual(d.get_status(), 'Clean')
self.assertEqual(e.get_status(), 'Clean')
self.assertEqual(f.get_status(), 'Clean')
infections.limited_infection(a, 3)
self.assertEqual(a.get_status(), 'Infected')
self.assertEqual(b.get_status(), 'Infected')
self.assertEqual(c.get_status(), 'Clean')
self.assertEqual(d.get_status(), 'Infected')
self.assertEqual(e.get_status(), 'Clean')
self.assertEqual(f.get_status(), 'Clean')
infections.limited_infection(f, 0)
self.assertEqual(e.get_status(), 'Clean')
self.assertEqual(f.get_status(), 'Clean')
infections.limited_infection(f, 10)
self.assertEqual(e.get_status(), 'Infected')
self.assertEqual(f.get_status(), 'Infected')
def test_total_infection1(self):
a = User()
b = User()
c = User()
d = User()
e = User()
f = User()
e.add_student(f)
a.add_coach(b)
b.add_student(c)
d.add_student(a)
self.assertEqual(a.get_status(), 'Clean')
self.assertEqual(b.get_status(), 'Clean')
self.assertEqual(c.get_status(), 'Clean')
self.assertEqual(d.get_status(), 'Clean')
self.assertEqual(e.get_status(), 'Clean')
self.assertEqual(f.get_status(), 'Clean')
infections.total_infection(a)
self.assertEqual(a.get_status(), 'Infected')
self.assertEqual(b.get_status(), 'Infected')
self.assertEqual(c.get_status(), 'Infected')
self.assertEqual(d.get_status(), 'Infected')
self.assertEqual(e.get_status(), 'Clean')
self.assertEqual(f.get_status(), 'Clean')
infections.total_infection(f)
self.assertEqual(e.get_status(), 'Infected')
self.assertEqual(f.get_status(), 'Infected')
def test_user_infect(self):
a = User()
a.infect()
self.assertEqual(a.get_status(), 'Infected')
b = User(None, a)
self.assertEqual(b.get_status(), 'Infected')
