def test_add_head_single_child_head_first(self):
tree = haltree.HalTree()
tree.add(self.head)
tree.add(self.child)
self.assertEquals(self.head, tree.head)
self.assertEquals(self.head, self.child.parent)
self.assertEquals(1, len(self.head.children))
self.assertEquals(self.child, self.head.children[0])
def test_add_device_no_head(self):
tree = haltree.HalTree()
tree.add(self.child)
self.assertEquals(None, tree.head)
def test_add_head(self):
tree = haltree.HalTree()
tree.add(self.head)
self.assertEquals(self.head, tree.head)