class TestTrunk: """Test the Trunk class""" def setup_method(self, method): self.trunk = Trunk() def teardown_method(self, method): self.trunk = None def test_name(self): assert self.trunk.get_name() == "Trunk"
def setup_method(self, method): self.trunk = Trunk()
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