def test_build_from_interactions():
model = PairWiseFiniteModel(10, 5)
j1 = np.random.random(size=(5, 5))
j2 = np.random.random(size=(5, 5))
model.add_interaction(0, 1, j1)
model.add_interaction(1, 0, j1)
model.add_interaction(1, 2, j2)
assert np.allclose(model.field, np.zeros((10, 5)))
assert model.edges == [(0, 1), (1, 2)]
assert np.allclose(model.get_interaction_matrix(0, 1), j1 + j1.T)
assert np.allclose(model.get_interaction_matrix(2, 1), j2.T)
def test_build_from_factors():
model = PairWiseFiniteModel(5, 2)
x = model.get_symbolic_variables()
# Field.
model *= np.exp(5 * x[1])
model *= np.exp(10 * x[2])
model *= DiscreteFactor(model, [3], np.exp([7, 8]))
# Interactions.
model *= np.exp(2 * x[0] * x[1])
model *= (1 + x[2] + x[3])
model *= np.exp(10 * x[0] * x[1]) # Should accumulate.
model *= DiscreteFactor(model, [0, 4], np.exp([[1, 2], [3, 4]]))
assert np.allclose(model.field,
np.array([[0, 0], [0, 5], [0, 10], [7, 8], [0, 0]]))
assert model.edges == [(0, 1), (2, 3), (0, 4)]
assert np.allclose(model.get_interaction_matrix(0, 1), [[0, 0], [0, 12]])
assert np.allclose(model.get_interaction_matrix(2, 3),
np.log([[1, 2], [2, 3]]))
assert np.allclose(model.get_interaction_matrix(0, 4), [[1, 2], [3, 4]])