Make test_save_image more robust (#1037)

test/test_utils.py

 import os
+import tempfile
 import torch
 import torchvision.utils as utils
 import unittest
@@ -37,11 +38,10 @@ class Tester(unittest.TestCase):
         assert torch.equal(norm_min, rounded_grid_min), 'Normalized min is not equal to 0'
 
     def test_save_image(self):
+        with tempfile.NamedTemporaryFile(suffix='.png') as f:
             t = torch.rand(2, 3, 64, 64)
-        file_name = 'test_image.png'
-        utils.save_image(t, file_name)
-        assert os.path.exists(file_name), 'The image is not present after save'
-        os.remove(file_name)
+            utils.save_image(t, f.name)
+            assert os.path.exists(f.name), 'The image is not present after save'
 
 
 if __name__ == '__main__':