From 1e5d582572b29b4fa85ed239a258de15df079581 Mon Sep 17 00:00:00 2001 From: Hiroki Takizawa Date: Thu, 7 Dec 2023 18:15:06 +0900 Subject: [PATCH] Update python_tests/test_csv_download.py Co-authored-by: Shuhei Watanabe <47781922+nabenabe0928@users.noreply.github.com> --- python_tests/test_csv_download.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python_tests/test_csv_download.py b/python_tests/test_csv_download.py index 3c615995..a93789e7 100644 --- a/python_tests/test_csv_download.py +++ b/python_tests/test_csv_download.py @@ -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])