from pathlib import Path

from skimage.io import imread, imsave

from falign.landmarks import read_image_and_get_landmarks
from falign.align import align


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

    image, landmarks = read_image_and_get_landmarks(path_image)
    aligned = align(image, landmarks, height=256, width=256)
    assert aligned.shape == (256, 256, 3)

    imsave(dir_gallery / "test_align.jpg", aligned)