def test_triangulator():
bbn = BbnUtil.get_huang_graph()
PotentialInitializer.init(bbn)
ug = Moralizer.moralize(bbn)
cliques = Triangulator.triangulate(ug)
def test_initializer():
"""
Tests initialization.
:return: None.
"""
bbn = BbnUtil.get_huang_graph()
PotentialInitializer.init(bbn)
ug = Moralizer.moralize(bbn)
cliques = Triangulator.triangulate(ug)
join_tree = Transformer.transform(cliques)
Initializer.initialize(join_tree)
e_potentials = {
'3=on,4=on,5=on': 0.01000,
'3=on,4=on,5=off': 0.99000,
'3=on,4=off,5=on': 0.01000,
'3=on,4=off,5=off': 0.99000,
'3=off,4=on,5=on': 0.01000,
'3=off,4=on,5=off': 0.99000,
'3=off,4=off,5=on': 0.99000,
'3=off,4=off,5=off': 0.01000,
'4=on,6=on,7=on': 0.05000,
'4=on,6=on,7=off': 0.95000,
'4=on,6=off,7=on': 0.95000,
'4=on,6=off,7=off': 0.05000,
'4=off,6=on,7=on': 0.95000,
'4=off,6=on,7=off': 0.05000,
'4=off,6=off,7=on': 0.95000,
'4=off,6=off,7=off': 0.05000,
'2=on,4=on,6=on': 0.80000,
'2=on,4=on,6=off': 0.20000,
'2=on,4=off,6=on': 0.80000,
'2=on,4=off,6=off': 0.20000,
'2=off,4=on,6=on': 0.10000,
'2=off,4=on,6=off': 0.90000,
'2=off,4=off,6=on': 0.10000,
'2=off,4=off,6=off': 0.90000,
'0=on,1=on,2=on': 0.17500,
'0=off,1=on,2=on': 0.04000,
'0=on,1=on,2=off': 0.07500,
'0=off,1=on,2=off': 0.16000,
'0=on,1=off,2=on': 0.17500,
'0=off,1=off,2=on': 0.06000,
'0=on,1=off,2=off': 0.07500,
'0=off,1=off,2=off': 0.24000,
'1=on,2=on,3=on': 0.90000,
'1=off,2=on,3=on': 0.50000,
'1=on,2=on,3=off': 0.10000,
'1=off,2=on,3=off': 0.50000,
'1=on,2=off,3=on': 0.90000,
'1=off,2=off,3=on': 0.50000,
'1=on,2=off,3=off': 0.10000,
'1=off,2=off,3=off': 0.50000,
'2=on,3=on,4=on': 0.30000,
'2=off,3=on,4=on': 0.60000,
'2=on,3=on,4=off': 0.70000,
'2=off,3=on,4=off': 0.40000,
'2=on,3=off,4=on': 0.30000,
'2=off,3=off,4=on': 0.60000,
'2=on,3=off,4=off': 0.70000,
'2=off,3=off,4=off': 0.40000,
'1=on,2=on': 1.00000,
'1=on,2=off': 1.00000,
'1=off,2=on': 1.00000,
'1=off,2=off': 1.00000,
'2=on,3=on': 1.00000,
'2=on,3=off': 1.00000,
'2=off,3=on': 1.00000,
'2=off,3=off': 1.00000,
'2=on,4=on': 1.00000,
'2=on,4=off': 1.00000,
'2=off,4=on': 1.00000,
'2=off,4=off': 1.00000,
'3=on,4=on': 1.00000,
'3=on,4=off': 1.00000,
'3=off,4=on': 1.00000,
'3=off,4=off': 1.00000,
'4=on,6=on': 1.00000,
'4=on,6=off': 1.00000,
'4=off,6=on': 1.00000,
'4=off,6=off': 1.00000
}
o_potentials = '\n'.join([str(v) for _, v in join_tree.potentials.items()
]).split('\n')
o_potentials = [p.split('|') for p in o_potentials]
o_potentials = {tokens[0]: float(tokens[1]) for tokens in o_potentials}
assert len(e_potentials) == len(o_potentials)
for k, lhs in o_potentials.items():
assert k in e_potentials
rhs = e_potentials[k]
assert lhs == rhs