from pathlib import Path

from skimage.io import imread

from falign.landmarks import read_image_and_get_landmarks
from falign.plot import imsave_with_landmarks


def test_get_landmarks():
    dir_gallery = Path(__file__).parent.parent.parent / "gallery"
    path_image = dir_gallery / "original.jpg"

    image, landmarks = read_image_and_get_landmarks(path_image)
    assert landmarks.shape == (68, 2)

    imsave_with_landmarks(
        dir_gallery / "test_get_landmarks.jpg",
        image, landmarks,
        size=5
    )