Implement align

test/test_falign/test_align.py

@@ -0,0 +1,17 @@
+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)