mirror of
https://github.com/wassname/optuna-dashboard.git
synced 2026-08-21 11:18:45 +08:00
107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
|
|
import numpy as np
|
|
import optuna
|
|
from optuna_dashboard._serializer import serialize_attrs
|
|
from optuna_dashboard._serializer import serialize_study_detail
|
|
from optuna_dashboard._serializer import serialize_study_summary
|
|
from optuna_dashboard._storage import get_study_summaries
|
|
from optuna_dashboard.preferential import create_study
|
|
from packaging import version
|
|
import pytest
|
|
|
|
|
|
def test_serialize_bytes() -> None:
|
|
serialized = serialize_attrs({"bytes": b"This is a bytes object."})
|
|
assert serialized[0]["value"] == "<binary object>"
|
|
|
|
|
|
def test_serialize_dict() -> None:
|
|
serialized = serialize_attrs(
|
|
{
|
|
"key": {"foo": "bar"},
|
|
}
|
|
)
|
|
assert len(serialized) <= 1
|
|
|
|
|
|
def test_serialize_numpy_integer() -> None:
|
|
serialized = serialize_attrs(
|
|
{
|
|
"int8": np.int8(1),
|
|
"int16": np.int16(1),
|
|
"int32": np.int32(1),
|
|
"int64": np.int64(1),
|
|
}
|
|
)
|
|
assert len(serialized) == 4
|
|
assert all([v["value"] == "1" for v in serialized])
|
|
|
|
|
|
def test_serialize_numpy_floating() -> None:
|
|
serialized = serialize_attrs(
|
|
{
|
|
"float16": np.float16(1.0),
|
|
"float32": np.float32(1.0),
|
|
"float64": np.float64(1.0),
|
|
"float128": np.float128(1.0),
|
|
}
|
|
)
|
|
assert len(serialized) == 4
|
|
assert all([v["value"] == "1.0" for v in serialized])
|
|
|
|
|
|
@pytest.mark.skipif(sys.version_info < (3, 8), reason="BoTorch dropped Python3.7 support")
|
|
@pytest.mark.skipif(
|
|
version.parse(optuna.__version__) < version.parse("3.2.0"), reason="Needs optuna.search_space"
|
|
)
|
|
def test_get_study_detail_is_preferential() -> None:
|
|
storage = optuna.storages.InMemoryStorage()
|
|
study = create_study(n_generate=4, storage=storage)
|
|
study_summaries = get_study_summaries(storage)
|
|
assert len(study_summaries) == 1
|
|
|
|
study_summary = study_summaries[0]
|
|
study_detail = serialize_study_detail(
|
|
study_summary, [], study.trials, [], [], [], False, {}, []
|
|
)
|
|
assert study_detail["is_preferential"]
|
|
|
|
|
|
def test_get_study_detail_is_not_preferential() -> None:
|
|
storage = optuna.storages.InMemoryStorage()
|
|
study = optuna.create_study(storage=storage)
|
|
study_summaries = get_study_summaries(storage)
|
|
assert len(study_summaries) == 1
|
|
|
|
study_summary = study_summaries[0]
|
|
study_detail = serialize_study_detail(
|
|
study_summary, [], study.trials, [], [], [], False, {}, []
|
|
)
|
|
assert not study_detail["is_preferential"]
|
|
|
|
|
|
@pytest.mark.skipif(sys.version_info < (3, 8), reason="BoTorch dropped Python3.7 support")
|
|
@pytest.mark.skipif(
|
|
version.parse(optuna.__version__) < version.parse("3.2.0"), reason="Needs optuna.search_space"
|
|
)
|
|
def test_get_study_summary_is_preferential() -> None:
|
|
storage = optuna.storages.InMemoryStorage()
|
|
create_study(n_generate=4, storage=storage)
|
|
study_summaries = get_study_summaries(storage)
|
|
assert len(study_summaries) == 1
|
|
|
|
study_summary = serialize_study_summary(study_summaries[0])
|
|
assert study_summary["is_preferential"]
|
|
|
|
|
|
def test_get_study_summary_is_not_preferential() -> None:
|
|
storage = optuna.storages.InMemoryStorage()
|
|
optuna.create_study(storage=storage)
|
|
study_summaries = get_study_summaries(storage)
|
|
assert len(study_summaries) == 1
|
|
study_summary = serialize_study_summary(study_summaries[0])
|
|
assert not study_summary["is_preferential"]
|