wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-10 12:23:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #650 from not522/test-orthants-mvn-gibbs-sampling

Browse Source 
Add test for `_orthants_MVN_Gibbs_sampling`
This commit is contained in:
c-bata
2023-12-18 17:11:23 +09:00
committed by GitHub
parent 9d4c2d5d07 bb31e36bed
commit bb2a7edb3e
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/preferential/samplers/test_gp.py
+8
View File
@@ -7,11 +7,19 @@ import pytest
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
import torch
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], dtype=torch.float64)) >= lower