1#!/usr/bin/env python
2# coding=utf-8
3
4
5"""Tests for the aggregated NIH CXR14-PadChest dataset"""
6
7from ..configs.datasets.nih_cxr14_re_pc import idiap as nih_pc
8from ..configs.datasets.nih_cxr14_re import default as nih
9from ..configs.datasets.padchest import no_tb_idiap as pc
10import pytest
11
12@pytest.mark.skip_if_rc_var_not_set('bob.med.tb.padchest.datadir')
13@pytest.mark.skip_if_rc_var_not_set('bob.med.tb.nih_cxr14_re.datadir')
14def test_dataset_consistency():
15
16 # Default protocol
17 nih_pc_dataset = nih_pc.dataset
18 assert isinstance(nih_pc_dataset, dict)
19
20 nih_dataset = nih.dataset
21 pc_dataset = pc.dataset
22
23 assert "train" in nih_pc_dataset
24 assert len(nih_pc_dataset["train"]) == len(nih_dataset["train"]) + len(pc_dataset["train"])
25
26 assert "validation" in nih_pc_dataset
27 assert len(nih_pc_dataset["validation"]) == len(nih_dataset["validation"]) + len(pc_dataset["validation"])