From fccd28b1e889f498bf8e120d872acf8592e8a634 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Tue, 20 Sep 2016 10:47:51 +0200
Subject: [PATCH] Solving Issue #38 - BioDatabase method training_files causes
tests to fail
---
bob/bio/base/test/test_database_implementations.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/bob/bio/base/test/test_database_implementations.py b/bob/bio/base/test/test_database_implementations.py
index de0d02ed..2cfcc5b1 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)
--
GitLab