@@ -142,18 +142,29 @@ def test_that_correlated_and_independent_observations_are_grouped_separately(
142
142
143
143
144
144
@pytest .mark .parametrize ("nr_observations" , [0 , 1 , 2 ])
145
- def test_misfit_preprocessor_single (nr_observations ):
146
- """We create nr_observations responses using the linear function y = ax.
147
- We don`t have enough information to cluster properly, so we don`t try
148
- to scale"""
149
- rng = np .random .default_rng (1234 )
145
+ def test_that_perfectly_correlated_responses_are_not_scaled (nr_observations ):
150
146
nr_realizations = 1000
151
147
Y = np .ones ((nr_observations , nr_realizations ))
148
+
149
+ rng = np .random .default_rng (1234 )
152
150
parameters_a = rng .normal (10 , 1 , nr_realizations )
151
+
153
152
for i in range (nr_observations ):
154
153
Y [i ] = (i + 1 ) * parameters_a
155
- result , * _ = main (Y , Y .mean (axis = 1 ))
154
+
155
+ scale_factors , clusters , nr_components = main (Y , Y .mean (axis = 1 ))
156
+
157
+ np .testing .assert_equal (
158
+ scale_factors ,
159
+ np .array (nr_observations * [1.0 ]),
160
+ )
161
+
162
+ np .testing .assert_equal (
163
+ clusters ,
164
+ np .array (nr_observations * [1.0 ]),
165
+ )
166
+
156
167
np .testing .assert_equal (
157
- result ,
168
+ nr_components ,
158
169
np .array (nr_observations * [1.0 ]),
159
170
)
0 commit comments