|
12 | 12 | make_example_3d_numpy_list,
|
13 | 13 | )
|
14 | 14 | 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 |
19 | 16 |
|
20 | 17 |
|
21 | 18 | @pytest.mark.parametrize(
|
@@ -48,39 +45,6 @@ def test_collection_transformer_invalid_input(dtype):
|
48 | 45 | t.fit_transform(y)
|
49 | 46 |
|
50 | 47 |
|
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 |
| - |
84 | 48 | def test_inverse_transform():
|
85 | 49 | """Test inverse transform."""
|
86 | 50 | d = _Dummy()
|
|
0 commit comments