mirror of
https://github.com/wassname/optuna-dashboard.git
synced 2026-10-28 13:11:26 +08:00
28 lines
750 B
Python
28 lines
750 B
Python
import optuna
|
|
from playwright.sync_api import Page
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize("study_id", range(10))
|
|
def test_study_list(
|
|
study_id: int,
|
|
page: Page,
|
|
dummy_storage: optuna.storages.InMemoryStorage,
|
|
server_url: str,
|
|
) -> None:
|
|
page.set_viewport_size({"width": 1000, "height": 3000})
|
|
|
|
summaries = optuna.get_all_study_summaries(dummy_storage)
|
|
study_ids = {s._study_id: s.study_name for s in summaries}
|
|
|
|
study_name = study_ids[study_id]
|
|
page.goto(server_url)
|
|
page.click(f"a[href='/dashboard/studies/{study_id}']")
|
|
|
|
element = page.query_selector(".MuiTypography-body1")
|
|
assert element is not None
|
|
|
|
title = element.text_content()
|
|
assert title is not None
|
|
assert study_name in title
|