wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-09 11:28:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

code fix

Browse Source
This commit is contained in:
Contramundum
2023-12-04 16:10:55 +09:00
parent 60888843db
commit 5bbb11cf46
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/artifact/test_backend.py
+14 -7
View File
@@ -181,10 +181,10 @@ def test_successful_trial_artifact_retrieval() -> None:
assert body == b"dummy_content"
DUMMY_DATA = f"data:text/plain; charset=utf-8,{base64.b64encode(b'dummy_content').decode('utf-8')}"
DUMMY_DATA_URL = f"data:text/plain; charset=utf-8,{base64.b64encode(b'dummy_content').decode('utf-8')}"
def test_upload_artifact_invalid() -> None:
def test_upload_artifact_invalid_no_trial() -> None:
storage = optuna.storages.InMemoryStorage()
with tempfile.TemporaryDirectory() as tmpdir:
@@ -193,24 +193,31 @@ def test_upload_artifact_invalid() -> None:
app = create_app(storage, artifact_store)
study = optuna.create_study(storage=storage)
# Invalid: no trial
status, _, body = send_request(
app,
f"/api/artifacts/{study._study_id}/0",
"POST",
body=json.dumps({"file": DUMMY_DATA}),
body=json.dumps({"file": DUMMY_DATA_URL}),
content_type="application/json",
)
assert status == 500 # TODO: This should return 400
# Invalid: complete trial
def test_upload_artifact_invalid_complete_trial() -> None:
storage = optuna.storages.InMemoryStorage()
with tempfile.TemporaryDirectory() as tmpdir:
artifact_store = FileSystemArtifactStore(tmpdir)
app = create_app(storage, artifact_store)
study = optuna.create_study(storage=storage)
study.add_trial(optuna.create_trial(value=1.0, distributions={}, params={}))
trial = study.trials[-1]
status, _, body = send_request(
app,
f"/api/artifacts/{study._study_id}/{trial._trial_id}",
"POST",
body=json.dumps({"file": DUMMY_DATA}),
body=json.dumps({"file": DUMMY_DATA_URL}),
content_type="application/json",
)
assert status == 400
@@ -231,7 +238,7 @@ def test_upload_artifact() -> None:
app,
f"/api/artifacts/{study._study_id}/{trial._trial_id}",
"POST",
body=json.dumps({"file": DUMMY_DATA}),
body=json.dumps({"file": DUMMY_DATA_URL}),
content_type="application/json",
)
assert status == 201