From d3859cf5034afd94ba69d47e7624ffa84c089d66 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Fri, 16 Sep 2016 11:09:44 +0200
Subject: [PATCH] [refactoring2016] Added more tests
---
bob/bio/face/test/test_databases.py | 127 +++++++++++++++-------------
1 file changed, 67 insertions(+), 60 deletions(-)
diff --git a/bob/bio/face/test/test_databases.py b/bob/bio/face/test/test_databases.py
index d6dacb9f..f1928301 100644
--- a/bob/bio/face/test/test_databases.py
+++ b/bob/bio/face/test/test_databases.py
@@ -18,47 +18,11 @@
# along with this program. If not, see <http://www.gnu.org/licenses/>.
-import os
from nose.plugins.skip import SkipTest
import bob.bio.base
from bob.bio.base.test.utils import db_available
-
-
-def _check_database(database, groups=('dev',), protocol=None, training_depends=False, models_depend=False):
- assert isinstance(database, bob.bio.base.database.BioDatabase)
-
- # load the directories
- if 'HOME' in os.environ:
- database.replace_directories(os.path.join(os.environ['HOME'], '.bob_bio_databases.txt'))
-
- if protocol: database.protocol = protocol
- if protocol is None: protocol = database.protocol
-
- assert len(database.all_files()) > 0
- assert len(database.training_files('train_extractor')) > 0
- assert len(database.arrange_by_client(database.training_files('train_enroller'))) > 0
-
- for group in groups:
- model_ids = database.model_ids_with_protocol(group, protocol=protocol)
- assert len(model_ids) > 0
- assert database.client_id_from_model_id(model_ids[0]) is not None
- assert len(database.enroll_files(model_ids[0], group)) > 0
- assert len(database.probe_files(model_ids[0], group)) > 0
-
- assert database.training_depends_on_protocol == training_depends
- assert database.models_depend_on_protocol == models_depend
-
-
-def _check_database_zt(database, groups=('dev', 'eval'), protocol=None, training_depends=False, models_depend=False):
- _check_database(database, groups, protocol, training_depends, models_depend)
- assert isinstance(database, bob.bio.base.database.ZTBioDatabase)
- for group in groups:
- t_model_ids = database.t_model_ids(group)
- assert len(t_model_ids) > 0
- assert database.client_id_from_model_id(t_model_ids[0]) is not None
- assert len(database.t_enroll_files(t_model_ids[0], group)) > 0
- assert len(database.z_probe_files(group)) > 0
+from bob.bio.base.test.test_database_implementations import check_database, check_database_zt
def _check_annotations(database):
@@ -74,7 +38,18 @@ def _check_annotations(database):
def test_arface():
database = bob.bio.base.load_resource('arface', 'database', preferred_package='bob.bio.face')
try:
- _check_database(database, groups=('dev', 'eval'))
+ check_database(database, groups=('dev', 'eval'))
+ _check_annotations(database)
+ except IOError as e:
+ raise SkipTest(
+ "The database could not queried; probably the db.sql3 file is missing. Here is the error: '%s'" % e)
+
+
+@db_available('atnt')
+def test_atnt():
+ database = bob.bio.base.load_resource('atnt', 'database', preferred_package='bob.bio.face')
+ try:
+ check_database(database)
_check_annotations(database)
except IOError as e:
raise SkipTest(
@@ -85,7 +60,7 @@ def test_arface():
def test_banca():
database = bob.bio.base.load_resource('banca', 'database', preferred_package='bob.bio.face')
try:
- _check_database_zt(database)
+ check_database_zt(database)
_check_annotations(database)
except IOError as e:
raise SkipTest(
@@ -96,8 +71,8 @@ def test_banca():
def test_caspeal():
database = bob.bio.base.load_resource('caspeal', 'database', preferred_package='bob.bio.face')
try:
- _check_database(database)
- _check_database(database, protocol='aging')
+ check_database(database)
+ check_database(database, protocol='aging')
_check_annotations(database)
except IOError as e:
raise SkipTest(
@@ -116,14 +91,35 @@ def test_caspeal():
# raise SkipTest(
# "The database could not be opened, probably the original directory is wrong. Here is the error: '%s'" % e)
+@db_available('cuhk_cufs')
+def test_cuhk_cufs():
+ database = bob.bio.base.load_resource('cuhk_cufs', 'database', preferred_package='bob.bio.face')
+ try:
+ check_database(database)
+ _check_annotations(database)
+ except IOError as e:
+ raise SkipTest(
+ "The database could not queried; probably the db.sql3 file is missing. Here is the error: '%s'" % e)
+
@db_available('gbu')
def test_gbu():
database = bob.bio.base.load_resource('gbu', 'database', preferred_package='bob.bio.face')
try:
- _check_database(database, models_depend=True)
- _check_database(database, protocol='Bad', models_depend=True)
- _check_database(database, protocol = 'Ugly', models_depend=True)
+ check_database(database, models_depend=True)
+ check_database(database, protocol='Bad', models_depend=True)
+ check_database(database, protocol = 'Ugly', models_depend=True)
+ _check_annotations(database)
+ except IOError as e:
+ raise SkipTest(
+ "The database could not queried; probably the db.sql3 file is missing. Here is the error: '%s'" % e)
+
+
+@db_available('ijba')
+def test_ijba():
+ database = bob.bio.base.load_resource('ijba', 'database', preferred_package='bob.bio.face')
+ try:
+ check_database(database)
_check_annotations(database)
except IOError as e:
raise SkipTest(
@@ -134,10 +130,10 @@ def test_gbu():
def test_lfw():
database = bob.bio.base.load_resource('lfw-restricted', 'database', preferred_package='bob.bio.face')
try:
- _check_database(database, training_depends=True, models_depend=True)
- _check_database(database, groups=('dev', 'eval'), protocol='fold1', training_depends=True, models_depend=True)
- _check_database(bob.bio.base.load_resource('lfw-unrestricted', 'database', preferred_package='bob.bio.face'),
- training_depends=True, models_depend=True)
+ check_database(database, training_depends=True, models_depend=True)
+ check_database(database, groups=('dev', 'eval'), protocol='fold1', training_depends=True, models_depend=True)
+ check_database(bob.bio.base.load_resource('lfw-unrestricted', 'database', preferred_package='bob.bio.face'),
+ training_depends=True, models_depend=True)
_check_annotations(database)
except IOError as e:
raise SkipTest(
@@ -148,12 +144,12 @@ def test_lfw():
def test_mobio():
database = bob.bio.base.load_resource('mobio-image', 'database', preferred_package='bob.bio.face')
try:
- _check_database_zt(database, models_depend=True)
- _check_database_zt(database, protocol='female', models_depend=True)
- _check_database_zt(bob.bio.base.load_resource('mobio-male', 'database', preferred_package='bob.bio.face'),
- models_depend=True)
- _check_database_zt(bob.bio.base.load_resource('mobio-female', 'database', preferred_package='bob.bio.face'),
- models_depend=True)
+ check_database_zt(database, models_depend=True)
+ check_database_zt(database, protocol='female', models_depend=True)
+ check_database_zt(bob.bio.base.load_resource('mobio-male', 'database', preferred_package='bob.bio.face'),
+ models_depend=True)
+ check_database_zt(bob.bio.base.load_resource('mobio-female', 'database', preferred_package='bob.bio.face'),
+ models_depend=True)
except IOError as e:
raise SkipTest(
"The database could not be queried; probably the db.sql3 file is missing. Here is the error: '%s'" % e)
@@ -169,9 +165,9 @@ def test_mobio():
def test_multipie():
database = bob.bio.base.load_resource('multipie', 'database', preferred_package='bob.bio.face')
try:
- _check_database_zt(database, training_depends=True)
- _check_database_zt(bob.bio.base.load_resource('multipie-pose', 'database', preferred_package='bob.bio.face'),
- training_depends=True)
+ check_database_zt(database, training_depends=True)
+ check_database_zt(bob.bio.base.load_resource('multipie-pose', 'database', preferred_package='bob.bio.face'),
+ training_depends=True)
except IOError as e:
raise SkipTest(
"The database could not queried; probably the db.sql3 file is missing. Here is the error: '%s'" % e)
@@ -190,7 +186,7 @@ def test_multipie():
def test_scface():
database = bob.bio.base.load_resource('scface', 'database', preferred_package='bob.bio.face')
try:
- _check_database_zt(database)
+ check_database_zt(database)
_check_annotations(database)
except IOError as e:
raise SkipTest(
@@ -201,8 +197,19 @@ def test_scface():
def test_xm2vts():
database = bob.bio.base.load_resource('xm2vts', 'database', preferred_package='bob.bio.face')
try:
- _check_database(database, groups=('dev', 'eval'))
- _check_database(database, groups=('dev', 'eval'), protocol='darkened-lp1')
+ check_database(database, groups=('dev', 'eval'))
+ check_database(database, groups=('dev', 'eval'), protocol='darkened-lp1')
+ _check_annotations(database)
+ except IOError as e:
+ raise SkipTest(
+ "The database could not be queried; probably the db.sql3 file is missing. Here is the error: '%s'" % e)
+
+
+@db_available('replay')
+def test_replay():
+ database = bob.bio.base.load_resource('replay', 'database', preferred_package='bob.bio.face')
+ try:
+ check_database(database, groups=('dev', 'eval'))
_check_annotations(database)
except IOError as e:
raise SkipTest(
--
GitLab