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)