def test_evaluate_weight():
formula = Ite(a, v1, v2)
variables = WMIVariables()
w = Weights(formula, variables)
value = w._evaluate_weight(w.weights, [True])
assert value == v1
value = w._evaluate_weight(w.weights, [False])
assert value == v2
def test_evaluate_weight_multiplication():
formula = Ite(a, v1, Times(v2, Ite(b, v1, v3)))
variables = WMIVariables()
w = Weights(formula, variables)
value = w._evaluate_weight(w.weights, [True, True])
assert value == v1
value = w._evaluate_weight(w.weights, [False, True])
assert value == Times(v2, v1)
value = w._evaluate_weight(w.weights, [False, False])
assert value == Times(v2, v3)