def test_print_args_2(self):
"""Assert that _print_args prints correct values for all parameters in regression mode."""
args_list = ['tests/tests.csv', '-mode', 'regression', '-is', ',']
args = self.parser.parse_args(args_list)
with captured_output() as (out, err):
_print_args(args)
output = out.getvalue()
expected_output = """
TPOT settings:
CHECKPOINT_FOLDER = None
CONFIG_FILE = None
CROSSOVER_RATE = 0.1
EARLY_STOP = None
GENERATIONS = 100
INPUT_FILE = tests/tests.csv
INPUT_SEPARATOR = ,
MAX_EVAL_MINS = 5
MAX_TIME_MINS = None
MEMORY = None
MUTATION_RATE = 0.9
NUM_CV_FOLDS = 5
NUM_JOBS = 1
OFFSPRING_SIZE = 100
OUTPUT_FILE = None
POPULATION_SIZE = 100
RANDOM_STATE = None
SCORING_FN = neg_mean_squared_error
SUBSAMPLE = 1.0
TARGET_NAME = class
TPOT_MODE = regression
VERBOSITY = 1
"""
self.assertEqual(_sort_lines(expected_output), _sort_lines(output))
def test_print_args_2(self):
"""Assert that _print_args prints correct values for all parameters in regression mode."""
args_list = [
'tests/tests.csv',
'-mode', 'regression',
'-is', ','
]
args = self.parser.parse_args(args_list)
with captured_output() as (out, err):
_print_args(args)
output = out.getvalue()
expected_output = """
TPOT settings:
CHECKPOINT_FOLDER = None
CONFIG_FILE = None
CROSSOVER_RATE = 0.1
EARLY_STOP = None
GENERATIONS = 100
INPUT_FILE = tests/tests.csv
INPUT_SEPARATOR = ,
MAX_EVAL_MINS = 5
MAX_TIME_MINS = None
MEMORY = None
MUTATION_RATE = 0.9
NUM_CV_FOLDS = 5
NUM_JOBS = 1
OFFSPRING_SIZE = 100
OUTPUT_FILE = None
POPULATION_SIZE = 100
RANDOM_STATE = None
SCORING_FN = neg_mean_squared_error
SUBSAMPLE = 1.0
TARGET_NAME = class
TPOT_MODE = regression
VERBOSITY = 1
"""
self.assertEqual(_sort_lines(expected_output), _sort_lines(output))
