Beispiel #1
0
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")
Beispiel #2
0
 def test_display_correct_stocks(self):
     shop1 = BikeRental(30)
     shop2 = BikeRental(15)
     self.assertEqual(shop1.displayStock(), 30, 'should be 30')
     self.assertEqual(shop2.displayStock(), 15, 'should be 15')
Beispiel #3
0
 def test_bike_rental_display_correct_stock(self):
     shop1 = BikeRental()
     shop2 = BikeRental(10)
     self.assertEqual(shop1.displayStock(), 0)
     self.assertEqual(shop2.displayStock(), 10)