|
22 | 22 | from parameterized import parameterized |
23 | 23 |
|
24 | 24 | from monai.data.image_reader import ITKReader, NibabelReader, NrrdReader, PILReader |
25 | | -from monai.data.image_writer import ITKWriter, NibabelWriter, PILWriter, register_writer, resolve_writer |
| 25 | +from monai.data.image_writer import FILETYPE_HINT, ITKWriter, NibabelWriter, PILWriter, register_writer, resolve_writer |
26 | 26 | from monai.data.meta_tensor import MetaTensor |
27 | 27 | from monai.transforms import LoadImage, SaveImage, moveaxis |
28 | 28 | from monai.utils import MetaKeys, OptionalImportError, optional_import |
@@ -188,24 +188,22 @@ def test_3d(self, reader, writer): |
188 | 188 | self.nrrd_rw(test_data, reader, writer, np.float32) |
189 | 189 |
|
190 | 190 |
|
191 | | -if __name__ == "__main__": |
192 | | - unittest.main() |
193 | | - |
194 | 191 | class TestResolveWriterHint(unittest.TestCase): |
195 | | - |
196 | 192 | def test_filetype_hint_content(self): |
197 | | - from monai.data.image_writer import FILETYPE_HINT |
198 | 193 | self.assertEqual(FILETYPE_HINT.get("nii"), "nibabel") |
199 | 194 | self.assertEqual(FILETYPE_HINT.get("nii.gz"), "nibabel") |
200 | 195 | self.assertEqual(FILETYPE_HINT.get("png"), "pillow") |
201 | 196 | self.assertEqual(FILETYPE_HINT.get("mha"), "itk") |
202 | 197 |
|
203 | 198 | def test_resolve_writer_error_message(self): |
204 | | - from monai.utils import OptionalImportError |
205 | 199 | # Test with an unknown extension to see the base error message |
206 | 200 | # Since EXT_WILDCARD might have backends, it might not raise unless they are missing. |
207 | 201 | # But we can at least verify the logic is reachable. |
208 | 202 | try: |
209 | 203 | resolve_writer("unknown_ext", error_if_not_found=True) |
210 | 204 | except OptionalImportError as e: |
211 | 205 | self.assertIn("No ImageWriter backend found for 'unknown_ext'", str(e)) |
| 206 | + |
| 207 | + |
| 208 | +if __name__ == "__main__": |
| 209 | + unittest.main() |
0 commit comments