DeepRegNet /
DeepReg
| @@ 239-283 (lines=45) @@ | ||
| 236 | assert is_equal_np(got[1], expected[1]) |
|
| 237 | loader.close() |
|
| 238 | ||
| 239 | @pytest.mark.parametrize( |
|
| 240 | "name,expected", |
|
| 241 | [ |
|
| 242 | ( |
|
| 243 | "paired", |
|
| 244 | [ |
|
| 245 | ("./data/test/nifti/paired/test", "case000025"), |
|
| 246 | ("./data/test/nifti/paired/test", "case000026"), |
|
| 247 | ], |
|
| 248 | ), |
|
| 249 | ( |
|
| 250 | "unpaired", |
|
| 251 | [ |
|
| 252 | ("./data/test/nifti/unpaired/test", "case000025"), |
|
| 253 | ("./data/test/nifti/unpaired/test", "case000026"), |
|
| 254 | ], |
|
| 255 | ), |
|
| 256 | ( |
|
| 257 | "grouped", |
|
| 258 | [ |
|
| 259 | ("./data/test/nifti/grouped/test", "group1", "case000025"), |
|
| 260 | ("./data/test/nifti/grouped/test", "group1", "case000026"), |
|
| 261 | ], |
|
| 262 | ), |
|
| 263 | ( |
|
| 264 | "multi_dirs_grouped", |
|
| 265 | [ |
|
| 266 | ("./data/test/nifti/grouped/test", "group1", "case000025"), |
|
| 267 | ("./data/test/nifti/grouped/test", "group1", "case000026"), |
|
| 268 | ("./data/test/nifti/grouped/train", "group1", "case000000"), |
|
| 269 | ("./data/test/nifti/grouped/train", "group1", "case000001"), |
|
| 270 | ("./data/test/nifti/grouped/train", "group1", "case000003"), |
|
| 271 | ("./data/test/nifti/grouped/train", "group1", "case000008"), |
|
| 272 | ("./data/test/nifti/grouped/train", "group2", "case000009"), |
|
| 273 | ("./data/test/nifti/grouped/train", "group2", "case000011"), |
|
| 274 | ("./data/test/nifti/grouped/train", "group2", "case000012"), |
|
| 275 | ], |
|
| 276 | ), |
|
| 277 | ], |
|
| 278 | ) |
|
| 279 | def test_get_data_ids(self, name, expected): |
|
| 280 | loader = get_loader(name) |
|
| 281 | got = loader.get_data_ids() |
|
| 282 | assert got == expected |
|
| 283 | loader.close() |
|
| 284 | ||
| 285 | @pytest.mark.parametrize( |
|
| 286 | "index,err_type", |
|
| @@ 173-211 (lines=39) @@ | ||
| 170 | assert is_equal_np(got[1], expected[1]) |
|
| 171 | loader.close() |
|
| 172 | ||
| 173 | @pytest.mark.parametrize( |
|
| 174 | "name,expected", |
|
| 175 | [ |
|
| 176 | ("paired", [("./data/test/h5/paired/test", "case000025.nii.gz")]), |
|
| 177 | ( |
|
| 178 | "unpaired", |
|
| 179 | [ |
|
| 180 | ("./data/test/h5/unpaired/test", "case000025.nii.gz"), |
|
| 181 | ("./data/test/h5/unpaired/test", "case000026.nii.gz"), |
|
| 182 | ], |
|
| 183 | ), |
|
| 184 | ( |
|
| 185 | "grouped", |
|
| 186 | [ |
|
| 187 | ("./data/test/h5/grouped/test", "1", "1"), |
|
| 188 | ("./data/test/h5/grouped/test", "1", "2"), |
|
| 189 | ], |
|
| 190 | ), |
|
| 191 | ( |
|
| 192 | "multi_dirs_grouped", |
|
| 193 | [ |
|
| 194 | ("./data/test/h5/grouped/train", "1", "1"), |
|
| 195 | ("./data/test/h5/grouped/train", "1", "2"), |
|
| 196 | ("./data/test/h5/grouped/train", "1", "3"), |
|
| 197 | ("./data/test/h5/grouped/train", "1", "4"), |
|
| 198 | ("./data/test/h5/grouped/train", "2", "1"), |
|
| 199 | ("./data/test/h5/grouped/train", "2", "2"), |
|
| 200 | ("./data/test/h5/grouped/train", "2", "3"), |
|
| 201 | ("./data/test/h5/grouped/test", "1", "1"), |
|
| 202 | ("./data/test/h5/grouped/test", "1", "2"), |
|
| 203 | ], |
|
| 204 | ), |
|
| 205 | ], |
|
| 206 | ) |
|
| 207 | def test_get_data_ids(self, name, expected): |
|
| 208 | loader = get_loader(name) |
|
| 209 | got = loader.get_data_ids() |
|
| 210 | assert got == expected |
|
| 211 | loader.close() |
|
| 212 | ||
| 213 | @pytest.mark.parametrize( |
|
| 214 | "index,err_type", |
|
