def setUp(self): damaged_bike = Bike() damaged_bike.damage() self.bike_container = BikeContainer(20) self.other_container = BikeContainer(20) self.bike = Bike() self.damaged_bike = damaged_bike
class BikeTestCase(unittest.TestCase): def setUp(self): self.bike = Bike() def test_default_bike_broken_state(self): self.assertEqual(self.bike.is_damaged, False) def test_can_be_damaged(self): self.bike.damage() self.assertEqual(self.bike.is_damaged, True) def test_can_be_fixed(self): self.bike.damage() self.bike.fix() self.assertEqual(self.bike.is_damaged, False)
def add_bikes_to(holder, count, damaged_bikes = True): for num in range(0, count): bike = Bike() if damaged_bikes == False: bike.damage() holder.dock(bike)