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

Shortcuts on this page

r m x   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

321 statements  

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"])