Skip to content
Snippets Groups Projects
Commit 1fb90fd3 authored by leufen1's avatar leufen1
Browse files

test update

parent c3bca050
Branches
Tags
3 merge requests!192include Develop,!191Resolve "release v1.1.0",!182Resolve "Mixed sampling types"
Pipeline #51132 passed
...@@ -61,7 +61,8 @@ class TestTraining: ...@@ -61,7 +61,8 @@ class TestTraining:
obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_train_gen"), "general.train") obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_train_gen"), "general.train")
obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_val_gen"), "general.val") obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_val_gen"), "general.val")
obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_test_gen"), "general.test") obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_test_gen"), "general.test")
os.makedirs(path) if not os.path.exists(path):
os.makedirs(path)
obj.data_store.set("experiment_path", path, "general") obj.data_store.set("experiment_path", path, "general")
os.makedirs(batch_path) os.makedirs(batch_path)
obj.data_store.set("batch_path", batch_path, "general") obj.data_store.set("batch_path", batch_path, "general")
...@@ -170,7 +171,8 @@ class TestTraining: ...@@ -170,7 +171,8 @@ class TestTraining:
@pytest.fixture @pytest.fixture
def ready_to_init(self, data_collection, model, callbacks, path, model_path, batch_path): def ready_to_init(self, data_collection, model, callbacks, path, model_path, batch_path):
os.makedirs(path) if not os.path.exists(path):
os.makedirs(path)
os.makedirs(model_path) os.makedirs(model_path)
obj = RunEnvironment() obj = RunEnvironment()
obj.data_store.set("data_collection", data_collection, "general.train") obj.data_store.set("data_collection", data_collection, "general.train")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment