def get_group_image(grouped_faces): pinned_faces = get_pinned_faces(grouped_faces["list"][0].first_letter_of_district) if len(pinned_faces) != 0: return get_face_image(pinned_faces[0]) else: return get_group_image_of_the_week(grouped_faces)
def test_get_pinned_faces(self): face1 = self.create_face(is_pinned=True) self.create_face(is_pinned=True) self.create_face(is_pinned=False) self.assertEqual(2, len(get_pinned_faces(face1.first_letter_of_district())))