Skip to content

Commit 5fc5be5

Browse files
author
yashnaiduu
committed
style: fix test_image_rw.py imports and ordering
Signed-off-by: yashnaiduu <yashnaidu@example.com>
1 parent c2274ff commit 5fc5be5

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

tests/data/test_image_rw.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from parameterized import parameterized
2323

2424
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
2626
from monai.data.meta_tensor import MetaTensor
2727
from monai.transforms import LoadImage, SaveImage, moveaxis
2828
from monai.utils import MetaKeys, OptionalImportError, optional_import
@@ -188,24 +188,22 @@ def test_3d(self, reader, writer):
188188
self.nrrd_rw(test_data, reader, writer, np.float32)
189189

190190

191-
if __name__ == "__main__":
192-
unittest.main()
193-
194191
class TestResolveWriterHint(unittest.TestCase):
195-
196192
def test_filetype_hint_content(self):
197-
from monai.data.image_writer import FILETYPE_HINT
198193
self.assertEqual(FILETYPE_HINT.get("nii"), "nibabel")
199194
self.assertEqual(FILETYPE_HINT.get("nii.gz"), "nibabel")
200195
self.assertEqual(FILETYPE_HINT.get("png"), "pillow")
201196
self.assertEqual(FILETYPE_HINT.get("mha"), "itk")
202197

203198
def test_resolve_writer_error_message(self):
204-
from monai.utils import OptionalImportError
205199
# Test with an unknown extension to see the base error message
206200
# Since EXT_WILDCARD might have backends, it might not raise unless they are missing.
207201
# But we can at least verify the logic is reachable.
208202
try:
209203
resolve_writer("unknown_ext", error_if_not_found=True)
210204
except OptionalImportError as e:
211205
self.assertIn("No ImageWriter backend found for 'unknown_ext'", str(e))
206+
207+
208+
if __name__ == "__main__":
209+
unittest.main()

0 commit comments

Comments
 (0)