Skip to content

Commit d5fdf13

Browse files
committed
remove CollectionToSeriesWrapper
1 parent cc76694 commit d5fdf13

File tree

1 file changed

+1
-37
lines changed

1 file changed

+1
-37
lines changed

aeon/transformations/collection/tests/test_base.py

+1-37
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,7 @@
1212
make_example_3d_numpy_list,
1313
)
1414
from aeon.testing.data_generation._legacy import make_series
15-
from aeon.transformations.collection import (
16-
BaseCollectionTransformer,
17-
CollectionToSeriesWrapper,
18-
)
15+
from aeon.transformations.collection import BaseCollectionTransformer
1916

2017

2118
@pytest.mark.parametrize(
@@ -48,39 +45,6 @@ def test_collection_transformer_invalid_input(dtype):
4845
t.fit_transform(y)
4946

5047

51-
@pytest.mark.parametrize("dtype", ["pd.Series", "pd.DataFrame", "np.ndarray"])
52-
def test_collection_transformer_wrapper_series(dtype):
53-
"""Test that the wrapper for regular transformers works with series input."""
54-
if dtype == "np.ndarray":
55-
y = make_series(return_numpy=True)
56-
else:
57-
y = make_series()
58-
if dtype == "pd.DataFrame":
59-
y = y.to_frame(name="series1")
60-
wrap = CollectionToSeriesWrapper(transformer=_Dummy())
61-
yt = wrap.fit_transform(y)
62-
63-
assert isinstance(yt, list)
64-
assert isinstance(yt[0], np.ndarray)
65-
assert len(yt) == 1
66-
assert yt[0].ndim == 2
67-
68-
69-
@pytest.mark.parametrize(
70-
"data_gen", [make_example_3d_numpy, make_example_3d_numpy_list]
71-
)
72-
def test_collection_transformer_wrapper_collection(data_gen):
73-
"""Test that the wrapper for regular transformers works with collection input."""
74-
X, y = data_gen()
75-
76-
wrap = CollectionToSeriesWrapper(transformer=_Dummy())
77-
Xt = wrap.fit_transform(X, y)
78-
79-
assert isinstance(Xt, list)
80-
assert isinstance(Xt[0], np.ndarray)
81-
assert Xt[0].ndim == 2
82-
83-
8448
def test_inverse_transform():
8549
"""Test inverse transform."""
8650
d = _Dummy()

0 commit comments

Comments
 (0)