def test_set_workspace_data():
data = {"dummy": "data"}
set_workspace_data(data)
assert get_loaded_workspace_data() == data
# clean up
reset_workspace()
def test_load_workspace():
with NamedTemporaryFile(suffix="yaml", mode='w') as workspace:
workspace.write("experiments:\n")
workspace.write(" chaosiq:\n")
workspace.write(" test:\n")
workspace.write(" dummy.json:\n")
workspace.write(" experiment_id: azerty123456\n")
workspace.seek(0)
load_workspace(workspace.name)
data = get_loaded_workspace_data()
assert data == {
"experiments": {
"chaosiq": {
"test": {
"dummy.json": {
"experiment_id": "azerty123456"
}
}
}
}
}
# clean up
reset_workspace()
def test_save_workspace():
set_workspace_data({"experiments": {}})
with NamedTemporaryFile(suffix="yaml", mode='w') as workspace:
save_workspace(workspace.name)
# clean up
reset_workspace()
def test_get_loaded_workspace_data_cannot_be_mutated():
set_workspace_data({"dummy": "data"})
data1 = get_loaded_workspace_data()
data2 = get_loaded_workspace_data()
assert data1 == data2
# mutate one of the data dict and check the other one is not
# neither the global workspace variable
data1["extra"] = "value"
assert data1 != data2
assert data1 != workspace_metadata
assert data1 != get_loaded_workspace_data()
assert data2 == get_loaded_workspace_data()
# clean up
reset_workspace()
def test_register_experiment(mock_exists, organizations, default_org_id,
default_team_id):
mock_exists.return_value = True
experiment = {
"title":
"hello",
"extensions": [{
"name": "chaosiq",
"experiment_id": "azerty123456",
"experiment_path": "/tmp/experiment.json"
}]
}
register_experiment_to_workspace(experiment, organizations)
assert get_loaded_workspace_data() != {}
data = get_loaded_workspace_data()
assert "experiment.json" in \
data["experiments"][default_org_id][default_team_id]
# clean up
reset_workspace()
def test_fetch_experiment_metadata(mock_exists, organizations, default_org_id,
default_team_id):
mock_exists.return_value = True
set_workspace_data({
"experiments": {
default_org_id: {
default_team_id: {
"experiment.json": {
"experiment_id": "azerty123456"
}
}
}
}
})
experiment = {"title": "hello"}
experiment_meta = get_experiment_metadata_from_workspace(
experiment, organizations, experiment_path="/tmp/experiment.json")
assert experiment_meta["experiment_id"] == "azerty123456"
# clean up
reset_workspace()
def test_reset_workspace_data():
set_workspace_data({"dummy": "data"})
assert get_loaded_workspace_data() not in EMPTY_VALUES
reset_workspace()
assert get_loaded_workspace_data() in EMPTY_VALUES