diff --git a/bob/db/morph/test.py b/bob/db/morph/test.py index 9857a2c39a87bcb218939d9dd69b16a7c653e6cb..827bea6d9732ac6d25242710a14ffd6c2c77e104 100644 --- a/bob/db/morph/test.py +++ b/bob/db/morph/test.py @@ -208,7 +208,7 @@ def test_verification_fold1_probes_dev(): expected_sample_count=1, match_sample_count_exactly=False, is_probes=True, - ref_count=None, + ref_count=10540//2, ) # ----------------------------------------------------------------------------- @@ -240,7 +240,7 @@ def test_verification_fold1_probes_eval(): expected_sample_count=1, match_sample_count_exactly=False, is_probes=True, - ref_count=None, + ref_count=10540//2, ) @@ -303,9 +303,15 @@ def test_debug_protocol_probes_dev(): expected_sample_count=1, match_sample_count_exactly=False, is_probes=True, - ref_count=None, + ref_count=100, ) + ids = [p.subject for p in probes_set_dev] + + assert len(set(ids)) == 100 + assert len(set(probes_set_dev[0].references)) == 100 + assert len(_commons(set(ids), set(probes_set_dev[0].references))) == 100 + # ----------------------------------------------------------------------------- # references eval # ----------------------------------------------------------------------------- @@ -335,5 +341,11 @@ def test_debug_protocol_probes_eval(): expected_sample_count=1, match_sample_count_exactly=False, is_probes=True, - ref_count=None, + ref_count=100, ) + + ids = [p.subject for p in probes_set_eval] + + assert len(set(ids)) == 100 + assert len(set(probes_set_eval[0].references)) == 100 + assert len(_commons(set(ids), set(probes_set_eval[0].references))) == 100