def main():
shop = BikeRental(100)
customer = Customer()
while True:
print("""
====== Bike Rental Shop =======
1. Display available bikes
2. Request a bike on hourly basis $5
3. Request a bike on daily basis $20
4. Request a bike on weekly basis $60
5. Return a bike
6. Exit
""")
choice = input("Enter your choice ? ")
try:
choice = int(choice)
except ValueError:
print("That's not an integer value you entered !")
continue
if choice == 1:
shop.displayStock()
elif choice == 2:
customer.rentalTime = shop.rentBikesOnHourlyBasis(
customer.requestBike())
customer.rentalBasics = 1
elif choice == 3:
customer.rentalBasics = 2
customer.rentalTime = shop.rentBikesOnDailyBasis(
customer.requestBike())
elif choice == 4:
customer.rentalBasics = 3
customer.rentalTime = shop.rentBikesOnWeeklyBasis(
customer.requestBike())
elif choice == 5:
customer.bill = shop.returnBikes(customer.returnBike())
customer.rentalBasics, customer.rentalTime, customer.bikes = 0, 0, 0
elif choice == 6:
break
else:
print("INVALID Choice")
def test_rentBikeOnHourlyBasis_for_negative_number_of_bikes(self):
shop = BikeRental(10)
self.assertEqual(shop.rentBikesOnHourlyBasis(-1), None)