Exemplo n.º 1
0
    def test_traverse(self):
        self.assertEqual({"root", "child1", "child2", "child3", "child4"},
                         set(t.obj
                             for t in Tree.from_tuples(self.tree).traverse()))

        self.assertEqual(
            {"t", "1", "2", "3", "4"},
            set(Tree.from_tuples(self.tree).traverse(lambda x: x.obj[-1])))
Exemplo n.º 2
0
    def test_traverse(self):
        self.assertEqual(
            {"root", "child1", "child2", "child3", "child4"},
            set(t.obj for t in Tree.from_tuples(self.tree).traverse())
        )

        self.assertEqual(
            {"t", "1", "2", "3", "4"},
            set(Tree.from_tuples(self.tree).traverse(lambda x: x.obj[-1]))
        )
Exemplo n.º 3
0
    def test_get_level(self):
        tree = Tree.from_tuples(self.tree)

        self.assertEqual(set(t.obj for t in tree.get_level(0)), {"root"})
        self.assertEqual(set(t.obj for t in tree.get_level(1)), {"child1", "child2"})
        self.assertEqual(set(t.obj for t in tree.get_level(2)), {"child3", "child4"})
        self.assertEqual(set(t.obj for t in tree.get_level(5)), set())
Exemplo n.º 4
0
    def test_get_level(self):
        tree = Tree.from_tuples(self.tree)

        self.assertEqual(set(t.obj for t in tree.get_level(0)), {"root"})
        self.assertEqual(set(t.obj for t in tree.get_level(1)),
                         {"child1", "child2"})
        self.assertEqual(set(t.obj for t in tree.get_level(2)),
                         {"child3", "child4"})
        self.assertEqual(set(t.obj for t in tree.get_level(5)), set())