From 0ec4f085f7232319618eb48fb8a0e983f9856c6b Mon Sep 17 00:00:00 2001
From: porink0424 <daichi2120@gmail.com>
Date: Fri, 10 May 2024 13:06:45 +0900
Subject: [PATCH] Add tests

---
 tslib/storage/test/generate_assets.py | 18 ++++++++++++++++++
 tslib/storage/test/journal.test.mjs   |  7 ++++++-
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/tslib/storage/test/generate_assets.py b/tslib/storage/test/generate_assets.py
index bd135720..c7cfaca4 100644
--- a/tslib/storage/test/generate_assets.py
+++ b/tslib/storage/test/generate_assets.py
@@ -78,6 +78,24 @@ def create_optuna_storage(storage: BaseStorage) -> None:
 
     study.optimize(objective_single_nan_report, n_trials=100)
 
+    # Multi-objective study with metric names
+    study = optuna.create_study(
+        study_name="multi-objective-metric-names",
+        storage=storage,
+        directions=["minimize", "minimize"],
+    )
+    print(f"Generating {study.study_name} for {type(storage).__name__}...")
+    study.set_metric_names(["value1", "value2"])
+
+    def objective_multi(trial: optuna.Trial) -> tuple[float, float]:
+        x = trial.suggest_float("x", 0, 5)
+        y = trial.suggest_float("y", 0, 3)
+        v0 = 4 * x**2 + 4 * y**2
+        v1 = (x - 5) ** 2 + (y - 5) ** 2
+        return v0, v1
+
+    study.optimize(objective_multi, n_trials=50)
+
 
 if __name__ == "__main__":
     remove_assets()
diff --git a/tslib/storage/test/journal.test.mjs b/tslib/storage/test/journal.test.mjs
index e1539247..5df85e7f 100644
--- a/tslib/storage/test/journal.test.mjs
+++ b/tslib/storage/test/journal.test.mjs
@@ -52,8 +52,13 @@ describe("Test Journal File Storage", async () => {
     )
   })
 
+  it("Check metric_names function", () => {
+    const study = studies.find((s) => s.name === "multi-objective-metric-names")
+    assert.deepStrictEqual(study.metric_names, ["value1", "value2"])
+  })
+
   it("Check the number of studies", () => {
-    const N_STUDIES = 4
+    const N_STUDIES = 5
     assert.strictEqual(studies.length, N_STUDIES)
   })
 })