Coverage for /scratch/builds/bob/bob.med.tb/miniconda/conda-bld/bob.med.tb_1637571489937/_test_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placeho/lib/python3.8/site-packages/bob/med/tb/test/test_mc_ch_in_pc.py: 55%

1#!/usr/bin/env python 

2# coding=utf-8 

3 

4 

5"""Tests for the aggregated Montgomery-Shenzhen-Indian-Padchest dataset""" 

6 

7import pytest 

8 

9from ..configs.datasets.mc_ch_in_pc import default as mc_ch_in_pc 

10from ..configs.datasets.montgomery import default as mc 

11from ..configs.datasets.shenzhen import default as ch 

12from ..configs.datasets.indian import default as indian 

13from ..configs.datasets.padchest import tb_idiap as pc 

14 

15 

16@pytest.mark.skip_if_rc_var_not_set("bob.med.tb.montgomery.datadir") 

17@pytest.mark.skip_if_rc_var_not_set("bob.med.tb.shenzhen.datadir") 

18@pytest.mark.skip_if_rc_var_not_set("bob.med.tb.indian.datadir") 

19@pytest.mark.skip_if_rc_var_not_set("bob.med.tb.padchest.datadir") 

20def test_dataset_consistency(): 

21 

22 # Default protocol 

23 mc_ch_in_pc_dataset = mc_ch_in_pc.dataset 

24 assert isinstance(mc_ch_in_pc_dataset, dict) 

25 

26 mc_dataset = mc.dataset 

27 ch_dataset = ch.dataset 

28 in_dataset = indian.dataset 

29 pc_dataset = pc.dataset 

30 

31 assert "train" in mc_ch_in_pc_dataset 

32 assert len(mc_ch_in_pc_dataset["train"]) == len(mc_dataset["train"]) + len( 

33 ch_dataset["train"] 

34 ) + len(in_dataset["train"]) + len(pc_dataset["train"]) 

35 

36 assert "test" in mc_ch_in_pc_dataset 

37 assert len(mc_ch_in_pc_dataset["test"]) == len(mc_dataset["test"]) + len( 

38 ch_dataset["test"] 

39 ) + len(in_dataset["test"]) + len(pc_dataset["test"])