class CustomerTest(unittest.TestCase):
def setUp(self):
self._frame = CarbonFrame(10, 100)
self._frontWheel = RoadWheel("Front RoadWheel", 10, 100)
self._backWheel1 = MountainWheel("Back MountainWheel", 10, 100)
self._backWheel2 = RoadWheel("Back RoadWheel", 10, 100)
self._bike = BicycleModel("Good Build", self._frontWheel,
self._backWheel2, self._frame)
self._customer = Customer('Customer', 10000)
def test_buy_bike(self):
self._customer.buy_bike(self._bike)
self.assertNotEqual(None, self._customer.get_bike())
self.assertEqual(self._customer.get_funds(), 9970)
class CustomerTest(unittest.TestCase):
def setUp(self):
self._frame = CarbonFrame(10, 100)
self._frontWheel = RoadWheel("Front RoadWheel", 10, 100)
self._backWheel1 = MountainWheel("Back MountainWheel", 10, 100)
self._backWheel2 = RoadWheel("Back RoadWheel", 10, 100)
self._bike = BicycleModel(
"Good Build", self._frontWheel, self._backWheel2, self._frame
)
self._customer = Customer('Customer', 10000)
def test_buy_bike(self):
self._customer.buy_bike(self._bike)
self.assertNotEqual(None, self._customer.get_bike())
self.assertEqual(self._customer.get_funds(), 9970)
