From 8cd56b443beeaf80e14a1c1783b1a3d59b28f4e4 Mon Sep 17 00:00:00 2001
From: ageorge <anjith.george@idiap.ch>
Date: Thu, 31 Jan 2019 17:05:16 +0100
Subject: [PATCH] Added unit tests for protocols in BATL HLDI
---
bob/pad/face/test/test_databases.py | 74 ++++++++++++++++++++++++++++-
1 file changed, 73 insertions(+), 1 deletion(-)
diff --git a/bob/pad/face/test/test_databases.py b/bob/pad/face/test/test_databases.py
index eebd65c2..a5af55b7 100644
--- a/bob/pad/face/test/test_databases.py
+++ b/bob/pad/face/test/test_databases.py
@@ -223,7 +223,7 @@ def test_casiasurf():
# Test the BATL database
def test_aggregated_db():
batl_db = bob.bio.base.load_resource(
- 'batl',
+ 'batl-db',
'database',
preferred_package='bob.pad.face',
package_prefix='bob.pad.')
@@ -274,6 +274,78 @@ def test_aggregated_db():
batl_db.objects(groups='eval',
protocol='grandtest-color-50-LOO_fakehead')) == 132
+ # test for LOO_flexiblemask
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev', 'eval'],
+ protocol='grandtest-color-50-LOO_flexiblemask')) == 1132
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev'], protocol='grandtest-color-50-LOO_flexiblemask')) == 880
+ assert len(
+ batl_db.objects(groups='eval',
+ protocol='grandtest-color-50-LOO_flexiblemask')) == 252
+
+ # test for LOO_glasses
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev', 'eval'],
+ protocol='grandtest-color-50-LOO_glasses')) == 1206
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev'], protocol='grandtest-color-50-LOO_glasses')) == 1069
+ assert len(
+ batl_db.objects(groups='eval',
+ protocol='grandtest-color-50-LOO_glasses')) == 137
+
+ # test for LOO_papermask
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev', 'eval'],
+ protocol='grandtest-color-50-LOO_papermask')) == 1308
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev'], protocol='grandtest-color-50-LOO_papermask')) == 1122
+ assert len(
+ batl_db.objects(groups='eval',
+ protocol='grandtest-color-50-LOO_papermask')) == 186
+
+ # test for LOO_prints
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev', 'eval'],
+ protocol='grandtest-color-50-LOO_prints')) == 1169
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev'], protocol='grandtest-color-50-LOO_prints')) == 988
+ assert len(
+ batl_db.objects(groups='eval',
+ protocol='grandtest-color-50-LOO_prints')) == 181
+
+ # test for LOO_replay
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev', 'eval'],
+ protocol='grandtest-color-50-LOO_replay')) == 1049
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev'], protocol='grandtest-color-50-LOO_replay')) == 854
+ assert len(
+ batl_db.objects(groups='eval',
+ protocol='grandtest-color-50-LOO_replay')) == 195
+
+ # test for LOO_rigidmask
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev', 'eval'],
+ protocol='grandtest-color-50-LOO_rigidmask')) == 1198
+ assert len(
+ batl_db.objects(
+ groups=['train', 'dev'], protocol='grandtest-color-50-LOO_rigidmask')) == 1034
+ assert len(
+ batl_db.objects(groups='eval',
+ protocol='grandtest-color-50-LOO_rigidmask')) == 164
+
except IOError as e:
raise SkipTest(
--
GitLab