def test_classmethod(self, mock_get_roll_call): mock_get_roll_call.return_value = [ Car('Ford'), ] self.assertEquals(get_roll_call(), [ Car('Ford'), ])
def test_context_manager(self, mock_open_car): car = Car() car.closed = 'Foo' def enter_car(car): car.closed = 'Bar' return car mock_open_car.return_value.__enter__ = enter_car #mock_open_car.return_value.__exit__ = exit_car states = close_car(car) self.assertEquals(states, ['Bar', 'Bar'])
def get_car_wheels(): return Car().wheels
def get_car_make(make=None): car = Car() if make: car = Car.for_make(make) return car.get_make()