def test_run_options_run_and_name_keys(self):
run_options = {
'name': 'test-run',
'run': Mock(),
}
with pytest.raises(ValueError):
parse_run_options(run_options)
run_options = {
'name': None,
'run': None,
}
assert parse_run_options(run_options)
run_options = {
'name': "",
'run': None,
}
assert parse_run_options(run_options)
run_options = {
'name': "",
'run': RunContext(Mock(), Mock(assignments={'a': 1})),
}
assert parse_run_options(run_options)
def test_run_options_run_context_object(self):
run_options = {'run': TrainingRun(Mock())}
with pytest.raises(TypeError):
parse_run_options(run_options)
run_options = {
'run': RunContext(Mock(), Mock(assignments={'a': 1})),
}
assert parse_run_options(run_options)
def test_run_options_fully_parsed(self):
num_of_options = random.randint(1, len(DEFAULT_RUN_OPTIONS))
run_options_keys = random.sample(DEFAULT_RUN_OPTIONS.keys(),
num_of_options)
run_options = {k: DEFAULT_RUN_OPTIONS[k] for k in run_options_keys}
parsed_options = parse_run_options(run_options)
assert set(parsed_options.keys()) == set(DEFAULT_RUN_OPTIONS.keys())
def test_run_options_autolog_not_bool(self):
run_options = {
'autolog_metrics': 12,
}
with pytest.raises(TypeError):
parse_run_options(run_options)
def test_run_options_wrong_keys(self):
run_options = {
'autolog_metric': True,
}
with pytest.raises(ValueError):
parse_run_options(run_options)
def test_run_options_wrong_type(self):
run_options = Mock(log_params=True)
with pytest.raises(TypeError):
parse_run_options(run_options)