optuna

1
import pytest
2

3
from optuna._imports import try_import
4

5

6
def test_try_import_is_successful() -> None:
7
    with try_import() as imports:
8
        pass
9
    assert imports.is_successful()
10
    imports.check()
11

12

13
def test_try_import_is_successful_other_error() -> None:
14
    with pytest.raises(NotImplementedError):
15
        with try_import() as imports:
16
            raise NotImplementedError
17
    assert imports.is_successful()  # No imports failed so `imports` is successful.
18
    imports.check()
19

20

21
def test_try_import_not_successful() -> None:
22
    with try_import() as imports:
23
        raise ImportError
24
    assert not imports.is_successful()
25
    with pytest.raises(ImportError):
26
        imports.check()
27

28
    with try_import() as imports:
29
        raise SyntaxError
30
    assert not imports.is_successful()
31
    with pytest.raises(ImportError):
32
        imports.check()
33