Skip to content
Snippets Groups Projects
Commit ca83b390 authored by lukas leufen's avatar lukas leufen
Browse files

adjust test

parent 028b9df8
No related branches found
No related tags found
3 merge requests!125Release v0.10.0,!124Update Master to new version v0.10.0,!105Resolve "Discussion: right location of epoch parameter"
...@@ -22,6 +22,7 @@ class TestModelSetup: ...@@ -22,6 +22,7 @@ class TestModelSetup:
obj.data_store.set("model_class", MyLittleModel) obj.data_store.set("model_class", MyLittleModel)
obj.data_store.set("lr_decay", "dummy_str", "general.model") obj.data_store.set("lr_decay", "dummy_str", "general.model")
obj.data_store.set("hist", "dummy_str", "general.model") obj.data_store.set("hist", "dummy_str", "general.model")
obj.data_store.set("epochs", 2)
obj.model_name = "%s.h5" obj.model_name = "%s.h5"
yield obj yield obj
RunEnvironment().__del__() RunEnvironment().__del__()
...@@ -49,8 +50,7 @@ class TestModelSetup: ...@@ -49,8 +50,7 @@ class TestModelSetup:
@pytest.fixture @pytest.fixture
def setup_with_model(self, setup): def setup_with_model(self, setup):
setup.model = AbstractModelClass() setup.model = AbstractModelClass()
setup.model.epochs = 2 setup.model.test_param = "42"
setup.model.batch_size = int(256)
yield setup yield setup
RunEnvironment().__del__() RunEnvironment().__del__()
...@@ -80,15 +80,15 @@ class TestModelSetup: ...@@ -80,15 +80,15 @@ class TestModelSetup:
setup_with_model.scope = "model_test" setup_with_model.scope = "model_test"
with pytest.raises(EmptyScope): with pytest.raises(EmptyScope):
self.current_scope_as_set(setup_with_model) # will fail because scope is not created self.current_scope_as_set(setup_with_model) # will fail because scope is not created
setup_with_model.get_model_settings() # this saves now the parameters epochs and batch_size into scope setup_with_model.get_model_settings() # this saves now the parameter test_param into scope
assert {"epochs", "batch_size"} <= self.current_scope_as_set(setup_with_model) assert {"test_param", "model_name"} <= self.current_scope_as_set(setup_with_model)
def test_build_model(self, setup_with_gen): def test_build_model(self, setup_with_gen):
assert setup_with_gen.model is None assert setup_with_gen.model is None
setup_with_gen.build_model() setup_with_gen.build_model()
assert isinstance(setup_with_gen.model, AbstractModelClass) assert isinstance(setup_with_gen.model, AbstractModelClass)
expected = {"window_history_size", "window_lead_time", "channels", "dropout_rate", "regularizer", "initial_lr", expected = {"window_history_size", "window_lead_time", "channels", "dropout_rate", "regularizer", "initial_lr",
"optimizer", "epochs", "batch_size", "activation"} "optimizer", "batch_size", "activation"}
assert expected <= self.current_scope_as_set(setup_with_gen) assert expected <= self.current_scope_as_set(setup_with_gen)
def test_set_channels(self, setup_with_gen_tiny): def test_set_channels(self, setup_with_gen_tiny):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment