def setUp(self):
data = MockarooResponse(users=TEST_DATA)
data.create_test_users()
data.assign_mentees()
all_users = KhanUser.objects.all()
self.relevant_users = remove_outliers(all_users)
self.origin_user = KhanUser.objects.get(id=12)
self.infection_result = InfectionResult(original_user_infected=self.origin_user, users=all_users, networked_users=self.relevant_users)
def test_users_who_are_outliers(self):
outliers = self.infection_result.users_who_are_outliers
outlier_users = [u.id for u in outliers]
outlier_ids = [4,10]
all_users = KhanUser.objects.all()
relevant_users = remove_outliers(all_users)
r_users = [u.id for u in relevant_users]
user_ids = [1,2,3,5,6,7,8,9,11,12]
self.assertEqual(r_users, user_ids)
self.assertEqual(outlier_users, outlier_ids)
