From fc0d08c1cd3f3dc2a8433fa099eb357a8687f3c3 Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Tue, 3 Jul 2018 14:31:06 +0200
Subject: [PATCH] [test] added unit tests for preprocessors
---
bob/pad/face/test/test.py | 52 ++++++++++++++++++++++++++++++++++++++-
1 file changed, 51 insertions(+), 1 deletion(-)
diff --git a/bob/pad/face/test/test.py b/bob/pad/face/test/test.py
index bbcf337a..dee19513 100644
--- a/bob/pad/face/test/test.py
+++ b/bob/pad/face/test/test.py
@@ -28,7 +28,11 @@ from ..extractor import LBPHistogram
from ..extractor import ImageQualityMeasure
-import random
+from ..preprocessor import LiICPR2016
+from ..preprocessor import Chrom
+from ..preprocessor import PPGSecure
+from ..preprocessor import SSR
+
from ..preprocessor.FaceCropAlign import detect_face_landmarks_in_image
@@ -371,3 +375,49 @@ def convert_array_to_list_of_frame_cont(data):
frame_container) # add current frame to FrameContainer
return frame_container_list
+
+
+def test_preprocessor_LiICPR2016():
+ preprocessor = LiICPR2016()
+ data = np.random((10, 3, 10, 10))
+ frames = bob.bio.video.FrameContainer()
+ for i in range(data.shape[0])
+ frames.add(i, data[i])
+ pulse = preprocessor(frames)
+
+ assert pulse.shape == (10, 3)
+ assert np.all(pulse == 0)
+
+
+def test_preprocessor_Chrom():
+ preprocessor = Chrom()
+ data = np.random((10, 3, 10, 10))
+ frames = bob.bio.video.FrameContainer()
+ for i in range(data.shape[0])
+ frames.add(i, data[i])
+ pulse = preprocessor(frames)
+
+ assert pulse.shape[0] == 10
+
+
+def test_preprocessor_PPGSecure():
+ preprocessor = PPGSecure()
+ data = np.random((10, 3, 10, 10))
+ frames = bob.bio.video.FrameContainer()
+ for i in range(data.shape[0])
+ frames.add(i, data[i])
+ pulse = preprocessor(frames)
+
+ assert pulse.shape == (5, 10)
+
+
+def test_preprocessor_SSR():
+ preprocessor = SSR()
+ data = np.random((10, 3, 10, 10))
+ frames = bob.bio.video.FrameContainer()
+ for i in range(data.shape[0])
+ frames.add(i, data[i])
+ pulse = preprocessor(frames)
+
+ assert pulse.shape[0] == 10
+
--
GitLab