def test_run_experiment_lr_predict_with_dictionary():
"""Test rsmpredict using the dictionary object, rather than a file."""
source = 'lr-predict-dict'
# set up a temporary directory since
# we will be using getcwd
temp_dir = tempfile.TemporaryDirectory(prefix=getcwd())
old_file_dict = {
'feature_file': 'data/files/test.csv',
'experiment_dir':
'data/experiments/lr-predict-dict/existing_experiment'
}
new_file_dict = copy_data_files(temp_dir.name, old_file_dict,
rsmtool_test_dir)
config_dict = {
"id_column": "ID",
"input_features_file": new_file_dict['feature_file'],
"experiment_dir": new_file_dict['experiment_dir'],
"experiment_id": "lr"
}
check_run_prediction(source,
given_test_dir=rsmtool_test_dir,
config_obj_or_dict=config_dict)
def test_run_experiment_lr_predict_with_object():
"""Test rsmpredict using the Configuration object, rather than a file."""
source = 'lr-predict-object'
configdir = join(rsmtool_test_dir, 'data', 'experiments', source)
config_dict = {
"id_column": "ID",
"input_features_file": "../../files/test.csv",
"experiment_dir": "existing_experiment",
"experiment_id": "lr"
}
config_obj = Configuration(config_dict,
context='rsmpredict',
configdir=configdir)
check_run_prediction(source,
given_test_dir=rsmtool_test_dir,
config_obj_or_dict=config_obj)
def test_run_experiment_parameterized(*args, **kwargs):
if TEST_DIR:
kwargs['given_test_dir'] = TEST_DIR
check_run_prediction(*args, **kwargs)
def test_run_experiment_parameterized(*args, **kwargs):
if TEST_DIR:
kwargs['given_test_dir'] = TEST_DIR
check_run_prediction(*args, **kwargs)