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_RS.py: 100%

1#!/usr/bin/env python 

2# coding=utf-8 

3 

4 

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

6 

7from ..configs.datasets.mc_ch_in_pc_RS import default as mc_ch_in_pc 

8from ..configs.datasets.montgomery_RS import default as mc_RS 

9from ..configs.datasets.shenzhen_RS import default as ch_RS 

10from ..configs.datasets.indian_RS import default as in_RS 

11from ..configs.datasets.padchest_RS import tb_idiap as pc_RS 

12 

13def test_dataset_consistency(): 

14 

15 # Default protocol 

16 mc_ch_in_pc_dataset = mc_ch_in_pc.dataset 

17 assert isinstance(mc_ch_in_pc_dataset, dict) 

18 

19 mc_RS_dataset = mc_RS.dataset 

20 ch_RS_dataset = ch_RS.dataset 

21 in_RS_dataset = in_RS.dataset 

22 pc_RS_dataset = pc_RS.dataset 

23 

24 assert "train" in mc_ch_in_pc_dataset 

25 assert len(mc_ch_in_pc_dataset["train"]) == len(mc_RS_dataset["train"]) + len(ch_RS_dataset["train"]) + len(in_RS_dataset["train"]) + len(pc_RS_dataset["train"]) 

26 

27 assert "validation" in mc_ch_in_pc_dataset 

28 assert len(mc_ch_in_pc_dataset["validation"]) == len(mc_RS_dataset["validation"]) + len(ch_RS_dataset["validation"]) + len(in_RS_dataset["validation"]) + len(pc_RS_dataset["validation"]) 

29 

30 assert "test" in mc_ch_in_pc_dataset 

31 assert len(mc_ch_in_pc_dataset["test"]) == len(mc_RS_dataset["test"]) + len(ch_RS_dataset["test"]) + len(in_RS_dataset["test"]) + len(pc_RS_dataset["test"])