class generate_bikes(unittest.TestCase):
def setUp(self):
self._manufacturer1 = BicycleManufacturer("Manufacturer1", 0.2)
self._manufacturer2 = BicycleManufacturer("Manufacturer2", 0.3)
self._shop = BicycleShop("Shop", 0.2,
[self._manufacturer1, self._manufacturer2])
self._customer = Customer('Customer', 10000)
def get_bikes_from_manufacturer(self):
self.assertEqual(6, len(self._shop.get_inventory()))
def test_sell_bike(self):
for i in range(6):
self._shop.sell_bike(self._customer, self._shop.get_inventory()[0])
self.assertEqual(6 - (i + 1), len(self._shop.get_inventory()))
class generate_bikes(unittest.TestCase):
def setUp(self):
self._manufacturer1 = BicycleManufacturer("Manufacturer1", 0.2)
self._manufacturer2 = BicycleManufacturer("Manufacturer2", 0.3)
self._shop = BicycleShop(
"Shop", 0.2, [self._manufacturer1, self._manufacturer2]
)
self._customer = Customer('Customer', 10000)
def get_bikes_from_manufacturer(self):
self.assertEqual(6, len(self._shop.get_inventory()))
def test_sell_bike(self):
for i in range(6):
self._shop.sell_bike(self._customer, self._shop.get_inventory()[0])
self.assertEqual(6-(i+1), len(self._shop.get_inventory()))
customer2 = Customer("Customer 2", 500)
customer3 = Customer("Customer 3", 1000)
customerList = [customer1, customer2, customer3]
print "\n"
print "====" * 5
print "Welcome to {0}".format(shop.get_name())
print "====" * 5
print "\nCurrent Inventory:"
print "-----" * 5
for bike in shop.get_inventory():
print "{0}".format(bike)
for customer in customerList:
print "\n{0}".format(customer)
temp = []
for bike in shop.get_inventory():
if (bike.get_cost() <= customer.get_funds()):
temp.append(bike)
print "You can buy the following bikes:"
for i in range(len(temp)):
print "{0} - {1}".format((i + 1), temp[i])
choice = int(raw_input("Which bike #? ")) - 1
shop.sell_bike(customer, temp[choice])
print "Bought {0} ${1} left".format(customer.get_bike(),
customer.get_funds())
print "\nShop's profit: ${0} !!!!!!".format(shop.get_profit())
customer3 = Customer("Customer 3", 1000)
customerList = [customer1, customer2, customer3]
print "\n"
print "====" * 5
print "Welcome to {0}".format(shop.get_name())
print "====" * 5
print "\nCurrent Inventory:"
print "-----" * 5
for bike in shop.get_inventory():
print "{0}".format(bike)
for customer in customerList:
print "\n{0}".format(customer)
temp = []
for bike in shop.get_inventory():
if(bike.get_cost() <= customer.get_funds()):
temp.append(bike)
print "You can buy the following bikes:"
for i in range(len(temp)):
print "{0} - {1}".format((i+1), temp[i])
choice = int(raw_input("Which bike #? ")) - 1
shop.sell_bike(customer, temp[choice])
print "Bought {0} ${1} left".format(
customer.get_bike(), customer.get_funds()
)
print "\nShop's profit: ${0} !!!!!!".format(shop.get_profit())
