def test_size(self):
     self.scene.add_object(Leaf())
     self.scene.add_object(Trunk())
     self.scene.add_object(Tree(2))
     assert self.scene.get_size() == 3
 def setup_method(self, method):
     self.nbleaf = 20
     self.tree = Tree(self.nbleaf)