From 1a89ddf3cd8e8d2fa0a2e23c7ee3b4a671d3a5e7 Mon Sep 17 00:00:00 2001
From: Olegs NIKISINS <onikisins@italix03.idiap.ch>
Date: Wed, 10 May 2017 16:52:58 +0200
Subject: [PATCH] Fixed the unit tests for HLDI of Replay DB

---
 bob/pad/face/test/test_databases.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/bob/pad/face/test/test_databases.py b/bob/pad/face/test/test_databases.py
index 23de4dbe..4cf6241e 100644
--- a/bob/pad/face/test/test_databases.py
+++ b/bob/pad/face/test/test_databases.py
@@ -11,11 +11,16 @@ from bob.bio.base.test.test_database_implementations import check_database
 
 @db_available('replay')
 def test_replay():
-    module = bob.bio.base.load_resource('replay', 'config',
-        preferred_package='bob.pad.face')
+    replay_database_instance = bob.bio.base.load_resource('replay', 'database', preferred_package='bob.pad.face', package_prefix='bob.pad.')
     try:
-        check_database(module.database, protocol=module.protocol,
-            groups=('train', 'dev', 'eval'))
+
+        assert len( replay_database_instance.objects(groups=['train', 'dev', 'eval']) )==  1200
+        assert len( replay_database_instance.objects(groups=['train', 'dev']) ) ==  720
+        assert len( replay_database_instance.objects(groups=['train']) ) ==  360
+        assert len( replay_database_instance.objects(groups=['train', 'dev', 'eval'], protocol = 'grandtest') )==  1200
+        assert len( replay_database_instance.objects(groups=['train', 'dev', 'eval'], protocol = 'grandtest', purposes='real') ) ==  200
+        assert len( replay_database_instance.objects(groups=['train', 'dev', 'eval'], protocol = 'grandtest', purposes='attack') ) == 1000
+
     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)
-- 
GitLab