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())