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