Skip to content

Commit 9c2b414

Browse files
Update test_cwt_wavelets.py
1 parent 368f272 commit 9c2b414

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

pywt/tests/test_cwt_wavelets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -428,7 +428,7 @@ def test_cwt_batch(axis, method):
428428
assert_equal(cfs.shape[1 + axis], sst.shape[axis])
429429

430430
# batch result on stacked input is the same as stacked 1d result
431-
assert_almost_equal(cfs, np.stack((cfs1,) * n_batch, axis=batch_axis + 1),
431+
assert_almost_equal(cfs, np.stack((cfs1,) * np.ceil(n_batch / hop_size).astype(int), axis=batch_axis + 1),
432432
decimal=12)
433433

434434

0 commit comments

Comments
 (0)