From 21baea570ef2b67a64074202a726ad1daea11f7b Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.anjos@idiap.ch>
Date: Tue, 4 Oct 2016 13:36:22 +0200
Subject: [PATCH] Fix tests (closes #6)

---
 bob/bio/vein/configurations/utfvp.py      |  5 ++++-
 bob/bio/vein/configurations/verafinger.py |  5 ++++-
 bob/bio/vein/database/utfvp.py            |  2 +-
 bob/bio/vein/database/verafinger.py       |  2 +-
 bob/bio/vein/tests/test_databases.py      | 14 +++++++++-----
 5 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/bob/bio/vein/configurations/utfvp.py b/bob/bio/vein/configurations/utfvp.py
index eb742b0..6b40cc5 100644
--- a/bob/bio/vein/configurations/utfvp.py
+++ b/bob/bio/vein/configurations/utfvp.py
@@ -22,7 +22,10 @@ from bob.bio.vein.database import UtfvpBioDatabase
 utfvp_directory = "[YOUR_UTFVP_DIRECTORY]"
 """Value of ``~/.bob_bio_databases.txt`` for this database"""
 
-database = UtfvpBioDatabase(original_directory = utfvp_directory)
+database = UtfvpBioDatabase(
+    original_directory = utfvp_directory,
+    original_extension = '.png',
+    )
 """The :py:class:`bob.bio.base.database.BioDatabase` derivative with UTFVP settings
 
 .. warning::
diff --git a/bob/bio/vein/configurations/verafinger.py b/bob/bio/vein/configurations/verafinger.py
index f0b9aba..9b64210 100644
--- a/bob/bio/vein/configurations/verafinger.py
+++ b/bob/bio/vein/configurations/verafinger.py
@@ -20,7 +20,10 @@ from bob.bio.vein.database import VerafingerBioDatabase
 verafinger_directory = "[YOUR_VERAFINGER_DIRECTORY]"
 """Value of ``~/.bob_bio_databases.txt`` for this database"""
 
-database = VerafingerBioDatabase(original_directory = verafinger_directory)
+database = VerafingerBioDatabase(
+    original_directory = verafinger_directory,
+    original_extension = '.png',
+    )
 """The :py:class:`bob.bio.base.database.BioDatabase` derivative with Verafinger
 database settings
 
diff --git a/bob/bio/vein/database/utfvp.py b/bob/bio/vein/database/utfvp.py
index 8fe5a5a..b8f09a0 100644
--- a/bob/bio/vein/database/utfvp.py
+++ b/bob/bio/vein/database/utfvp.py
@@ -3,7 +3,7 @@
 # Tue 27 Sep 2016 16:49:05 CEST
 
 from .database import VeinBioFile
-from bob.bio.base.database import BioDatabase, BioFile
+from bob.bio.base.database import BioDatabase
 
 
 class UtfvpBioDatabase(BioDatabase):
diff --git a/bob/bio/vein/database/verafinger.py b/bob/bio/vein/database/verafinger.py
index a33e2e4..23d3ff2 100644
--- a/bob/bio/vein/database/verafinger.py
+++ b/bob/bio/vein/database/verafinger.py
@@ -4,7 +4,7 @@
 
 
 from .database import VeinBioFile
-from bob.bio.base.database import BioDatabase, BioFile
+from bob.bio.base.database import BioDatabase
 
 
 class VerafingerBioDatabase(BioDatabase):
diff --git a/bob/bio/vein/tests/test_databases.py b/bob/bio/vein/tests/test_databases.py
index 3903a35..08c7c55 100644
--- a/bob/bio/vein/tests/test_databases.py
+++ b/bob/bio/vein/tests/test_databases.py
@@ -6,14 +6,16 @@ from nose.plugins.skip import SkipTest
 
 import bob.bio.base
 from bob.bio.base.test.utils import db_available
-from bob.bio.base.test.test_database_implementations import check_database, check_database_zt
+from bob.bio.base.test.test_database_implementations import check_database
 
 
 @db_available('utfvp')
 def test_utfvp():
-    database = bob.bio.base.load_resource('utfvp', 'database', preferred_package='bob.bio.vein')
+    module = bob.bio.base.load_resource('utfvp', 'config',
+        preferred_package='bob.bio.vein')
     try:
-        check_database(database, protocol='nomLeftRing', groups=('dev', 'eval'))
+        check_database(module.database, protocol=module.protocol,
+            groups=('dev', 'eval'))
     except IOError as e:
         raise SkipTest(
             "The database could not queried; probably the db.sql3 file is missing. Here is the error: '%s'" % e)
@@ -21,9 +23,11 @@ def test_utfvp():
 
 @db_available('verafinger')
 def test_verafinger():
-    database = bob.bio.base.load_resource('verafinger', 'database', preferred_package='bob.bio.vein')
+    module = bob.bio.base.load_resource('verafinger', 'config',
+        preferred_package='bob.bio.vein')
     try:
-        check_database(database, protocol='Fifty', groups=('dev', 'eval'))
+        check_database(module.database, protocol='Fifty', groups=('dev',
+          'eval'))
     except IOError as e:
         raise SkipTest(
             "The database could not queried; probably the db.sql3 file is missing. Here is the error: '%s'" % e)
-- 
GitLab