Add test for RGB warning

fepegar · Aug 19, 2020 · 2cd6d36 · 2cd6d36
1 parent f101198
commit 2cd6d36
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/data/test_io.py b/tests/data/test_io.py
@@ -38,9 +38,14 @@ def write_dicom(self):
         writer.Execute(image)
 
     def test_read_image(self):
-        # I need to find something readable by nib but not sitk (MINC?)
+        # I need to find something readable by nib but not sitk
         io.read_image(self.nii_path)
 
+    def test_save_rgb(self):
+        im = ScalarImage(tensor=torch.rand(1, 4, 5, 1))
+        with self.assertWarns(UserWarning):
+            im.save(self.dir / 'test.jpg')
+
     def test_read_dicom_file(self):
         io.read_image(self.dicom_path)