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())
class TestGuest(unittest.TestCase): def setUp(self): self.song = Song("Castle Theme") self.guest = Guest("Mario", self.song, 100.00) self.room = Room("Nintendo") def test_guest_has_name(self): self.assertEqual("Mario", self.guest.name) def test_guest_has_favourite_song(self): self.assertEqual(self.song, self.guest.favourite_song) def test_guest_has_wallet(self): self.assertEqual(100.00, self.guest.wallet) def test_pay_entry_fee_removes_money_from_wallet(self): self.guest.pay_entry_fee(self.room) self.assertEqual(90.00, self.guest.wallet) def test_guest_can_cheer(self): self.assertEqual("Woop!", self.guest.cheer())
class TestGuest(unittest.TestCase): def setUp(self): self.room_2 = Room("Studio 54", 45.00) self.guest_1 = Guest("Mick", "Jagger", 20, 12000.00, "Dead Flowers") self.guest_2 = Guest("Justin", "Bieber", 16, 5.00, "Sorry") self.guest_3 = Guest("Joni", "Mitchell", 21, 100.00, "Juice") def test_guest_has_whole_name(self): self.assertEqual("Justin", self.guest_2.get_guest_first_name()) self.assertEqual("Bieber", self.guest_2.get_guest_surname()) def test_guest_has_age(self): self.assertEqual(20, self.guest_1.get_guest_age()) def test_guest_has_funds(self): self.assertEqual(100.00, self.guest_3.wallet_cash_count()) def test_remove_cash_from_wallet(self): self.guest_1.remove_cash_from_wallet(self.room_2.entry_fee) self.assertEqual(11955.00, self.guest_1.wallet_cash_count()) def test_cheer(self): self.cheer() self.assertEqual("CHOON!", self.guest_3.cheer())