test resize avec check des nouvelles tailles

skluc/data/mldatasets/ImageDataset.py

@@ -79,6 +79,7 @@ class ImageDataset(Dataset):
                 length_by_chanel = int(length_by_chanel)
             images_mat = np.reshape(images_vec, (images_vec.shape[0], length_by_chanel, self.DEPTH), order='F')
             images = np.reshape(images_mat, (images_mat.shape[0], self.HEIGHT, self.WIDTH, self.DEPTH), order='C')
+            logger.debug("Images mat shape: {}".format(images.shape))
             setattr(self, kw, LabeledData(images, labels))
 
     def flatten(self):

skluc/test/test_transformation/TestResizeTransformer.py

@@ -33,6 +33,9 @@ class TestResizeTransformer(unittest.TestCase):
                 d.to_image()
                 trans = ResizeTransformer(data_name=data_name, output_shape=size)
                 d.apply_transformer(transformer=trans)
+                new_size = d.train.data[0].shape[:-1]
+                self.assertEqual(new_size[0], size[0])
+                self.assertEqual(new_size[1], size[1])
                 del trans
 
     def test_init(self):