def test_add_to_tree_when_adding_tree_root(self):
dt = DecisionTree(tree={'a': {'b': {'c': dict()}}})
dt.add_to_tree([('a', 'b'), ('c', 'd')], 'e', True)
self.assertEqual(dt.get_tree(),
{'a': {
'b': {
'c': {
'd': {
'e': dict()
}
}
}
}})
def test_add_to_tree_when_path_is_empty(self):
dt = DecisionTree()
dt.add_to_tree([], 'a', True)
self.assertEqual(dt.get_tree(), {'a': dict()})
def test_add_to_tree_when_adding_decision_node(self):
dt = DecisionTree(tree={'a': {'b': {'c': dict()}}})
dt.add_to_tree([('a', 'b'), ('c', 'd')], 'e')
self.assertEqual(dt.get_tree(), {'a': {'b': {'c': {'d': 'e'}}}})
def __init__(self, in_attr_list, out_attr):
self.in_attr_list = in_attr_list
self.out_attr = out_attr
self.tree = DecisionTree()