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

adjusted tests

parent ae6b6f49
Branches
No related tags found
1 merge request!50release for v0.7.0
Pipeline #31032 passed
...@@ -54,7 +54,8 @@ class TestPreProcessing: ...@@ -54,7 +54,8 @@ class TestPreProcessing:
assert obj_with_exp_setup.data_store.search_name("generator") == [] assert obj_with_exp_setup.data_store.search_name("generator") == []
obj_with_exp_setup.split_train_val_test() obj_with_exp_setup.split_train_val_test()
data_store = obj_with_exp_setup.data_store data_store = obj_with_exp_setup.data_store
assert data_store.search_scope("general.train") == sorted(["generator", "start", "end", "stations"]) expected_params = ["generator", "start", "end", "stations", "permute_data"]
assert data_store.search_scope("general.train") == sorted(expected_params)
assert data_store.search_name("generator") == sorted(["general.train", "general.val", "general.test", assert data_store.search_name("generator") == sorted(["general.train", "general.val", "general.test",
"general.train_val"]) "general.train_val"])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment