1#!/usr/bin/env python
2# coding=utf-8
3
4
5"""Tests for the aggregated Montgomery-Shenzhen dataset"""
6
7from ..configs.datasets.mc_ch_RS import default as mc_ch_RS
8from ..configs.datasets.montgomery_RS import default as mc_RS
9from ..configs.datasets.shenzhen_RS import default as ch_RS
10from ..configs.datasets.mc_ch_RS import fold_0 as mc_ch_f0
11from ..configs.datasets.montgomery_RS import fold_0 as mc_f0
12from ..configs.datasets.shenzhen_RS import fold_0 as ch_f0
13from ..configs.datasets.mc_ch_RS import fold_1 as mc_ch_f1
14from ..configs.datasets.montgomery_RS import fold_1 as mc_f1
15from ..configs.datasets.shenzhen_RS import fold_1 as ch_f1
16from ..configs.datasets.mc_ch_RS import fold_2 as mc_ch_f2
17from ..configs.datasets.montgomery_RS import fold_2 as mc_f2
18from ..configs.datasets.shenzhen_RS import fold_2 as ch_f2
19from ..configs.datasets.mc_ch_RS import fold_3 as mc_ch_f3
20from ..configs.datasets.montgomery_RS import fold_3 as mc_f3
21from ..configs.datasets.shenzhen_RS import fold_3 as ch_f3
22from ..configs.datasets.mc_ch_RS import fold_4 as mc_ch_f4
23from ..configs.datasets.montgomery_RS import fold_4 as mc_f4
24from ..configs.datasets.shenzhen_RS import fold_4 as ch_f4
25from ..configs.datasets.mc_ch_RS import fold_5 as mc_ch_f5
26from ..configs.datasets.montgomery_RS import fold_5 as mc_f5
27from ..configs.datasets.shenzhen_RS import fold_5 as ch_f5
28from ..configs.datasets.mc_ch_RS import fold_6 as mc_ch_f6
29from ..configs.datasets.montgomery_RS import fold_6 as mc_f6
30from ..configs.datasets.shenzhen_RS import fold_6 as ch_f6
31from ..configs.datasets.mc_ch_RS import fold_7 as mc_ch_f7
32from ..configs.datasets.montgomery_RS import fold_7 as mc_f7
33from ..configs.datasets.shenzhen_RS import fold_7 as ch_f7
34from ..configs.datasets.mc_ch_RS import fold_8 as mc_ch_f8
35from ..configs.datasets.montgomery_RS import fold_8 as mc_f8
36from ..configs.datasets.shenzhen_RS import fold_8 as ch_f8
37from ..configs.datasets.mc_ch_RS import fold_9 as mc_ch_f9
38from ..configs.datasets.montgomery_RS import fold_9 as mc_f9
39from ..configs.datasets.shenzhen_RS import fold_9 as ch_f9
40
41def test_dataset_consistency():
42
43 # Default protocol
44 mc_ch_RS_dataset = mc_ch_RS.dataset
45 assert isinstance(mc_ch_RS_dataset, dict)
46
47 mc_RS_dataset = mc_RS.dataset
48 ch_RS_dataset = ch_RS.dataset
49
50 assert "train" in mc_ch_RS_dataset
51 assert len(mc_ch_RS_dataset["train"]) == len(mc_RS_dataset["train"]) + len(ch_RS_dataset["train"])
52
53 assert "validation" in mc_ch_RS_dataset
54 assert len(mc_ch_RS_dataset["validation"]) == len(mc_RS_dataset["validation"]) + len(ch_RS_dataset["validation"])
55
56 assert "test" in mc_ch_RS_dataset
57 assert len(mc_ch_RS_dataset["test"]) == len(mc_RS_dataset["test"]) + len(ch_RS_dataset["test"])
58
59 # f0 protocol
60 mc_ch_dataset = mc_ch_f0.dataset
61 assert isinstance(mc_ch_dataset, dict)
62
63 mc_dataset = mc_f0.dataset
64 ch_dataset = ch_f0.dataset
65
66 assert "train" in mc_ch_dataset
67 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
68
69 assert "validation" in mc_ch_dataset
70 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
71
72 assert "test" in mc_ch_dataset
73 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])
74
75 # f1 protocol
76 mc_ch_dataset = mc_ch_f1.dataset
77 assert isinstance(mc_ch_dataset, dict)
78
79 mc_dataset = mc_f1.dataset
80 ch_dataset = ch_f1.dataset
81
82 assert "train" in mc_ch_dataset
83 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
84
85 assert "validation" in mc_ch_dataset
86 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
87
88 assert "test" in mc_ch_dataset
89 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])
90
91 # f2 protocol
92 mc_ch_dataset = mc_ch_f2.dataset
93 assert isinstance(mc_ch_dataset, dict)
94
95 mc_dataset = mc_f2.dataset
96 ch_dataset = ch_f2.dataset
97
98 assert "train" in mc_ch_dataset
99 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
100
101 assert "validation" in mc_ch_dataset
102 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
103
104 assert "test" in mc_ch_dataset
105 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])
106
107 # f3 protocol
108 mc_ch_dataset = mc_ch_f3.dataset
109 assert isinstance(mc_ch_dataset, dict)
110
111 mc_dataset = mc_f3.dataset
112 ch_dataset = ch_f3.dataset
113
114 assert "train" in mc_ch_dataset
115 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
116
117 assert "validation" in mc_ch_dataset
118 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
119
120 assert "test" in mc_ch_dataset
121 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])
122
123 # f4 protocol
124 mc_ch_dataset = mc_ch_f4.dataset
125 assert isinstance(mc_ch_dataset, dict)
126
127 mc_dataset = mc_f4.dataset
128 ch_dataset = ch_f4.dataset
129
130 assert "train" in mc_ch_dataset
131 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
132
133 assert "validation" in mc_ch_dataset
134 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
135
136 assert "test" in mc_ch_dataset
137 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"]
138 )
139
140 # f5 protocol
141 mc_ch_dataset = mc_ch_f5.dataset
142 assert isinstance(mc_ch_dataset, dict)
143
144 mc_dataset = mc_f5.dataset
145 ch_dataset = ch_f5.dataset
146
147 assert "train" in mc_ch_dataset
148 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
149
150 assert "validation" in mc_ch_dataset
151 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
152
153 assert "test" in mc_ch_dataset
154 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])
155
156 # f6 protocol
157 mc_ch_dataset = mc_ch_f6.dataset
158 assert isinstance(mc_ch_dataset, dict)
159
160 mc_dataset = mc_f6.dataset
161 ch_dataset = ch_f6.dataset
162
163 assert "train" in mc_ch_dataset
164 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
165
166 assert "validation" in mc_ch_dataset
167 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
168
169 assert "test" in mc_ch_dataset
170 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])
171
172 # f7 protocol
173 mc_ch_dataset = mc_ch_f7.dataset
174 assert isinstance(mc_ch_dataset, dict)
175
176 mc_dataset = mc_f7.dataset
177 ch_dataset = ch_f7.dataset
178
179 assert "train" in mc_ch_dataset
180 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
181
182 assert "validation" in mc_ch_dataset
183 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
184
185 assert "test" in mc_ch_dataset
186 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])
187
188 # f8 protocol
189 mc_ch_dataset = mc_ch_f8.dataset
190 assert isinstance(mc_ch_dataset, dict)
191
192 mc_dataset = mc_f8.dataset
193 ch_dataset = ch_f8.dataset
194
195 assert "train" in mc_ch_dataset
196 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
197
198 assert "validation" in mc_ch_dataset
199 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
200
201 assert "test" in mc_ch_dataset
202 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])
203
204 # f9 protocol
205 mc_ch_dataset = mc_ch_f9.dataset
206 assert isinstance(mc_ch_dataset, dict)
207
208 mc_dataset = mc_f9.dataset
209 ch_dataset = ch_f9.dataset
210
211 assert "train" in mc_ch_dataset
212 assert len(mc_ch_dataset["train"]) == len(mc_dataset["train"]) + len(ch_dataset["train"])
213
214 assert "validation" in mc_ch_dataset
215 assert len(mc_ch_dataset["validation"]) == len(mc_dataset["validation"]) + len(ch_dataset["validation"])
216
217 assert "test" in mc_ch_dataset
218 assert len(mc_ch_dataset["test"]) == len(mc_dataset["test"]) + len(ch_dataset["test"])