Exemplo n.º 1
0
 def testAllLeavesWithTrees(self, tree):
   leaves = tree_util.tree_leaves(tree)
   self.assertTrue(tree_util.all_leaves(leaves))
   self.assertFalse(tree_util.all_leaves([tree]))
Exemplo n.º 2
0
 def testAllLeavesWithLeaves(self, leaf):
   self.assertTrue(tree_util.all_leaves([leaf]))
Exemplo n.º 3
0
 def tree_unflatten(cls, meta, data):
   if not tree_util.all_leaves(data):
     data, meta = tree_util.tree_flatten(tree_util.tree_unflatten(meta, data))
   return FlatCache(None, leaves=data, treedef=meta)