def test_environment_params_path(env_1, exp_knc_0):
assert env_1.results_path.startswith(assets_dir)
assert env_1.target_column == ["diagnosis"]
assert env_1.cv_type.__name__ == "StratifiedKFold"
assert "heartbeat" in env_1.file_blacklist
assert has_experiment_result_file(
assets_dir, exp_knc_0, ["Descriptions", "PredictionsOOF", "ScriptBackups"]
)
assert not has_experiment_result_file(assets_dir, exp_knc_0, ["Heartbeats"])
def test_multi_metric(env_4, exp_svc_0, opt_svc_0):
assert len(opt_svc_0.similar_experiments) > 0
for similar_experiment in opt_svc_0.similar_experiments:
assert has_experiment_result_file(
assets_dir, similar_experiment[2],
["Descriptions", "Heartbeats", "PredictionsOOF"])
def test_lambda_callback(env_5, exp_gbc_1):
assert has_experiment_result_file(assets_dir, exp_gbc_1)
def test_holdout_test_datasets(env_2, exp_gbc_1):
assert has_experiment_result_file(assets_dir, exp_gbc_1, "ALL")
def test_do_full_save(env_0, exp_gbc_0, exp_gbc_1):
assert has_experiment_result_file(assets_dir, exp_gbc_0,
["Descriptions", "ScriptBackups"])
assert not has_experiment_result_file(assets_dir, exp_gbc_0,
["PredictionsOOF"])
assert has_experiment_result_file(assets_dir, exp_gbc_1)
def test_recorder(env_5, exp_gbc_1):
assert has_experiment_result_file(assets_dir, exp_gbc_1)
def test_classification_experiment(env_0, exp_xgb_0):
assert has_experiment_result_file(assets_dir, exp_xgb_0)