def test_shapes(params, shape_desired): """Test that the shape of the output is the expected one.""" transformer = MultivariateTransformer(**params) assert transformer.fit(X).transform(X).shape == shape_desired assert transformer.fit_transform(X).shape == shape_desired
def test_ndim(params, ndim_desired): """Test that the ndim of the output is the expected one.""" transformer = MultivariateTransformer(**params) assert transformer.fit(X).transform(X).ndim == ndim_desired assert transformer.fit_transform(X).ndim == ndim_desired