diff --git a/bob/bio/base/test/test_database_implementations.py b/bob/bio/base/test/test_database_implementations.py index de0d02ed3f39e7989d1d5f51125600f48efc6b53..2cfcc5b1d22bfb208f09c4ed39f161f4ad3022dd 100644 --- a/bob/bio/base/test/test_database_implementations.py +++ b/bob/bio/base/test/test_database_implementations.py @@ -12,7 +12,7 @@ import os from bob.bio.base.database import BioDatabase, ZTBioDatabase -def check_database(database, groups=('dev',), protocol=None, training_depends=False, models_depend=False): +def check_database(database, groups=('dev',), protocol=None, training_depends=False, models_depend=False, skip_train=False): assert isinstance(database, BioDatabase) # load the directories @@ -23,8 +23,9 @@ def check_database(database, groups=('dev',), protocol=None, training_depends=Fa 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 + if not skip_train: + 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)