def test_parse_YATSM_config_3(YATSM_cfg):
""" Test parsing of config without "refit" section
"""
del YATSM_cfg['YATSM']['refit']
cfg = config_parser._parse_YATSM_config(YATSM_cfg)
assert 'refit' in cfg['YATSM']
assert cfg['YATSM']['refit']['prefix'] == []
assert cfg['YATSM']['refit']['prediction'] == []
def test_parse_YATSM_config_7(YATSM_cfg):
""" Test parsing of config with "stay_regularized" section
"""
n = len(YATSM_cfg['YATSM']['refit']['prediction'])
YATSM_cfg['YATSM']['refit']['stay_regularized'] = [True] * n
config_parser._parse_YATSM_config(YATSM_cfg)
def test_parse_YATSM_config_6(YATSM_cfg):
""" Test parsing of config with "stay_regularized" section
"""
YATSM_cfg['YATSM']['refit']['stay_regularized'] = True
config_parser._parse_YATSM_config(YATSM_cfg)
def test_parse_YATSM_config_5(YATSM_cfg):
""" Test parsing of config with misspecified "refit" section
"""
YATSM_cfg['YATSM']['refit']['prefix'] = ['just_one_prefix']
with pytest.raises(KeyError):
config_parser._parse_YATSM_config(YATSM_cfg)
def test_parse_YATSM_config_4(YATSM_cfg):
""" Test parsing of config with "refit" estimators that don't exist
"""
YATSM_cfg['YATSM']['refit']['prediction'] = 'not_an_estimator'
with pytest.raises(KeyError):
config_parser._parse_YATSM_config(YATSM_cfg)
def test_parse_YATSM_config_2(YATSM_cfg):
""" Test retrieval of packaged estimators that don't exist
"""
with pytest.raises(KeyError):
YATSM_cfg['YATSM']['prediction'] = 'not_an_estimator'
config_parser._parse_YATSM_config(YATSM_cfg)
def test_parse_YATSM_config_1(YATSM_cfg):
""" Test retrieval of packaged estimators
"""
for estimator in packaged_regressions:
YATSM_cfg['YATSM']['prediction'] = estimator
config_parser._parse_YATSM_config(YATSM_cfg)
