from image_prediction.estimator.adapter.adapter import EstimatorAdapter


class DummyEstimator:
    @staticmethod
    def predict(_):
        return True


class EstimatorAdapterMock(EstimatorAdapter):

    def __init__(self, estimator):
        super().__init__(estimator=estimator)

    def predict(self, batch):
        return self.estimator.predict(batch)