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)

test/test_falign/test_landmarks.py

@@ -2,19 +2,19 @@ from pathlib import Path
 
 from skimage.io import imread
 
-from falign.landmarks import get_landmarks
+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"
-    landmarks = get_landmarks(path_image)
 
+    image, landmarks = read_image_and_get_landmarks(path_image)
     assert landmarks.shape == (68, 2)
 
     imsave_with_landmarks(
         dir_gallery / "test_get_landmarks.jpg",
-        imread(path_image), landmarks,
+        image, landmarks,
         size=5
     )