DeepRegNet /
DeepReg
| @@ 215-237 (lines=23) @@ | ||
| 212 | assert "No data collected" in str(err_info.value) |
|
| 213 | os.removedirs(dir_path) |
|
| 214 | ||
| 215 | @pytest.mark.parametrize( |
|
| 216 | "name,index,expected", |
|
| 217 | [ |
|
| 218 | ("paired", 0, [(44, 59, 41), [255.0, 0.0, 68.359276, 65.84009]]), |
|
| 219 | ("unpaired", 0, [(64, 64, 60), [255.0, 0.0, 60.073948, 47.27648]]), |
|
| 220 | ("grouped", (0, 1), [(64, 64, 60), [255.0, 0.0, 85.67942, 49.193127]]), |
|
| 221 | ( |
|
| 222 | "multi_dirs_grouped", |
|
| 223 | (0, 1), |
|
| 224 | [(64, 64, 60), [255.0, 0.0, 85.67942, 49.193127]], |
|
| 225 | ), |
|
| 226 | ], |
|
| 227 | ) |
|
| 228 | def test_get_data(self, name, index, expected): |
|
| 229 | loader = get_loader(name) |
|
| 230 | array = loader.get_data(index) |
|
| 231 | got = [ |
|
| 232 | np.shape(array), |
|
| 233 | [np.amax(array), np.amin(array), np.mean(array), np.std(array)], |
|
| 234 | ] |
|
| 235 | assert got[0] == expected[0] |
|
| 236 | assert is_equal_np(got[1], expected[1]) |
|
| 237 | loader.close() |
|
| 238 | ||
| 239 | @pytest.mark.parametrize( |
|
| 240 | "name,expected", |
|
| @@ 149-171 (lines=23) @@ | ||
| 146 | assert "No data collected" in str(err_info.value) |
|
| 147 | os.remove(file_path) |
|
| 148 | ||
| 149 | @pytest.mark.parametrize( |
|
| 150 | "name,index,expected", |
|
| 151 | [ |
|
| 152 | ("paired", 0, [(44, 59, 41), [255.0, 0.0, 68.359276, 65.84009]]), |
|
| 153 | ("unpaired", 0, [(64, 64, 60), [255.0, 0.0, 60.073948, 47.27648]]), |
|
| 154 | ("grouped", (0, 1), [(64, 64, 60), [255.0, 0.0, 60.073948, 47.27648]]), |
|
| 155 | ( |
|
| 156 | "multi_dirs_grouped", |
|
| 157 | (0, 1), |
|
| 158 | [(64, 64, 60), [255.0, 0.0, 60.073948, 47.27648]], |
|
| 159 | ), |
|
| 160 | ], |
|
| 161 | ) |
|
| 162 | def test_get_data(self, name, index, expected): |
|
| 163 | loader = get_loader(name) |
|
| 164 | array = loader.get_data(index) |
|
| 165 | got = [ |
|
| 166 | np.shape(array), |
|
| 167 | [np.amax(array), np.amin(array), np.mean(array), np.std(array)], |
|
| 168 | ] |
|
| 169 | assert got[0] == expected[0] |
|
| 170 | assert is_equal_np(got[1], expected[1]) |
|
| 171 | loader.close() |
|
| 172 | ||
| 173 | @pytest.mark.parametrize( |
|
| 174 | "name,expected", |
|
