class TestLeaf:
    """Test the Leaf class"""
    def setup_method(self, method):
        self.leaf = Leaf()

    def teardown_method(self, method):
        self.leaf = None

    def test_name(self):
        assert self.leaf.get_name() == "Leaf"
class TestLeaf:
    """Test the Leaf class"""

    def setup_method(self, method):
        self.leaf = Leaf()

    def teardown_method(self, method):
        self.leaf = None

    def test_name(self):
        assert self.leaf.get_name() == "Leaf"
 def setup_method(self, method):
     self.leaf = Leaf()
 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.leaf = Leaf()