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)