def get_face_image(face):
face_images = get_pinned_face_images(face)
if len(face_images) == 0:
week_of_the_year = get_week_of_the_year()
image_count = face.images.all().count()
image_index = week_of_the_year % image_count
return list(face.images.all())[image_index]
else:
return face_images[0]
def test_get_pinned_face_images(self):
face_image1 = self.create_face_image(is_pinned=True)
face_image2 = self.create_face_image(is_pinned=True)
face_image3 = self.create_face_image(is_pinned=False)
face = face_image1.face
face.images.add(face_image2)
face.images.add(face_image3)
self.assertEqual(2, len(get_pinned_face_images(face_image2.face)))
