def test_jointree_creation():
"""
Tests join tree creation.
:return: None.
"""
n0 = BbnNode(Variable(0, 'n0', ['t', 'f']), [])
n1 = BbnNode(Variable(1, 'n1', ['t', 'f']), [])
n2 = BbnNode(Variable(2, 'n2', ['t', 'f']), [])
clique0 = Clique([n0, n1])
clique1 = Clique([n1, n2])
sep_set0 = clique0.get_sep_set(clique1)
sep_set1 = clique0.get_sep_set(clique1)
sep_set2 = clique1.get_sep_set(clique0)
sep_set3 = clique0.get_sep_set(clique0)
e0 = JtEdge(sep_set0)
e1 = JtEdge(sep_set1)
e2 = JtEdge(sep_set2)
e3 = JtEdge(sep_set3)
g = JoinTree().add_edge(e0).add_edge(e1).add_edge(e2).add_edge(e3)
nodes = g.get_nodes()
edges = g.get_edges()
assert len(nodes) == 3
assert len(edges) == 1
assert len(g.get_flattened_edges()) == 2
def transform(cliques):
"""
Transforms the cliques into a join tree.
:param cliques: List of cliques.
:return: Join tree.
"""
join_tree = JoinTree()
sep_sets = Transformer.get_sep_sets(cliques)
n = (len(cliques) - 1) * 2
total = 0
for i in range(len(sep_sets)):
join_tree.add_edge(JtEdge(sep_sets[i]))
total += 2
if total == n:
break
return join_tree
def transform(cliques):
"""
Transforms the cliques into a join tree.
:param cliques: List of cliques.
:return: Join tree.
"""
join_tree = JoinTree()
for clique in cliques:
join_tree.add_node(clique)
sep_sets = Transformer.get_sep_sets(cliques)
for i in range(len(sep_sets)):
jtEdge = JtEdge(sep_sets[i])
join_tree.add_edge(jtEdge)
return join_tree