def setUp(self):
self.guest1 = Guest("Stevie", 39, "The Chain", 10)
self.guest2 = Guest("Luke", 45, "The Rocky Road to Dublin", 150)
self.guest3 = Guest("Cathy", 31, "Wuthering Heights", 20)
# self.room1 = Room("Osaka", 5, 20)
self.room = Room("Tokyo", 3, 20)
def setUp(self):
self.room = Room("Room 1", 45, "Wild West")
self.guest_1 = Guest("Michael Jackson", 50)
self.guest_2 = Guest("50 Cent", 150)
self.song_1 = Song("Beautiful Day", "U2")
self.song_2 = Song("Bunsen", "Frontierer")
self.song_3 = Song("Axe To Fall", "Converge")
def setUp(self):
self.room_01 = Room("The Party Room", 4, 5.00)
self.room_02 = Room("The Date Room", 2, 10.00)
self.drink_01 = Drink("Gaffer's Home Brew", 5, 1.20)
self.drink_02 = Drink("Dry Martini", 2, 3.40)
self.drink_03 = Drink("Margarita", 2, 3.40)
self.drink_04 = Drink("Manhattan", 2, 3.40)
self.drink_05 = Drink("Sazerac", 3, 4.25)
self.drink_06 = Drink("Highball", 1, 2.60)
self.song_01 = Song("Come on Eileen", "Dexys Midnight Runners")
self.song_02 = Song("Mr Blue Sky", "ELO")
self.song_03 = Song("Bat out of Hell", "Meatloaf")
self.song_04 = Song("You're so Vain", "Carly Simon")
self.song_05 = Song("Tubthumping", "Chumbawamba")
self.song_06 = Song("Accidentally in Love", "Counting Crows")
self.guest_01 = Guest("Frodo Baggins", 51, 30.50, self.song_04,
self.drink_06)
self.guest_02 = Guest("Samwise Gamgee", 36, 10.75, self.song_03,
self.drink_01)
self.guest_03 = Guest("Meriadoc Brandybuck", 37, 23.60, self.song_01,
self.drink_05)
self.guest_04 = Guest("Peregrin Took", 29, 17.85, self.song_05,
self.drink_03)
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest = Guest("jack", 28, 50.00, "Dancing Queen")
self.guest2 = Guest("andy", 35, 75.00, "In the end")
self.drink1 = Bar("Vodka", 4.00)
self.drink2 = Bar("Beer", 3.00)
self.drink3 = Bar("Rum", 6.00)
def test_guest_has_name(self):
self.assertEqual("jack", self.guest.name)
def test_guest_has_age(self):
self.assertEqual(28, self.guest.age)
def test_guest_has_wallet(self):
self.assertEqual(50.00, self.guest.wallet)
def test_guest_has_fav_song(self):
self.assertEqual("Dancing Queen", self.guest.fav_song)
def test_guest_has_sufficient_funds_for_drink(self):
self.assertEqual(True, self.guest.sufficient_funds(self.drink1))
def test_guest_can_buy_drink(self):
self.guest.buy_drink(self.drink3)
self.assertEqual(44.00, self.guest.wallet)
def test_cannot_check_guest_in_if_cannot_afford_it(self):
tam = Guest("Tam", 2, self._song_1)
self._room.check_in_guest(tam)
self.assertEqual(0, self._room.number_of_guests())
self.assertEqual(0, self._room.get_till())
self.assertEqual(2, tam.get_cash())
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest = Guest("Pedram Valiani", 50)
self.room_1 = Room("Room 1", 40, "Outer Space")
self.favourite_song_1 = Song("Noctourniquet", "The Mars Volta")
# - This test determines that a song exists by comparing the object self.guest with attribute "name" to the value of "Pedram Valiani"
# - self.guest.name reads as self.guest -> the guest object from classes.guest and .name is the attribute of the song as defined in the Guest class
def test_guest_has_name(self):
self.assertEqual(self.guest.name, "Pedram Valiani")
# - Check that the guest has money in their wallet to pay a bill
def test_guest_has_wallet_money(self):
self.assertEqual(self.guest.wallet, 50)
# - Check that a guest can pay a bill - use the entry as an example
def test_customer_can_afford_entry(self):
self.guest.pay_the_bill(3)
self.assertEqual(47, self.guest.wallet)
# - Check whether a guest can afford to enter the room at the bar
def test_customer_can_afford_entry_to_room(self):
self.assertTrue(self.guest.customer_can_afford_entry(self.room_1))
class TestBar(unittest.TestCase):
def setUp(self):
self.song1 = Song("The Rolling Stones", "She's a rainbow")
self.song2 = Song("The Kinks", "All day and all of the night")
self.song3 = Song("The Beatles", "Norwegian Wood")
self.song4 = Song("The Troggs", "Wild Thing")
self.songs = [self.song1, self.song2, self.song3, self.song4]
self.drink = Drink("rum", 5.00, 30)
self.food = Food("peanuts", 2.00, 25)
self.guest1 = Guest("Holly Golighty", 20, 50, self.song1, 4)
self.guest2 = Guest("Paul Varjak", 27, 100, self.song2, 8)
self.guest3 = Guest("Cat", 100, 0, self.song4, 2)
self.guests = [self.guest1, self.guest2, self.guest3]
def test_guest_age(self):
self.assertEqual(20, self.guest1.age)
def test_guest_wallet(self):
self.assertEqual(50.00, self.guest1.wallet)
def test_sufficient_funds__true_if_enough(self):
self.assertEqual(True, self.guest1.sufficient_funds(self.drink))
def test_sufficient_funds__false_if_not_enough(self):
self.assertEqual(False, self.guest1.sufficient_funds(self.drink))
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest = Guest("Cam", 100.00, "My Favourite Song",
"My Favourite Artist")
self.song = Song("My Favourite Song", "My Favourite Artist")
self.room = Room(1, 10)
def test_guest_has_name(self):
self.assertEqual("Cam", self.guest.name)
def test_guest_has_wallet(self):
self.assertEqual(100.00, self.guest.wallet)
def test_guest_has_fav_song(self):
self.assertEqual("My Favourite Song", self.guest.fav_song)
def test_guest_has_fav_artist(self):
self.assertEqual("My Favourite Artist", self.guest.fav_artist)
def test_pay_entry(self):
self.guest.pay_entry(10.00)
self.assertEqual(90.00, self.guest.wallet)
def test_cheer_for_song(self):
self.room.add_song(self.song)
self.room.play_song()
self.assertEqual("This is my favourite song, yeehaw",
self.guest.cheer_for_favourite_song(self.song))
class TestGuest(unittest.TestCase):
def setUp(self):
self.song = Song("Don't Stop Believin'", "Journey")
self.guest = Guest("Tim", 32, 20.5, self.song)
def test_guest_has_name(self):
self.assertEqual("Tim", self.guest.name)
def test_guest_has_age(self):
self.assertEqual(32, self.guest.age)
def test_guest_has_wallet(self):
self.assertEqual(20.5, self.guest.wallet)
def test_guest_has_no_fav_song_if_not_provided(self):
guest_no_fav = Guest("Bob", 60, 100)
self.assertEqual(None, guest_no_fav.fav_song)
def test_guest_has_fav_song_if_provided(self):
self.assertEqual(self.song, self.guest.fav_song)
def test_guest_wallet_reduction(self):
self.guest.reduce_wallet(5)
self.assertEqual(15.5, self.guest.wallet)
def test_guest_cheer(self):
self.assertEqual("Whoo! They have my favourite song!",
self.guest.cheer())
def setUp(self):
self.song1 = Song("Help", "The Beatles")
self.song2 = Song("Saturday Night", "Sam Cooke")
self._songs = [self.song1, self.song2]
self.guest = Guest("Hannah")
def setUp(self):
self.guest_1 = Guest("Colin the Crooner", 34.23, "Isn't it Pythonic?")
self.room_1 = Room("Lizard Lounge", 3)
self.room_2 = Room("Viper's Nest", 10)
self.room_3 = Room("Clucking Hell", 15)
room_list = [self.room_1, self.room_2, self.room_3]
self.karaoke_bar = KaraokeBar("The Hit Re-Factory", room_list, 100.00,
10.00)
def setUp(self):
self.room = Room("Karoke Room 1", "Saviour", 0, 3, 2)
self.guest_1 = Guest("Baron Von Humperdink", 25, 40, "Saviour")
self.guest_2 = Guest("Baron Von Jon", 45, 40, "Faith Alone")
self.guest_3 = Guest("Baron Steve", 65, 40, "Blitzkrieg Bop")
self.song_1 = Song("Saviour", "Rise Against")
self.song_2 = Song("Blitzkrieg Bop", "Ramones")
self.song_3 = Song("Faith Alone", "Bad Religion")
def setUp(self):
self.song_1 = Song("Highway to Hell", "AC/DC")
self.song_2 = Song("The Clansman", "Iron Maiden")
self.song_3 = Song("Ace of Spades", "Motorhead")
self.songs = [self.song_1, self.song_2, self.song_3]
song = Song("Ace of Spades", "Motorhead")
self.guest = Guest("Jack", 20, song)
def setUp(self):
self.room = Room("Room 1", 3, 5.00, 100.00)
self.song_1 = Song("Africa", "Toto")
self.song_2 = Song("Purple Rain", "Prince")
self.song_3 = Song("Dancing Queen", "Abba")
self.guest_1 = Guest("Elton", 30.00, "Purple Rain")
self.guest_2 = Guest("Dolly", 40.00, "Wannabe")
self.guest_3 = Guest("Kylie", 50.00, "Africa")
self.guest_4 = Guest("Bruce", 50.00, "Enter Sandman")
def setUp(self):
self.room1 = Room("Room 1", 2, 5)
self.room2 = Room("Room 2", 3, 7.5)
self.guest1 = Guest("Sam", 5)
self.guest2 = Guest("Bob", 10)
self.guest3 = Guest("Joe", 15)
self.song1 = Song("People Like Us")
self.song2 = Song("Waterloo")
self.song3 = Song("Ice Cream")
def setUp(self):
# self.room1 = Room("Osaka", 5, 20)
self.room = Room("Tokyo", 3, 20)
self.song1 = Song("Fleetwood Mac", "The Chain")
self.song2 = Song("Dubliners", "The Rocky Road to Dublin")
self.song3 = Song("Kate Bush", "Wuthering Heights")
self.guest1 = Guest("Stevie", 39, "The Chain", 10)
self.guest2 = Guest("Luke", 45, "The Rocky Road to Dublin", 150)
self.guest3 = Guest("Cathy", 31, "Wuthering Heights", 200)
def test_can_add_guest_to_room_room_full(self):
guest_1 = Guest("Rocky", 100, "Eye of the tiger")
guest_2 = Guest("Apollo", 100, "Living in America")
guest_3 = Guest("Adrian", 100, "No easy way out")
guest_4 = Guest("Paulie", 100, "Heart's on fire")
guest_5 = Guest("Ivan", 100, "Burning heart")
self.room.guests.append(guest_1)
self.room.guests.append(guest_2)
self.room.guests.append(guest_3)
self.room.guests.append(guest_4)
self.assertEqual("Room is full", self.room.add_guest_to_room(guest_5))
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest_1 = Guest("Elton", 30.00, "Purple Rain")
self.room = Room("Room 1", 3, 5.00, 100.00)
def test_guest_has_name(self):
self.assertEqual("Elton", self.guest_1.name)
def test_guest_can_pay_entry_fee(self):
self.guest_1.pay_entry_fee(self.room.entry_fee)
self.assertEqual(25.00, self.guest_1.wallet)
class TestGuest(unittest.TestCase):
def setUp(self):
song_1 = Song("Run-D.M.C.", "Christmas in Hollis")
song_2 = Song("Skeletons", "Stevie Wonder")
song_3 = Song("Beethoven", "Ode to Joy")
song_4 = Song("Let It Snow! Let It Snow! Let It Snow!", "Vaughn Monroe")
self.guest_1 = Guest("John McLean", 39, song_1, 100.00, "Yippee-ky-ay!!")
self.guest_2 = Guest(
"Al Powell",
43,
song_2,
50.00,
"Now means now goddammit!, I'm under automatic rifle fire at Nakatomi, I need backup assistance now! NOW, GODDAMMIT, NOW!",
)
self.guest_3 = Guest(
"Hans Gruber", 47, song_3, 10000.00, "WHERE ARE MY DETONATORS!"
)
self.guest_4 = Guest(
"Karl Vreski", 35, song_4, 0.00, "No one kills him but me!"
)
def test_guest_name(self):
self.assertEqual("John McLean", self.guest_1.name)
def test_guest_age(self):
self.assertEqual(39, self.guest_1.age)
def test_guest_favourite_song(self):
self.assertEqual("Run-D.M.C.", self.guest_1.favourite_song.artist)
self.assertEqual("Christmas in Hollis", self.guest_1.favourite_song.title)
def test_guest_cheer(self):
self.assertEqual("No one kills him but me!", self.guest_4.cheer)
def test_guest_wallet(self):
self.assertEqual(100, self.guest_1.wallet)
def test_check_wallet_balance__exact_amount(self):
balance_check = self.guest_1.check_wallet_balance(100)
self.assertEqual(True, balance_check)
def test_check_wallet_balance__under_balance(self):
balance_check = self.guest_1.check_wallet_balance(99)
self.assertEqual(True, balance_check)
def test_check_wallet_balance__over_balance(self):
balance_check = self.guest_1.check_wallet_balance(101)
self.assertEqual(False, balance_check)
def test_remove_from_wallet(self):
self.guest_2.remove_from_wallet(30)
self.assertEqual(20, self.guest_2.wallet)
def setUp(self):
self.room_1 = Room("Karaoke Star", 50)
self.room_2 = Room("Karaoke Rooms", 20)
self.room_3 = Room("Scotland's got talent", 10)
self.guest_1 = Guest("Arnold")
self.guest_2 = Guest("John")
self.guest_3 = Guest("Tamara")
self.song_1 = Song("Hotel California", "Eagles", "rock")
self.song_2 = Song("Viva Las Vegas", "Elvis Presley", "Rock and roll")
self.song_3 = Song("Billie Jean", "Michael Jackson", "Pop")
def setUp(self):
self.song1 = Song("Help", "The Beatles")
self.song2 = Song("Saturday Night", "Sam Cooke")
self.songs = [self.song1, self.song2]
self.guest1 = Guest("Hannah")
self.guest2 = Guest("Louise")
self.guest3 = Guest("Harry")
self.guests = [self.guest1, self.guest2, self.guest3]
self.room = Room("Motown", 3)
def setUp(self):
self.room = Room("Cover Your Ears!", 15, 10)
self.room_small = Room("Tight Squeeze", 1, 15)
self.song = Song("Hold a chicken in the air")
self.song_not_fav = Song("Put that chair back")
# fav_song should be song object, not just a string
self.guest = Guest("Sid The Sloth", 33.25, "Hold a chicken in the air")
# like...
# self.guest = Guest("Sid The Sloth", 33.25, song)
# fav_song should be song object, not just a string
self.guest_no_money = Guest("Scrooge", 3.99, "Yeah, right!")
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest = Guest('TJ', 100.00)
self.room = Room('Smooth Room', 50.00, 3, 5.00)
def test_check_guest_name(self):
self.assertEqual('TJ', self.guest.name)
def test_check_guest_cash(self):
self.assertEqual(100.00, self.guest.cash)
def test_pay_entry_fee(self):
self.guest.pay_entry_fee(self.room)
self.assertEqual(95.00, self.guest.cash)
def setUp(self):
self.song = Song("Clint Eastwood", "Gorrilaz", "Electropop")
self.song2 = Song("All Eyez On Me", "Tupac", "Rap")
self.song3 = Song("Timeless", "Il Divo", "Classical")
self.song4 = Song("Hurt", "Johnny Cash", "Country")
self.song5 = Song("Mustang Sally", "Wilson Pinkett", "Blues")
self.song6 = Song("Think Twice", "Celine Dion", "Soft Rock")
self.song7 = Song("Bye Bye", "Mariah Carey", "R&B")
self.song8 = Song("Perfect", "Ed Sheeran", "Pop")
self.room = Room("300", 30.00, 0, [self.song, self.song2])
self.guest = Guest("Frank Samson", 50.00, 30, self.song3)
self.guest2 = Guest("Johnny Bravo", 100.00, 55, self.song4)
self.guest3 = Guest("Lucy Lou", 85.00, 40, self.song5)
self.guest4 = Guest("Paul Rand", 20.00, 17, self.song6)
def setUp(self):
self.room = Room("The Viper Room", 150, 5, 20)
self.song1 = Song("The Rolling Stones", "She's a rainbow")
self.song2 = Song("The Kinks", "All day and all of the night")
self.song3 = Song("The Beatles", "Norwegian Wood")
self.song4 = Song("The Troggs", "Wild Thing")
self.songs = [self.song1, self.song2, self.song3, self.song4]
self.guest1 = Guest("Holly Golighty", 20, 50, self.song1, 4)
self.guest2 = Guest("Paul Varjak", 27, 100, self.song2, 8)
self.guest3 = Guest("Cat", 100, 0, self.song4, 2)
self.guests = [self.guest1, self.guest2, self.guest3]
def setUp(self):
self.song_1 = Song("Highway to Hell", "AC/DC")
self.song_2 = Song("The Clansman", "Iron Maiden")
self.song_3 = Song("Ace of Spades", "Motorhead")
self.songs = [self.song_1, self.song_2, self.song_3]
self.jack = Guest("Jack", 20, self.song_1)
self.victor = Guest("Victor", 15, self.song_2)
self.isa = Guest("Isa", 100, self.song_3)
self.guests = [self.jack, self.victor, self.isa]
self.winston = Guest("Winston", 10, self.song_2)
self.room = Room("The Metal Room", 3, 10)
def setUp(self):
self.room1 = Room("Boiler room", 20, 5)
self.room2 = Room("Grinder room", 35, 3)
self.guest = Guest("Jack", 28, 50.00, "In the end")
self.guest1 = Guest("Andy", 35, 75.00, "In the end")
self.guest2 = Guest("Alex", 25, 30.00, "Unwritten")
self.guest3 = Guest("Collin", 30, 45.00, "Dancing in the moonlight")
self.guests1 = [self.guest, self.guest1, self.guest2]
self.guests2 = [self.guest, self.guest1, self.guest2, self.guest3]
self.guests3 = [self.guest1, self.guest2, self.guest3]
self.song = Song("Queen", "Bohemian Rhapsody")
self.song1 = Song("Linken park", "In the end")
self.song2 = Song("Natasha Bedingfield", "Unwritten")
self.song4 = Song("Top Loader", "Dancing in the moonlight")
self.playlist = [self.song, self.song1, self.song2]
def setUp(self):
self.song1 = Song("The Rolling Stones", "She's a rainbow")
self.song2 = Song("The Kinks", "All day and all of the night")
self.song3 = Song("The Beatles", "Norwegian Wood")
self.song4 = Song("The Troggs", "Wild Thing")
self.songs = [self.song1, self.song2, self.song3, self.song4]
self.drink = Drink("rum", 5.00, 30)
self.food = Food("peanuts", 2.00, 25)
self.guest1 = Guest("Holly Golighty", 20, 50, self.song1, 4)
self.guest2 = Guest("Paul Varjak", 27, 100, self.song2, 8)
self.guest3 = Guest("Cat", 100, 0, self.song4, 2)
self.guests = [self.guest1, self.guest2, self.guest3]
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest_1 = Guest("Colin the Crooner", 34.23, "Isn't it Pythonic?")
self.room_1 = Room("Lizard Lounge", 3)
self.room_2 = Room("Viper's Nest", 10)
self.room_3 = Room("Clucking Hell", 15)
room_list = [self.room_1, self.room_2, self.room_3]
self.karaoke_bar = KaraokeBar("The Hit Re-Factory", room_list, 100.00,
10.00)
def test_can_find_guest_by_name(self):
self.assertEqual("Colin the Crooner", self.guest_1.name)
def test_entry_fee_deducted_from_guest_wallet(self):
self.guest_1.pay_entry_fee(self.karaoke_bar.entry_fee)
self.assertEqual(24.23, round(self.guest_1.wallet, 2))
def setUp(self):
drink_1 = Drink("Guinness", 3, 3)
drink_2 = Drink("Whiskey Sour", 6, 5)
drink_3 = Drink("Tsingtao", 4, 3)
drinks_list = [drink_1, drink_2, drink_3]
self.bar_1 = Bar(500, drinks_list)
self.song_1 = Song(10001, "Let's Make Love and Listen to Death from Above", "CSS", [3,30], 2005)
self.song_4 = Song(10004, "Club Tropicana", "Wham!", [4, 28], 1983)
self.guest_1 = Guest("Lovefoxxx", 36, 50, self.song_1)
self.guest_2 = Guest("Britney Spears", 38, 100, self.song_4)
self.room_1 = Room("Plush Paradise", 3)
def create_guests(mode = None, n = None, p_eta = None, p_order = None): if n != None and p_eta != None and p_order != None: d = dict.fromkeys(string.ascii_lowercase, 0) etas = [np.random.poisson(p_eta) for i in range(n)] names = [i for i in d] orders = [np.random.poisson(p_order) for i in range(n)] # Pre-defined etas if mode == 'case_1': etas = [6, 2, 5, 1, 4, 7] names = ['a','b','c','d','e','f'] orders = [1, 10, 4, 4, 2, 3] n = 6 arrivals = [] for i in range(n): new_guest = Guest() # new_guest.eta = np.random.random_integers(1,3)*5 new_guest.arrive = datetime.timedelta(minutes=etas[i]) new_guest.name = names[i] new_guest.id = 'gst'+str(i) new_guest.orders = orders[i] new_guest.prep = datetime.timedelta(minutes=orders[i]*1) arrivals.append(new_guest) if arrivals[0].arrive.seconds == 0: arrivals[0].arrive = datetime.timedelta(minutes=1) return arrivals
def copy_guests(A, B): for i in A: new_guest = Guest() new_guest.arrive = i.arrive new_guest.name = i.name new_guest.id = i.id new_guest.orders = i.orders new_guest.prep = i.prep B.append(new_guest)
