Example #1
0
 def test_bfs_1(self):
     t = Tree('a')
     b = t._add(t.root(), 'b')
     t._add(b, 'd')
     c = t._add(t.root(), 'c')
     t._add(c, 'e')
     f = t._add(c, 'f')
     t._add(f, 'g')
     self.assertEqual([x.element() for x in t.bfs()],
                      ['a', 'b', 'c', 'd', 'e', 'f', 'g'])
Example #2
0
    def test_add_between(self):

        t = Tree('r')
        r = t.root()
        a = t.add(r, 'a')
        b = t.add(a, 'b')
        c = t.add(a, 'c')
        d = t.add(a, 'd')
        e = t.add_between(r, a, 'e')
        f = t.add(e, 'f')
        self.assertEqual([x.element() for x in t.bfs()],
                         ['r', 'e', 'a', 'f', 'b', 'c', 'd'])
        self.assertEqual(len(t), 7)
        self.assertEqual(t.height(), 3)