Update python_tests/test_csv_download.py

Co-authored-by: Shuhei Watanabe <47781922+nabenabe0928@users.noreply.github.com>
This commit is contained in:
Hiroki Takizawa
2023-12-07 18:15:06 +09:00
committed by GitHub
co-authored by Shuhei Watanabe
parent a4fb8fdcdb
commit 1e5d582572
+3 -1
View File
@@ -71,7 +71,9 @@ def test_download_csv_fail(study_id: int) -> None:
study = optuna.create_study(storage=storage)
optuna.logging.set_verbosity(optuna.logging.ERROR)
study.optimize(objective, n_trials=10)
_validate_output(storage, 404 if study_id != 0 else 200, study_id)
expect_no_result = study_id != 0
cols = ["Param x", "Param y", "Value"]
_validate_output(storage, 404 if expect_no_result else 200, study_id, expect_no_result, cols)
@pytest.mark.parametrize("is_multi_obj", [True, False])