def mock_layer_and_mock(self):
from mink.layers import Layer
def fit_transform(self, X, y, **kwargs):
return mock.fit(X, y, **kwargs).transform(X, **kwargs)
def transform(self, X, **kwargs):
return X
mock = Mock(spec=Layer)
mock.fit = Mock(return_value=mock)
mock.transform = Mock(side_effect=lambda X, **kwargs: X)
mock.fit_transform = fit_transform
mock.incoming = None
class MockLayer(Layer):
fit = mock.fit
transform = mock.transform
fit_transform = mock.fit_transform
def __init__(self, incoming=None):
self.incoming = incoming
def initialize(self, Xs, ys, **kwargs):
self.fit(Xs, ys, **kwargs)
return self
def get_output(self, Xs, **kwargs):
return Xs
return MockLayer(), mock
