diff --git a/python_tests/preferential/samplers/test_gp.py b/python_tests/preferential/samplers/test_gp.py index 33f33cfc..08acca46 100644 --- a/python_tests/preferential/samplers/test_gp.py +++ b/python_tests/preferential/samplers/test_gp.py @@ -8,10 +8,18 @@ import torch if sys.version_info >= (3, 8): from optuna_dashboard.preferential.samplers.gp import _one_side_trunc_norm_sampling + from optuna_dashboard.preferential.samplers.gp import _orthants_MVN_Gibbs_sampling else: pytest.skip("BoTorch dropped Python3.7 support", allow_module_level=True) +def test_orthants_MVN_Gibbs_sampling() -> None: + cov_inv = torch.Tensor([[0.1, 0.3], [0.4, 0.2]]) + initial_sample = torch.Tensor([0.5, 0.6]) + ret = _orthants_MVN_Gibbs_sampling(cov_inv, 2, initial_sample) + assert ret.shape == (3, 2) + + def test_one_side_trunc_norm_sampling() -> None: for lower in np.linspace(-10, 10, 100): assert _one_side_trunc_norm_sampling(torch.Tensor([lower])) >= lower