From 947ed211041c3a12fbf060a72d14f20274145423 Mon Sep 17 00:00:00 2001 From: Maximilian Gruber Date: Tue, 3 Mar 2020 09:42:00 +0100 Subject: [PATCH] fix(test_decomposition_realtime): adjust concat statement --- test/test_propagate_DWT.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/test_propagate_DWT.py b/test/test_propagate_DWT.py index 1dcbb0c5a..48f6db8fb 100644 --- a/test/test_propagate_DWT.py +++ b/test/test_propagate_DWT.py @@ -172,8 +172,9 @@ def test_decomposition_realtime(): coeffs_b, Ucoeffs_b, ol_b, z_b = wave_dec_realtime(x_batch, Ux_batch, ld, hd, n=2, level_states=z_b) coeffs_list.append(coeffs_b) Ucoeffs_list.append(Ucoeffs_b) - coeffs_b = np.concatenate(coeffs_list, axis=0) - Ucoeffs_b = np.concatenate(Ucoeffs_list, axis=0) + + coeffs_b = [np.concatenate([coeffs[level] for coeffs in coeffs_list], axis=0) for level in range(len(coeffs_list[0]))] + Ucoeffs_b = [np.concatenate([Ucoeffs[level] for Ucoeffs in Ucoeffs_list], axis=0) for level in range(len(Ucoeffs_list[0]))] # compare output depth assert len(coeffs_a) == len(coeffs_b)