class NodeCase(unittest.TestCase):
def setUp(self):
self.node1 = Node("Test One", "identifier 1")
self.node2 = Node("Test Two", "identifier 2")
def test_initialization(self):
self.assertEqual(self.node1.tag, "Test One")
self.assertEqual(self.node1.identifier, "identifier 1")
self.assertEqual(self.node1.expanded, True)
self.assertEqual(self.node1.bpointer, None)
self.assertEqual(self.node1.fpointer, [])
self.assertEqual(self.node1.data, None)
def test_set_tag(self):
self.node1.tag = "Test 1"
self.assertEqual(self.node1.tag, "Test 1")
self.node1.tag = "Test One"
def test_object_as_node_tag(self):
node = Node(tag=(0, 1))
self.assertEqual(node.tag, (0, 1))
self.assertTrue(node.__repr__().startswith('Node'))
def test_set_identifier(self):
self.node1.identifier = "ID1"
self.assertEqual(self.node1.identifier, "ID1")
self.node1.identifier = "identifier 1"
def test_set_fpointer(self):
self.node1.update_fpointer("identifier 2")
self.assertEqual(self.node1.fpointer, ['identifier 2'])
self.node1.fpointer = []
def test_set_bpointer(self):
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node2.bpointer, 'identifier 1')
self.node2.bpointer = None
def test_set_is_leaf(self):
self.node1.update_fpointer("identifier 2")
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node1.is_leaf(), False)
self.assertEqual(self.node2.is_leaf(), True)
def test_data(self):
class Flower(object):
def __init__(self, color):
self.color = color
def __str__(self):
return "%s" % self.color
self.node1.data = Flower("red")
self.assertEqual(self.node1.data.color, "red")
class NodeCase(unittest.TestCase):
def setUp(self):
self.node1 = Node("Test One", "identifier 1")
self.node2 = Node("Test Two", "identifier 2")
def test_initialization(self):
self.assertEqual(self.node1.tag, "Test One")
self.assertEqual(self.node1.identifier, "identifier 1")
self.assertEqual(self.node1.expanded, True)
self.assertEqual(self.node1.bpointer, None)
self.assertEqual(self.node1.fpointer, [])
self.assertEqual(self.node1.data, None)
def test_set_tag(self):
self.node1.tag = "Test 1"
self.assertEqual(self.node1.tag, "Test 1")
self.node1.tag = "Test One"
def test_object_as_node_tag(self):
node = Node(tag=(0, 1))
self.assertEqual(node.tag, (0, 1))
self.assertTrue(node.__repr__().startswith('Node'))
def test_set_identifier(self):
self.node1.identifier = "ID1"
self.assertEqual(self.node1.identifier, "ID1")
self.node1.identifier = "identifier 1"
def test_set_fpointer(self):
self.node1.update_fpointer("identifier 2")
self.assertEqual(self.node1.fpointer, ['identifier 2'])
self.node1.fpointer = []
def test_set_bpointer(self):
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node2.bpointer, 'identifier 1')
self.node2.bpointer = None
def test_set_is_leaf(self):
self.node1.update_fpointer("identifier 2")
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node1.is_leaf(), False)
self.assertEqual(self.node2.is_leaf(), True)
def test_data(self):
class Flower(object):
def __init__(self, color):
self.color = color
def __str__(self):
return "%s" % self.color
self.node1.data = Flower("red")
self.assertEqual(self.node1.data.color, "red")
class NodeCase(unittest.TestCase):
def setUp(self):
self.node1 = Node("Test One", "identifier 1")
self.node2 = Node("Test Two", "identifier 2")
def test_initialization(self):
self.assertEqual(self.node1.tag, "Test One")
self.assertEqual(self.node1.identifier, "identifier 1")
self.assertEqual(self.node1.expanded, True)
self.assertEqual(self.node1.bpointer, None)
self.assertEqual(self.node1.fpointer, [])
self.assertEqual(self.node1.data, None)
def test_set_tag(self):
self.node1.tag = "Test 1"
self.assertEqual(self.node1.tag, "Test 1")
self.node1.tag = "Test One"
def test_set_identifier(self):
self.node1.identifier = "ID1"
self.assertEqual(self.node1.identifier, "ID1")
self.node1.identifier = "identifier 1"
def test_set_fpointer(self):
self.node1.update_fpointer("identifier 2")
self.assertEqual(self.node1.fpointer, ["identifier 2"])
self.node1.fpointer = []
def test_set_bpointer(self):
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node2.bpointer, "identifier 1")
self.node2.bpointer = None
def test_set_is_leaf(self):
self.node1.update_fpointer("identifier 2")
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node1.is_leaf(), False)
self.assertEqual(self.node2.is_leaf(), True)
def test_data(self):
class Flower(object):
def __init__(self, color):
self.color = color
def __str__(self):
return "%s" % self.color
self.node1.data = Flower("red")
self.assertEqual(self.node1.data.color, "red")
def tearDown(self):
pass
def _walk_to_leaf_node(self, node: Node, instance):
if (node.is_leaf()):
return node.decision
if (self.attribute_types[node.attribute] == 'discrete'):
for c in self.children(node.identifier):
if (instance[node.attribute] == c.parent_attribute_value):
return self._walk_to_leaf_node(c, instance)
else:
for c in self.children(node.identifier):
if (c.division_criterion(instance[node.attribute])):
return self._walk_to_leaf_node(c, instance)
class NodeCase(unittest.TestCase):
def setUp(self):
self.node1 = Node("Test One", "identifier 1")
self.node2 = Node("Test Two", "identifier 2")
def test_initialization(self):
self.assertEqual(self.node1.tag, "Test One")
self.assertEqual(self.node1.identifier, "identifier 1")
self.assertEqual(self.node1.expanded, True)
self.assertEqual(self.node1.bpointer, None)
self.assertEqual(self.node1.fpointer, [])
def test_set_tag(self):
self.node1.tag = "Test 1"
self.assertEqual(self.node1.tag, "Test 1")
self.node1.tag = "Test One"
def test_set_identifier(self):
self.node1.identifier = "ID1"
self.assertEqual(self.node1.identifier, "ID1")
self.node1.identifier = "identifier 1"
def test_set_fpointer(self):
self.node1.update_fpointer("identifier 2")
self.assertEqual(self.node1.fpointer, ['identifier 2'])
self.node1.fpointer = []
def test_set_bpointer(self):
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node2.bpointer, 'identifier 1')
self.node2.bpointer = None
def test_set_is_leaf(self):
self.node1.update_fpointer("identifier 2")
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node1.is_leaf(), False)
self.assertEqual(self.node2.is_leaf(), True)
def tearDown(self):
pass
class NodeCase(unittest.TestCase):
def setUp(self):
self.node1 = Node("Test One", "identifier 1")
self.node2 = Node("Test Two", "identifier 2")
def test_initialization(self):
self.assertEqual(self.node1.tag, "Test One")
self.assertEqual(self.node1.identifier, "identifier 1")
self.assertEqual(self.node1.expanded, True)
self.assertEqual(self.node1.bpointer, None)
self.assertEqual(self.node1.fpointer, [])
def test_set_tag(self):
self.node1.tag = "Test 1"
self.assertEqual(self.node1.tag, "Test 1")
self.node1.tag = "Test One"
def test_set_identifier(self):
self.node1.identifier = "ID1"
self.assertEqual(self.node1.identifier, "ID1")
self.node1.identifier = "identifier 1"
def test_set_fpointer(self):
self.node1.update_fpointer("identifier 2")
self.assertEqual(self.node1.fpointer, ['identifier 2'])
self.node1.fpointer = []
def test_set_bpointer(self):
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node2.bpointer, 'identifier 1')
self.node2.bpointer = None
def test_set_is_leaf(self):
self.node1.update_fpointer("identifier 2")
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node1.is_leaf(), False)
self.assertEqual(self.node2.is_leaf(), True)
def tearDown(self):
pass
def _isLeaf(self, path: Node) -> bool:
return path.is_leaf()
class NodeCase(unittest.TestCase):
def setUp(self):
self.node1 = Node("Test One", "identifier 1")
self.node2 = Node("Test Two", "identifier 2")
def test_initialization(self):
self.assertEqual(self.node1.tag, "Test One")
self.assertEqual(self.node1.identifier, "identifier 1")
# retro-compatibility
self.assertEqual(self.node1.bpointer, None)
self.assertEqual(self.node1.fpointer, [])
self.assertEqual(self.node1.expanded, True)
self.assertEqual(self.node1._predecessor, {})
self.assertEqual(self.node1._successors, defaultdict(list))
self.assertEqual(self.node1.data, None)
def test_set_tag(self):
self.node1.tag = "Test 1"
self.assertEqual(self.node1.tag, "Test 1")
self.node1.tag = "Test One"
def test_object_as_node_tag(self):
node = Node(tag=(0, 1))
self.assertEqual(node.tag, (0, 1))
self.assertTrue(node.__repr__().startswith('Node'))
def test_set_identifier(self):
self.node1.identifier = "ID1"
self.assertEqual(self.node1.identifier, "ID1")
self.node1.identifier = "identifier 1"
def test_set_fpointer(self):
# retro-compatibility
self.node1.update_fpointer("identifier 2")
self.assertEqual(self.node1.fpointer, ['identifier 2'])
self.node1.fpointer = []
self.assertEqual(self.node1.fpointer, [])
def test_update_successors(self):
self.node1.update_successors("identifier 2", tree_id="tree 1")
self.assertEqual(self.node1.successors("tree 1"), ['identifier 2'])
self.assertEqual(self.node1._successors["tree 1"], ['identifier 2'])
self.node1.set_successors([], tree_id="tree 1")
self.assertEqual(self.node1._successors["tree 1"], [])
def test_set_bpointer(self):
# retro-compatibility
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node2.bpointer, 'identifier 1')
self.node2.bpointer = None
self.assertEqual(self.node2.bpointer, None)
def test_set_predecessor(self):
self.node2.set_predecessor("identifier 1", "tree 1")
self.assertEqual(self.node2.predecessor("tree 1"), 'identifier 1')
self.assertEqual(self.node2._predecessor["tree 1"], 'identifier 1')
self.node2.set_predecessor(None, "tree 1")
self.assertEqual(self.node2.predecessor("tree 1"), None)
def test_set_is_leaf(self):
self.node1.update_fpointer("identifier 2")
self.node2.update_bpointer("identifier 1")
self.assertEqual(self.node1.is_leaf(), False)
self.assertEqual(self.node2.is_leaf(), True)
def test_tree_wise_is_leaf(self):
self.node1.update_successors("identifier 2", tree_id="tree 1")
self.node2.set_predecessor("identifier 1", "tree 1")
self.assertEqual(self.node1.is_leaf("tree 1"), False)
self.assertEqual(self.node2.is_leaf("tree 1"), True)
def test_data(self):
class Flower(object):
def __init__(self, color):
self.color = color
def __str__(self):
return "%s" % self.color
self.node1.data = Flower("red")
self.assertEqual(self.node1.data.color, "red")
