test vgg19 transformer working

skluc/test/test_data/test_transformation/TestVGG19Transformer.py

 import unittest
 
-from skluc.main.data.mldatasets.Cifar100FineDataset import Cifar100FineDataset
 from skluc.main.data.mldatasets.Cifar10Dataset import Cifar10Dataset
-from skluc.main.data.mldatasets.SVHNDataset import SVHNDataset
+
 from skluc.main.data.transformation.VGG19Transformer import VGG19Transformer
+
 from skluc.main.utils import logger
 
 
 class TestVGG19Transformer(unittest.TestCase):
     def setUp(self):
         self.lst_name_cut_layers = [
-            "block5_conv4",
-            "block4_conv4",
-            "block5_pool",
             "block3_pool"
         ]
         self.dict_datasets = {
             "cifar10": Cifar10Dataset,
-            "cifar100": Cifar100FineDataset,
-            "svhn": SVHNDataset
         }
 
     def test_transform(self):
@@ -45,6 +40,7 @@ class TestVGG19Transformer(unittest.TestCase):
             for name_cut_layer in self.lst_name_cut_layers:
                 logger.info("Testing cut layer {}".format(name_cut_layer))
                 trans = VGG19Transformer(data_name=data_name, cut_layer_name=name_cut_layer)
+                logger.debug(trans.keras_model)
             del trans