def setUp(self): self.lst = [ TopNode("top"), [ Node("a"), Node("b"), Node("c"), [ Node("ca"), Node("cb") ], Node("d") ] ] self.tt = tinytree.constructFromList(self.lst)[0]
def test_foo(self): pages = [ Node("one"), [ Node("two"), Node("three"), [ Node("four"), Node("five"), ] ], Node("six"), ] one = pages[0] six = pages[2] heads = tinytree.constructFromList(pages) assert one.count() == 5 assert six.count() == 1 assert len(heads) == 2