def test_get_variables_with_several_childs(self): root_node = BayesNetNode("root") child1 = BayesNetNode("child1") child2 = BayesNetNode("child2") child3 = BayesNetNode("child3") child1.influenced_by(root_node) child2.influenced_by(root_node) child3.influenced_by(root_node) bn = BayesNet([root_node]) vars = bn.get_variables() self.assertSameElements(["root", "child1", "child2", "child3"], vars)
def test_get_root_variable(self): root_node = BayesNetNode("root") bn = BayesNet([root_node]) vars = bn.get_variables() self.assertSameElements(["root"], vars)