Exemplo n.º 1
0
def test_bad_columns_mock():
    tmp_dir = scratch_path / "bad_columns_mock"
    dmf = DMF(path=tmp_dir, create=True)
    with AlamoMock() as mock:
        m = surrmod.SurrogateModel(Experiment(dmf))
        with pytest.raises(KeyError):
            m.set_input_data(model_data, ["snork"], "cost")
Exemplo n.º 2
0
def _test_run(tmp_dmf, model_data):
    sml = surrmod.SurrogateModel(Experiment(tmp_dmf))
    sml.set_input_data(model_data, ['scoops'], 'cost')
    try:
        _ = sml.run()
    except errors.AlamoError:
        raise
Exemplo n.º 3
0
def test_mock_alamo():
    tmp_dir = scratch_path / "mock_alamo"
    dmf = DMF(path=tmp_dir, create=True)
    with AlamoMock() as mock:
        sml = surrmod.SurrogateModel(Experiment(dmf))
        assert not mock.doalamo_called
        sml.set_input_data(model_data, ["scoops", "bowl"], "cost")
        sml.run()
        assert mock.doalamo_called
        # assert tuple(mock.xdata) == tuple(model_data["scoops"],model_data["bowl"])
        assert tuple(mock.zdata) == tuple(model_data["cost"])
Exemplo n.º 4
0
def _test_init(tmp_dmf):
    _ = surrmod.SurrogateModel(Experiment(tmp_dmf))
Exemplo n.º 5
0
def test_bad_columns_mock(tmp_dmf, model_data):
    with AlamoMock() as mock:
        m = surrmod.SurrogateModel(Experiment(tmp_dmf))
        with pytest.raises(KeyError):
            m.set_input_data(model_data, ['snork'], 'cost')
Exemplo n.º 6
0
def test_run():
    tmp_dir = scratch_path / "run"
    dmf = DMF(path=tmp_dir, create=True)
    sml = surrmod.SurrogateModel(Experiment(dmf))
    sml.set_input_data(model_data, ["scoops", "bowl"], "cost")
    sml.run()
Exemplo n.º 7
0
def test_init():
    tmp_dir = scratch_path / "init"
    dmf = DMF(path=tmp_dir, create=True)
    surrmod.SurrogateModel(Experiment(dmf))