Skip to content
Snippets Groups Projects
Commit 9ba8da7f authored by Yannick DAYER's avatar Yannick DAYER
Browse files

Update replaymobile-img tests

parent 50190761
No related branches found
No related tags found
1 merge request!128Using sparse CSVDataset for replay-mobile
Pipeline #52223 canceled
...@@ -26,7 +26,6 @@ from bob.bio.base.test.utils import db_available ...@@ -26,7 +26,6 @@ from bob.bio.base.test.utils import db_available
from bob.bio.base.test.test_database_implementations import check_database from bob.bio.base.test.test_database_implementations import check_database
import bob.core import bob.core
from bob.extension.download import get_file from bob.extension.download import get_file
from nose.plugins.skip import SkipTest
from bob.extension import rc from bob.extension import rc
logger = bob.core.log.setup("bob.bio.face") logger = bob.core.log.setup("bob.bio.face")
...@@ -278,8 +277,9 @@ def test_replaymobile(): ...@@ -278,8 +277,9 @@ def test_replaymobile():
"replaymobile-img", "database", preferred_package="bob.bio.face" "replaymobile-img", "database", preferred_package="bob.bio.face"
) )
samples = database.all_samples(groups=("dev", "eval")) samples = database.all_samples(groups=("dev", "eval"))
assert len(samples) == 8300, len(samples) assert len(samples) == 1557600, len(samples)
sample = samples[0] sample = samples[0]
del samples
assert hasattr(sample, "annotations") assert hasattr(sample, "annotations")
assert "reye" in sample.annotations assert "reye" in sample.annotations
assert "leye" in sample.annotations assert "leye" in sample.annotations
...@@ -289,7 +289,8 @@ def test_replaymobile(): ...@@ -289,7 +289,8 @@ def test_replaymobile():
assert len(database.references(group="eval")) == 12 assert len(database.references(group="eval")) == 12
assert len(database.probes()) == 4160 assert len(database.probes()) == 4160
assert len(database.probes(group="eval")) == 3020 assert len(database.probes(group="eval")) == 3020
try: # Only if data is available
if rc.get("bob.db.replaymobile.directory", None):
assert sample.annotations == { assert sample.annotations == {
"bottomright": [785, 395], "bottomright": [785, 395],
"topleft": [475, 167], "topleft": [475, 167],
...@@ -301,8 +302,6 @@ def test_replaymobile(): ...@@ -301,8 +302,6 @@ def test_replaymobile():
} }
assert sample.data.shape == (3, 1280, 720) assert sample.data.shape == (3, 1280, 720)
assert sample.data[0, 0, 0] == 87 assert sample.data[0, 0, 0] == 87
except RuntimeError as e:
raise SkipTest(e)
@pytest.mark.skipif( @pytest.mark.skipif(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment