class TestGuest(unittest.TestCase):
def setUp(self):
self.guest = Guest("Singy McSingface", 36, 50,
"Never Gonna Give You Up", "Beer")
def test_guest_has_name(self):
self.assertEqual("Singy McSingface", self.guest.name)
def test_guest_has_age(self):
self.assertEqual(36, self.guest.age)
def test_guest_has_cash(self):
self.assertEqual(50, self.guest.cash)
def test_remove_cash(self):
self.guest.remove_cash(5)
self.assertEqual(45, self.guest.cash)
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest1 = Guest("John", 10.00, "Wannabe")
def test_guest_has_name(self):
self.assertEqual("John", self.guest1.name)
def test_guest_has_wallet(self):
self.assertEqual(10.00, self.guest1.wallet)
def test_guest_remove_cash(self):
self.guest1.remove_cash(5.00)
self.assertEqual(5.00, self.guest1.wallet)
def test_guest_has_favourite_song(self):
self.assertEqual("Wannabe", self.guest1.favourite_song)
def test_guest_can_cheer(self):
self.assertEqual("Woo!", self.guest1.cheer())