def test_covar_index_name():
d = np.zeros((5, 5, 5, 10))
mvn = MVN(d, param_names=["a", "b", "c"])
assert mvn.var_index("a", "b") == 1
assert mvn.var_index("a", "c") == 3
assert mvn.var_index("b", "a") == 1
assert mvn.var_index("b", "c") == 4
assert mvn.var_index("c", "a") == 3
assert mvn.var_index("c", "b") == 4
def test_covar_index():
d = np.zeros((5, 5, 5, 10))
mvn = MVN(d)
assert mvn.var_index(0, 1) == 1
assert mvn.var_index(0, 2) == 3
assert mvn.var_index(1, 0) == 1
assert mvn.var_index(1, 2) == 4
assert mvn.var_index(2, 0) == 3
assert mvn.var_index(2, 1) == 4
def test_update_var_arr():
d = np.zeros((5, 5, 5, 10))
mvn = MVN(d)
for param in range(3):
img = np.random.rand(5, 5, 5)
mvn.update(param, var=img)
assert np.all(mvn[..., mvn.var_index(param)] == img)
def test_update_var_const():
d = np.zeros((5, 5, 5, 10))
mvn = MVN(d)
for param in range(3):
mvn.update(param, var=float(param + 1))
for param in range(3):
assert np.all(mvn[..., mvn.var_index(param)] == float(param + 1))
def test_var_index_name():
d = np.zeros((5, 5, 5, 10))
mvn = MVN(d, param_names=["a", "b", "c"])
assert mvn.var_index("a") == 0
assert mvn.var_index("b") == 2
assert mvn.var_index("c") == 5
def test_var_index():
d = np.zeros((5, 5, 5, 10))
mvn = MVN(d)
assert mvn.var_index(0) == 0
assert mvn.var_index(1) == 2
assert mvn.var_index(2) == 5