1#!/usr/bin/env python
2# coding=utf-8
3
4
5"""Tests for the aggregated Montgomery-Shenzhen-Indian dataset"""
6
7import pytest
8
9from ..configs.datasets.mc_ch_in import default as mc_ch_in
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.mc_ch_in import fold_0 as mc_ch_in_f0
14from ..configs.datasets.montgomery import fold_0 as mc_f0
15from ..configs.datasets.shenzhen import fold_0 as ch_f0
16from ..configs.datasets.indian import fold_0 as indian_f0
17from ..configs.datasets.mc_ch_in import fold_1 as mc_ch_in_f1
18from ..configs.datasets.montgomery import fold_1 as mc_f1
19from ..configs.datasets.shenzhen import fold_1 as ch_f1
20from ..configs.datasets.indian import fold_1 as indian_f1
21from ..configs.datasets.mc_ch_in import fold_2 as mc_ch_in_f2
22from ..configs.datasets.montgomery import fold_2 as mc_f2
23from ..configs.datasets.shenzhen import fold_2 as ch_f2
24from ..configs.datasets.indian import fold_2 as indian_f2
25from ..configs.datasets.mc_ch_in import fold_3 as mc_ch_in_f3
26from ..configs.datasets.montgomery import fold_3 as mc_f3
27from ..configs.datasets.shenzhen import fold_3 as ch_f3
28from ..configs.datasets.indian import fold_3 as indian_f3
29from ..configs.datasets.mc_ch_in import fold_4 as mc_ch_in_f4
30from ..configs.datasets.montgomery import fold_4 as mc_f4
31from ..configs.datasets.shenzhen import fold_4 as ch_f4
32from ..configs.datasets.indian import fold_4 as indian_f4
33from ..configs.datasets.mc_ch_in import fold_5 as mc_ch_in_f5
34from ..configs.datasets.montgomery import fold_5 as mc_f5
35from ..configs.datasets.shenzhen import fold_5 as ch_f5
36from ..configs.datasets.indian import fold_5 as indian_f5
37from ..configs.datasets.mc_ch_in import fold_6 as mc_ch_in_f6
38from ..configs.datasets.montgomery import fold_6 as mc_f6
39from ..configs.datasets.shenzhen import fold_6 as ch_f6
40from ..configs.datasets.indian import fold_6 as indian_f6
41from ..configs.datasets.mc_ch_in import fold_7 as mc_ch_in_f7
42from ..configs.datasets.montgomery import fold_7 as mc_f7
43from ..configs.datasets.shenzhen import fold_7 as ch_f7
44from ..configs.datasets.indian import fold_7 as indian_f7
45from ..configs.datasets.mc_ch_in import fold_8 as mc_ch_in_f8
46from ..configs.datasets.montgomery import fold_8 as mc_f8
47from ..configs.datasets.shenzhen import fold_8 as ch_f8
48from ..configs.datasets.indian import fold_8 as indian_f8
49from ..configs.datasets.mc_ch_in import fold_9 as mc_ch_in_f9
50from ..configs.datasets.montgomery import fold_9 as mc_f9
51from ..configs.datasets.shenzhen import fold_9 as ch_f9
52from ..configs.datasets.indian import fold_9 as indian_f9
53from ..configs.datasets.mc_ch_in import fold_0_rgb as mc_ch_in_f0_rgb
54from ..configs.datasets.montgomery import fold_0_rgb as mc_f0_rgb
55from ..configs.datasets.shenzhen import fold_0_rgb as ch_f0_rgb
56from ..configs.datasets.indian import fold_0_rgb as indian_f0_rgb
57from ..configs.datasets.mc_ch_in import fold_1_rgb as mc_ch_in_f1_rgb
58from ..configs.datasets.montgomery import fold_1_rgb as mc_f1_rgb
59from ..configs.datasets.shenzhen import fold_1_rgb as ch_f1_rgb
60from ..configs.datasets.indian import fold_1_rgb as indian_f1_rgb
61from ..configs.datasets.mc_ch_in import fold_2_rgb as mc_ch_in_f2_rgb
62from ..configs.datasets.montgomery import fold_2_rgb as mc_f2_rgb
63from ..configs.datasets.shenzhen import fold_2_rgb as ch_f2_rgb
64from ..configs.datasets.indian import fold_2_rgb as indian_f2_rgb
65from ..configs.datasets.mc_ch_in import fold_3_rgb as mc_ch_in_f3_rgb
66from ..configs.datasets.montgomery import fold_3_rgb as mc_f3_rgb
67from ..configs.datasets.shenzhen import fold_3_rgb as ch_f3_rgb
68from ..configs.datasets.indian import fold_3_rgb as indian_f3_rgb
69from ..configs.datasets.mc_ch_in import fold_4_rgb as mc_ch_in_f4_rgb
70from ..configs.datasets.montgomery import fold_4_rgb as mc_f4_rgb
71from ..configs.datasets.shenzhen import fold_4_rgb as ch_f4_rgb
72from ..configs.datasets.indian import fold_4_rgb as indian_f4_rgb
73from ..configs.datasets.mc_ch_in import fold_5_rgb as mc_ch_in_f5_rgb
74from ..configs.datasets.montgomery import fold_5_rgb as mc_f5_rgb
75from ..configs.datasets.shenzhen import fold_5_rgb as ch_f5_rgb
76from ..configs.datasets.indian import fold_5_rgb as indian_f5_rgb
77from ..configs.datasets.mc_ch_in import fold_6_rgb as mc_ch_in_f6_rgb
78from ..configs.datasets.montgomery import fold_6_rgb as mc_f6_rgb
79from ..configs.datasets.shenzhen import fold_6_rgb as ch_f6_rgb
80from ..configs.datasets.indian import fold_6_rgb as indian_f6_rgb
81from ..configs.datasets.mc_ch_in import fold_7_rgb as mc_ch_in_f7_rgb
82from ..configs.datasets.montgomery import fold_7_rgb as mc_f7_rgb
83from ..configs.datasets.shenzhen import fold_7_rgb as ch_f7_rgb
84from ..configs.datasets.indian import fold_7_rgb as indian_f7_rgb
85from ..configs.datasets.mc_ch_in import fold_8_rgb as mc_ch_in_f8_rgb
86from ..configs.datasets.montgomery import fold_8_rgb as mc_f8_rgb
87from ..configs.datasets.shenzhen import fold_8_rgb as ch_f8_rgb
88from ..configs.datasets.indian import fold_8_rgb as indian_f8_rgb
89from ..configs.datasets.mc_ch_in import fold_9_rgb as mc_ch_in_f9_rgb
90from ..configs.datasets.montgomery import fold_9_rgb as mc_f9_rgb
91from ..configs.datasets.shenzhen import fold_9_rgb as ch_f9_rgb
92from ..configs.datasets.indian import fold_9_rgb as indian_f9_rgb
93
94
95@pytest.mark.skip_if_rc_var_not_set("bob.med.tb.montgomery.datadir")
96@pytest.mark.skip_if_rc_var_not_set("bob.med.tb.shenzhen.datadir")
97@pytest.mark.skip_if_rc_var_not_set("bob.med.tb.indian.datadir")
98def test_dataset_consistency():
99
100 # Default protocol
101 mc_ch_in_dataset = mc_ch_in.dataset
102 assert isinstance(mc_ch_in_dataset, dict)
103
104 mc_dataset = mc.dataset
105 ch_dataset = ch.dataset
106 in_dataset = indian.dataset
107
108 assert "train" in mc_ch_in_dataset
109 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
110 ch_dataset["train"]
111 ) + len(in_dataset["train"])
112
113 assert "validation" in mc_ch_in_dataset
114 assert len(mc_ch_in_dataset["validation"]) == len(
115 mc_dataset["validation"]
116 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
117
118 assert "test" in mc_ch_in_dataset
119 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
120 ch_dataset["test"]
121 ) + len(in_dataset["test"])
122
123 # Fold 0
124 mc_ch_in_dataset = mc_ch_in_f0.dataset
125 assert isinstance(mc_ch_in_dataset, dict)
126
127 mc_dataset = mc_f0.dataset
128 ch_dataset = ch_f0.dataset
129 in_dataset = indian_f0.dataset
130
131 assert "train" in mc_ch_in_dataset
132 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
133 ch_dataset["train"]
134 ) + len(in_dataset["train"])
135
136 assert "validation" in mc_ch_in_dataset
137 assert len(mc_ch_in_dataset["validation"]) == len(
138 mc_dataset["validation"]
139 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
140
141 assert "test" in mc_ch_in_dataset
142 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
143 ch_dataset["test"]
144 ) + len(in_dataset["test"])
145
146 # Fold 1
147 mc_ch_in_dataset = mc_ch_in_f1.dataset
148 assert isinstance(mc_ch_in_dataset, dict)
149
150 mc_dataset = mc_f1.dataset
151 ch_dataset = ch_f1.dataset
152 in_dataset = indian_f1.dataset
153
154 assert "train" in mc_ch_in_dataset
155 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
156 ch_dataset["train"]
157 ) + len(in_dataset["train"])
158
159 assert "validation" in mc_ch_in_dataset
160 assert len(mc_ch_in_dataset["validation"]) == len(
161 mc_dataset["validation"]
162 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
163
164 assert "test" in mc_ch_in_dataset
165 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
166 ch_dataset["test"]
167 ) + len(in_dataset["test"])
168
169 # Fold 2
170 mc_ch_in_dataset = mc_ch_in_f2.dataset
171 assert isinstance(mc_ch_in_dataset, dict)
172
173 mc_dataset = mc_f2.dataset
174 ch_dataset = ch_f2.dataset
175 in_dataset = indian_f2.dataset
176
177 assert "train" in mc_ch_in_dataset
178 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
179 ch_dataset["train"]
180 ) + len(in_dataset["train"])
181 assert "validation" in mc_ch_in_dataset
182 assert len(mc_ch_in_dataset["validation"]) == len(
183 mc_dataset["validation"]
184 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
185
186 assert "test" in mc_ch_in_dataset
187 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
188 ch_dataset["test"]
189 ) + len(in_dataset["test"])
190
191 # Fold 3
192 mc_ch_in_dataset = mc_ch_in_f3.dataset
193 assert isinstance(mc_ch_in_dataset, dict)
194
195 mc_dataset = mc_f3.dataset
196 ch_dataset = ch_f3.dataset
197 in_dataset = indian_f3.dataset
198
199 assert "train" in mc_ch_in_dataset
200 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
201 ch_dataset["train"]
202 ) + len(in_dataset["train"])
203
204 assert "validation" in mc_ch_in_dataset
205 assert len(mc_ch_in_dataset["validation"]) == len(
206 mc_dataset["validation"]
207 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
208
209 assert "test" in mc_ch_in_dataset
210 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
211 ch_dataset["test"]
212 ) + len(in_dataset["test"])
213
214 # Fold 4
215 mc_ch_in_dataset = mc_ch_in_f4.dataset
216 assert isinstance(mc_ch_in_dataset, dict)
217
218 mc_dataset = mc_f4.dataset
219 ch_dataset = ch_f4.dataset
220 in_dataset = indian_f4.dataset
221
222 assert "train" in mc_ch_in_dataset
223 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
224 ch_dataset["train"]
225 ) + len(in_dataset["train"])
226
227 assert "validation" in mc_ch_in_dataset
228 assert len(mc_ch_in_dataset["validation"]) == len(
229 mc_dataset["validation"]
230 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
231
232 assert "test" in mc_ch_in_dataset
233 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
234 ch_dataset["test"]
235 ) + len(in_dataset["test"])
236
237 # Fold 5
238 mc_ch_in_dataset = mc_ch_in_f5.dataset
239 assert isinstance(mc_ch_in_dataset, dict)
240
241 mc_dataset = mc_f5.dataset
242 ch_dataset = ch_f5.dataset
243 in_dataset = indian_f5.dataset
244
245 assert "train" in mc_ch_in_dataset
246 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
247 ch_dataset["train"]
248 ) + len(in_dataset["train"])
249
250 assert "validation" in mc_ch_in_dataset
251 assert len(mc_ch_in_dataset["validation"]) == len(
252 mc_dataset["validation"]
253 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
254
255 assert "test" in mc_ch_in_dataset
256 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
257 ch_dataset["test"]
258 ) + len(in_dataset["test"])
259
260 # Fold 6
261 mc_ch_in_dataset = mc_ch_in_f6.dataset
262 assert isinstance(mc_ch_in_dataset, dict)
263
264 mc_dataset = mc_f6.dataset
265 ch_dataset = ch_f6.dataset
266 in_dataset = indian_f6.dataset
267
268 assert "train" in mc_ch_in_dataset
269 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
270 ch_dataset["train"]
271 ) + len(in_dataset["train"])
272
273 assert "validation" in mc_ch_in_dataset
274 assert len(mc_ch_in_dataset["validation"]) == len(
275 mc_dataset["validation"]
276 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
277
278 assert "test" in mc_ch_in_dataset
279 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
280 ch_dataset["test"]
281 ) + len(in_dataset["test"])
282
283 # Fold 7
284 mc_ch_in_dataset = mc_ch_in_f7.dataset
285 assert isinstance(mc_ch_in_dataset, dict)
286
287 mc_dataset = mc_f7.dataset
288 ch_dataset = ch_f7.dataset
289 in_dataset = indian_f7.dataset
290
291 assert "train" in mc_ch_in_dataset
292 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
293 ch_dataset["train"]
294 ) + len(in_dataset["train"])
295
296 assert "validation" in mc_ch_in_dataset
297 assert len(mc_ch_in_dataset["validation"]) == len(
298 mc_dataset["validation"]
299 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
300
301 assert "test" in mc_ch_in_dataset
302 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
303 ch_dataset["test"]
304 ) + len(in_dataset["test"])
305
306 # Fold 8
307 mc_ch_in_dataset = mc_ch_in_f8.dataset
308 assert isinstance(mc_ch_in_dataset, dict)
309
310 mc_dataset = mc_f8.dataset
311 ch_dataset = ch_f8.dataset
312 in_dataset = indian_f8.dataset
313
314 assert "train" in mc_ch_in_dataset
315 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
316 ch_dataset["train"]
317 ) + len(in_dataset["train"])
318
319 assert "validation" in mc_ch_in_dataset
320 assert len(mc_ch_in_dataset["validation"]) == len(
321 mc_dataset["validation"]
322 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
323 assert "test" in mc_ch_in_dataset
324 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
325 ch_dataset["test"]
326 ) + len(in_dataset["test"])
327
328 # Fold 9
329 mc_ch_in_dataset = mc_ch_in_f9.dataset
330 assert isinstance(mc_ch_in_dataset, dict)
331
332 mc_dataset = mc_f9.dataset
333 ch_dataset = ch_f9.dataset
334 in_dataset = indian_f9.dataset
335
336 assert "train" in mc_ch_in_dataset
337 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
338 ch_dataset["train"]
339 ) + len(in_dataset["train"])
340
341 assert "validation" in mc_ch_in_dataset
342 assert len(mc_ch_in_dataset["validation"]) == len(
343 mc_dataset["validation"]
344 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
345
346 assert "test" in mc_ch_in_dataset
347 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
348 ch_dataset["test"]
349 ) + len(in_dataset["test"])
350
351 # Fold 0, RGB
352 mc_ch_in_dataset = mc_ch_in_f0_rgb.dataset
353 assert isinstance(mc_ch_in_dataset, dict)
354
355 mc_dataset = mc_f0_rgb.dataset
356 ch_dataset = ch_f0_rgb.dataset
357 in_dataset = indian_f0_rgb.dataset
358
359 assert "train" in mc_ch_in_dataset
360 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
361 ch_dataset["train"]
362 ) + len(in_dataset["train"])
363
364 assert "validation" in mc_ch_in_dataset
365 assert len(mc_ch_in_dataset["validation"]) == len(
366 mc_dataset["validation"]
367 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
368
369 assert "test" in mc_ch_in_dataset
370 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
371 ch_dataset["test"]
372 ) + len(in_dataset["test"])
373
374 # Fold 1, RGB
375 mc_ch_in_dataset = mc_ch_in_f1_rgb.dataset
376 assert isinstance(mc_ch_in_dataset, dict)
377
378 mc_dataset = mc_f1_rgb.dataset
379 ch_dataset = ch_f1_rgb.dataset
380 in_dataset = indian_f1_rgb.dataset
381
382 assert "train" in mc_ch_in_dataset
383 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
384 ch_dataset["train"]
385 ) + len(in_dataset["train"])
386
387 assert "validation" in mc_ch_in_dataset
388 assert len(mc_ch_in_dataset["validation"]) == len(
389 mc_dataset["validation"]
390 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
391
392 assert "test" in mc_ch_in_dataset
393 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
394 ch_dataset["test"]
395 ) + len(in_dataset["test"])
396
397 # Fold 2, RGB
398 mc_ch_in_dataset = mc_ch_in_f2_rgb.dataset
399 assert isinstance(mc_ch_in_dataset, dict)
400
401 mc_dataset = mc_f2_rgb.dataset
402 ch_dataset = ch_f2_rgb.dataset
403 in_dataset = indian_f2_rgb.dataset
404
405 assert "train" in mc_ch_in_dataset
406 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
407 ch_dataset["train"]
408 ) + len(in_dataset["train"])
409
410 assert "validation" in mc_ch_in_dataset
411 assert len(mc_ch_in_dataset["validation"]) == len(
412 mc_dataset["validation"]
413 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
414
415 assert "test" in mc_ch_in_dataset
416 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
417 ch_dataset["test"]
418 ) + len(in_dataset["test"])
419
420 # Fold 3, RGB
421 mc_ch_in_dataset = mc_ch_in_f3_rgb.dataset
422 assert isinstance(mc_ch_in_dataset, dict)
423
424 mc_dataset = mc_f3_rgb.dataset
425 ch_dataset = ch_f3_rgb.dataset
426 in_dataset = indian_f3_rgb.dataset
427
428 assert "train" in mc_ch_in_dataset
429 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
430 ch_dataset["train"]
431 ) + len(in_dataset["train"])
432
433 assert "validation" in mc_ch_in_dataset
434 assert len(mc_ch_in_dataset["validation"]) == len(
435 mc_dataset["validation"]
436 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
437
438 assert "test" in mc_ch_in_dataset
439 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
440 ch_dataset["test"]
441 ) + len(in_dataset["test"])
442
443 # Fold 4, RGB
444 mc_ch_in_dataset = mc_ch_in_f4_rgb.dataset
445 assert isinstance(mc_ch_in_dataset, dict)
446
447 mc_dataset = mc_f4_rgb.dataset
448 ch_dataset = ch_f4_rgb.dataset
449 in_dataset = indian_f4_rgb.dataset
450
451 assert "train" in mc_ch_in_dataset
452 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
453 ch_dataset["train"]
454 ) + len(in_dataset["train"])
455
456 assert "validation" in mc_ch_in_dataset
457 assert len(mc_ch_in_dataset["validation"]) == len(
458 mc_dataset["validation"]
459 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
460
461 assert "test" in mc_ch_in_dataset
462 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
463 ch_dataset["test"]
464 ) + len(in_dataset["test"])
465
466 # Fold 5, RGB
467 mc_ch_in_dataset = mc_ch_in_f5_rgb.dataset
468 assert isinstance(mc_ch_in_dataset, dict)
469
470 mc_dataset = mc_f5_rgb.dataset
471 ch_dataset = ch_f5_rgb.dataset
472 in_dataset = indian_f5_rgb.dataset
473
474 assert "train" in mc_ch_in_dataset
475 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
476 ch_dataset["train"]
477 ) + len(in_dataset["train"])
478
479 assert "validation" in mc_ch_in_dataset
480 assert len(mc_ch_in_dataset["validation"]) == len(
481 mc_dataset["validation"]
482 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
483
484 assert "test" in mc_ch_in_dataset
485 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
486 ch_dataset["test"]
487 ) + len(in_dataset["test"])
488
489 # Fold 6, RGB
490 mc_ch_in_dataset = mc_ch_in_f6_rgb.dataset
491 assert isinstance(mc_ch_in_dataset, dict)
492
493 mc_dataset = mc_f6_rgb.dataset
494 ch_dataset = ch_f6_rgb.dataset
495 in_dataset = indian_f6_rgb.dataset
496
497 assert "train" in mc_ch_in_dataset
498 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
499 ch_dataset["train"]
500 ) + len(in_dataset["train"])
501
502 assert "validation" in mc_ch_in_dataset
503 assert len(mc_ch_in_dataset["validation"]) == len(
504 mc_dataset["validation"]
505 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
506
507 assert "test" in mc_ch_in_dataset
508 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
509 ch_dataset["test"]
510 ) + len(in_dataset["test"])
511
512 # Fold 7, RGB
513 mc_ch_in_dataset = mc_ch_in_f7_rgb.dataset
514 assert isinstance(mc_ch_in_dataset, dict)
515
516 mc_dataset = mc_f7_rgb.dataset
517 ch_dataset = ch_f7_rgb.dataset
518 in_dataset = indian_f7_rgb.dataset
519
520 assert "train" in mc_ch_in_dataset
521 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
522 ch_dataset["train"]
523 ) + len(in_dataset["train"])
524
525 assert "validation" in mc_ch_in_dataset
526 assert len(mc_ch_in_dataset["validation"]) == len(
527 mc_dataset["validation"]
528 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
529
530 assert "test" in mc_ch_in_dataset
531 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
532 ch_dataset["test"]
533 ) + len(in_dataset["test"])
534
535 # Fold 8, RGB
536 mc_ch_in_dataset = mc_ch_in_f8_rgb.dataset
537 assert isinstance(mc_ch_in_dataset, dict)
538
539 mc_dataset = mc_f8_rgb.dataset
540 ch_dataset = ch_f8_rgb.dataset
541 in_dataset = indian_f8_rgb.dataset
542
543 assert "train" in mc_ch_in_dataset
544 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
545 ch_dataset["train"]
546 ) + len(in_dataset["train"])
547
548 assert "validation" in mc_ch_in_dataset
549 assert len(mc_ch_in_dataset["validation"]) == len(
550 mc_dataset["validation"]
551 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
552
553 assert "test" in mc_ch_in_dataset
554 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
555 ch_dataset["test"]
556 ) + len(in_dataset["test"])
557
558 # Fold 9, RGB
559 mc_ch_in_dataset = mc_ch_in_f9_rgb.dataset
560 assert isinstance(mc_ch_in_dataset, dict)
561
562 mc_dataset = mc_f9_rgb.dataset
563 ch_dataset = ch_f9_rgb.dataset
564 in_dataset = indian_f9_rgb.dataset
565
566 assert "train" in mc_ch_in_dataset
567 assert len(mc_ch_in_dataset["train"]) == len(mc_dataset["train"]) + len(
568 ch_dataset["train"]
569 ) + len(in_dataset["train"])
570
571 assert "validation" in mc_ch_in_dataset
572 assert len(mc_ch_in_dataset["validation"]) == len(
573 mc_dataset["validation"]
574 ) + len(ch_dataset["validation"]) + len(in_dataset["validation"])
575
576 assert "test" in mc_ch_in_dataset
577 assert len(mc_ch_in_dataset["test"]) == len(mc_dataset["test"]) + len(
578 ch_dataset["test"]
579 ) + len(in_dataset["test"])