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
Test that the shape of the output is the expected one.
pyts/multivariate/transformation/tests/test_multivariate.py