optuna

1
import copy
2

3
import pytest
4

5
from optuna import create_study
6
from optuna import get_all_study_summaries
7
from optuna.storages import RDBStorage
8

9

10
def test_study_summary_eq_ne() -> None:
11
    storage = RDBStorage("sqlite:///:memory:")
12

13
    create_study(storage=storage)
14
    study = create_study(storage=storage)
15

16
    summaries = get_all_study_summaries(study._storage, include_best_trial=True)
17
    assert len(summaries) == 2
18

19
    assert summaries[0] == copy.deepcopy(summaries[0])
20
    assert not summaries[0] != copy.deepcopy(summaries[0])
21

22
    assert not summaries[0] == summaries[1]
23
    assert summaries[0] != summaries[1]
24

25
    assert not summaries[0] == 1
26
    assert summaries[0] != 1
27

28

29
def test_study_summary_lt_le() -> None:
30
    storage = RDBStorage("sqlite:///:memory:")
31

32
    create_study(storage=storage)
33
    study = create_study(storage=storage)
34

35
    summaries = get_all_study_summaries(study._storage, include_best_trial=True)
36
    assert len(summaries) == 2
37

38
    summary_0 = summaries[0]
39
    summary_1 = summaries[1]
40

41
    assert summary_0 < summary_1
42
    assert not summary_1 < summary_0
43

44
    with pytest.raises(TypeError):
45
        summary_0 < 1
46

47
    assert summary_0 <= summary_0
48
    assert not summary_1 <= summary_0
49

50
    with pytest.raises(TypeError):
51
        summary_0 <= 1
52

53
    # A list of StudySummaries is sortable.
54
    summaries.reverse()
55
    summaries.sort()
56
    assert summaries[0] == summary_0
57
    assert summaries[1] == summary_1
58