from .basic_params import BasicParams


class TestParams(BasicParams):
    """
    This class is a son class of BasicParams.
    This class includes parameters for testing and parameters inherited from the father class.
    """
    def initialize(self, parser):
        parser = BasicParams.initialize(self, parser)

        # Testing parameters
        parser.add_argument('--save_latent_space', action='store_true', help='save the latent space of input data to disc')

        # Logging and visualization
        parser.add_argument('--print_freq', type=int, default=1,
                            help='frequency of showing results on console')

        self.isTrain = False
        self.isTest = True
        return parser