Пример #1
0
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'] == []
Пример #2
0
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)
Пример #3
0
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)
Пример #4
0
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)
Пример #5
0
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)
Пример #6
0
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)
Пример #7
0
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)