def create_crewman():
crewman_name = input("Name your crew member")
crewman = Staff(crewman_name)
print(repr(crewman))
available_stations = list()
for station in stations_list:
station.retrieve_bonuses()
print(station)
pop_cap = station.section_cap
current_pop = len(station.staff_list)
if current_pop < pop_cap:
available_stations.append(station)
print("available stations:")
options_dict: Dict = dict()
for counter, station in enumerate(available_stations):
options_dict[counter] = station
print("{}) {}".format(counter, station.station_name))
chosen_option: int = int(input("Input the station number"))
chosen_station = options_dict[chosen_option]
chosen_station.staff_list.append(crewman)
def test_room_remove_person(self):
room = Office('Iroko')
person = Staff('Anthiny', 'Nandaa')
room.allocate(person)
self.assertNotEqual(room.people, [])
room.remove_person(person)
self.assertEqual(room.people, [])
self.assertEqual(room.remove_person(person), False)
def setUp(self):
##### Staff #####
self.staff1 = Staff("Bob", 1, 21, 4, Job.CHEF, [FoodAbility.CHINESE])
self.staff2 = Staff("Barbera", 10, 92, 4, Job.DOORMAN,
[FoodAbility.INDIAN])
self.staff3 = Staff("George", 100, 12, 4, Job.CHEF, [FoodAbility.NONE])
##### Drinks #####
self.wine = Drink("Wine", 899, 67)
self.cola = Drink("Cola", 199, 0)
self.vodka = Drink("Vodka", 299, 0)
##### Food #####
self.food = Food("Soup", 399, 2, None)
self.food_1 = Food("Curry", 399, 2, FoodAbility.INDIAN)
self.food_not = Food("Chinese", 899, 2, FoodAbility.CHINESE)
##### Songs #####
self.song = [
Song("Cafo", "Animals as Leaders", 2009, Genre.ALTERNATIVE_METAL),
Song("Electric Sunrise", "Plini", 2016, Genre.PROG_METAL),
Song("Pastures", "Plini", 2016, Genre.PROG_METAL)
]
##### Duke Box #####
self.dukebox = DukeBox(self.song, 99, [])
##### Bar #####
self.bar = Bar(self.dukebox, [self.staff1, self.staff2, self.staff3],
[self.cola, self.wine], [self.food, self.food_1])
self.bar.drinks_list[0].add_stock(60)
self.bar.drinks_list[1].add_stock(90)
self.bar.food_list[0].add_stock(100)
self.bar.food_list[1].add_stock(70)
def setUp(self):
##### Staff #####
self.staff = [
Staff("Bob", 1, 21, 4, Job.CHEF, [FoodAbility.CHINESE]),
Staff("Barbera", 10, 92, 4, Job.DOORMAN, [FoodAbility.INDIAN]),
Staff("George", 100, 12, 4, Job.CHEF, [FoodAbility.NONE]),
Staff("Margery", 100, 30, 4, Job.BARMAN, [FoodAbility.NONE])
]
##### Drinks #####
self.wine = Drink("Wine", 899, 67)
self.cola = Drink("Cola", 199, 0)
self.vodka = Drink("Vodka", 299, 0)
##### Food #####
self.food = Food("Soup", 399, 2, None)
self.food_1 = Food("Curry", 399, 2, FoodAbility.INDIAN)
self.food_not = Food("Chinese", 899, 2, FoodAbility.CHINESE)
##### Songs #####
self.song = [
Song("Cafo", "Animals as Leaders", 2009, Genre.ALTERNATIVE_METAL),
Song("Electric Sunrise", "Plini", 2016, Genre.PROG_METAL),
Song("Pastures", "Plini", 2016, Genre.PROG_METAL)
]
##### Duke Box #####
self.dukebox = DukeBox(self.song, 99, [])
##### Bar #####
self.bar = Bar(self.dukebox, [self.staff[0], self.staff[3]],
[self.cola, self.wine], [self.food, self.food_1])
self.bar.drinks_list[0].add_stock(60)
self.bar.drinks_list[1].add_stock(90)
self.bar.food_list[0].add_stock(100)
self.bar.food_list[1].add_stock(70)
##### Bathroom #####
self.bath = Bath("Bath Extra", 7, 100, 10)
self.toilet = Toilet("ToLet Amaze", 7, 100, 10)
self.sink = Sink("sink Extra", 7, 100, 10)
self.bathroom = Bathroom(self.sink, self.bath, self.toilet, None)
##### Rooms #####
self.rooms = [
Room([self.bathroom], self.dukebox, RoomType.SINGLE),
Room([self.bathroom], self.dukebox, RoomType.DOUBLE),
Room([self.bathroom], self.dukebox, RoomType.TRIPLE),
Room([self.bathroom], self.dukebox, RoomType.DELUXE)
]
##### Customers #####
self.customers = [
Customer("Jeff", 24, 80, 70, 60, 10, 0, 90, None, 100),
Customer("Margaret", 15, 83, 70, 99, 10, 0, 90, None, 200),
Customer("Dorothy", 47, 83, 70, 99, 10, 0, 90, None, 200),
Customer("Humphrey", 15, 83, 70, 14, 10, 0, 90, None, 200)
]
self.customers[1].set_partner(self.customers[0])
##### Building #####
self.building = Building(self.rooms, self.bar, self.customers,
self.staff)
def setUp(self):
self.staff_member = Staff("staff_name", 2, 3, ["bob", "george"],
"Nothing")
def test_room_allocate_fails_when_person_type_is_staff_and_room_type_is_L(
self):
room = LivingSpace('Iroko')
person = Staff('Anthony', 'Nandaa')
self.assertEqual(room.allocate(person), False)
self.assertEqual(room.people, [])
def test_room_allocate_able(self):
room = Office('Iroko')
person = Staff('Anthiny', 'Nandaa')
room.allocate(person)
self.assertEqual(room.allocate_able(person), False)
def test_room_people_list_with_room_name(self):
room = Office('Iroko')
person = Staff('Anthiny', 'Nandaa')
room.allocate(person)
data = room.people_list_with_room_name(False)
self.assertNotEqual(data, [])
def test_room_allocate_fails_when_is_filled(self):
room = Office('Iroko')
person = Staff('Anthiny', 'Nandaa')
room.capacity = 0
self.assertEqual(room.allocate(person), False)
def test_room_allocate_staff(self):
room = Office('Iroko')
person = Staff('Anthony', 'Nandaa')
room.allocate(person)
self.assertNotEqual(room.people, [])
self.assertEqual(type(person), type(room.people[0]))