From 2b66f0c30314a8043efe0b8d5c9374b1468a3689 Mon Sep 17 00:00:00 2001 From: wilsaj Date: Tue, 24 Jul 2012 23:21:42 -0500 Subject: [PATCH] role duplicate logic for Image._repr_png_() and Image._repr_jpeg_() into a common method --- skimage/io/_io.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/skimage/io/_io.py b/skimage/io/_io.py index 8320ef9f..95cb5abc 100644 --- a/skimage/io/_io.py +++ b/skimage/io/_io.py @@ -53,15 +53,14 @@ class Image(np.ndarray): return tuple(object_state) def _repr_png_(self): - str_buffer = StringIO.StringIO() - imsave(str_buffer, self, format_str='png') - return_str = str_buffer.getvalue() - str_buffer.close() - return return_str + return self._repr_image_format('png') def _repr_jpeg_(self): + return self._repr_image_format('jpeg') + + def _repr_image_format(self, format_str): str_buffer = StringIO.StringIO() - imsave(str_buffer, self, format_str='jpeg') + imsave(str_buffer, self, format_str=format_str) return_str = str_buffer.getvalue() str_buffer.close() return return_str