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)