From 9b1b030382fa8035a432bac4a541f7e2cf00dab0 Mon Sep 17 00:00:00 2001
From: Yannick DAYER <yannick.dayer@idiap.ch>
Date: Wed, 9 Jun 2021 10:58:12 +0200
Subject: [PATCH] Change tests for replay-mobile to csv dataset
---
bob/bio/face/test/test_databases.py | 79 ++++++++++-------------------
1 file changed, 28 insertions(+), 51 deletions(-)
diff --git a/bob/bio/face/test/test_databases.py b/bob/bio/face/test/test_databases.py
index 306a659d..ff6213f4 100644
--- a/bob/bio/face/test/test_databases.py
+++ b/bob/bio/face/test/test_databases.py
@@ -26,6 +26,7 @@ from bob.bio.base.test.utils import db_available
from bob.bio.base.test.test_database_implementations import check_database
import bob.core
from bob.extension.download import get_file
+from nose.plugins.skip import SkipTest
logger = bob.core.log.setup("bob.bio.face")
@@ -271,60 +272,36 @@ def test_replay_spoof():
)
-@db_available("replaymobile")
-def test_replaymobile_csv():
- from bob.bio.face.database.replaymobile_csv import ReplayMobileDatabase # TODO: load resource instead
- database = ReplayMobileDatabase("grandtest", protocol_definition_path="./csv_datasets", data_path="")
- samples = database.all_samples(groups=("dev","eval"))
- assert len(samples) == 8300, len(samples)
- assert all([s.path for s in samples])
- # assert samples[0].data.shape == (3, 1280, 720)# TODO data sample?
- assert hasattr(samples[0], "annotations")
- assert "topleft" in samples[0].annotations
- assert "bottomright" in samples[0].annotations
- assert hasattr(samples[0], "frame")
-
-
-@db_available("replaymobile")
-def test_replaymobile_licit():
+def test_replaymobile():
database = bob.bio.base.load_resource(
- "replaymobile-img-licit", "database", preferred_package="bob.bio.face"
+ "replaymobile-img", "database", preferred_package="bob.bio.face"
)
+ samples = database.all_samples(groups=("dev", "eval"))
+ assert len(samples) == 8300, len(samples)
+ sample = samples[0]
+ assert hasattr(sample, "annotations")
+ assert "reye" in sample.annotations
+ assert "leye" in sample.annotations
+ assert hasattr(sample, "path")
+ assert hasattr(sample, "frame")
+ assert len(database.references()) == 16
+ assert len(database.references(group="eval")) == 12
+ assert len(database.probes()) == 4160
+ assert len(database.probes(group="eval")) == 3020
try:
- check_database(database, groups=("dev", "eval"))
- except IOError as e:
- pytest.skip(
- "The database could not be queried; probably the db.sql3 file is missing. Here is the error: '%s'"
- % e
- )
- try:
- _check_annotations(database, topleft=True, limit_files=20)
- except IOError as e:
- pytest.skip(
- "The annotations could not be queried; probably the annotation files are missing. Here is the error: '%s'"
- % e
- )
-
-
-@db_available("replaymobile")
-def test_replaymobile_spoof():
- database = bob.bio.base.load_resource(
- "replaymobile-img-spoof", "database", preferred_package="bob.bio.face"
- )
- try:
- check_database(database, groups=("dev", "eval"))
- except IOError as e:
- pytest.skip(
- "The database could not be queried; probably the db.sql3 file is missing. Here is the error: '%s'"
- % e
- )
- try:
- _check_annotations(database, topleft=True, limit_files=20)
- except IOError as e:
- pytest.skip(
- "The annotations could not be queried; probably the annotation files are missing. Here is the error: '%s'"
- % e
- )
+ assert sample.annotations == {
+ "bottomright": [785, 395],
+ "topleft": [475, 167],
+ "leye": [587, 336],
+ "reye": [588, 238],
+ "mouthleft": [705, 252],
+ "mouthright": [706, 326],
+ "nose": [643, 295],
+ }
+ assert sample.data.shape == (3, 1280, 720)
+ assert sample.data[0, 0, 0] == 87
+ except RuntimeError as e:
+ raise SkipTest(e)
def test_ijbc():
--
GitLab