Skip to content

Commit 78e8763

Browse files
Update test_cwt_wavelets.py
verify number of time steps reduced by hop_size
1 parent 4a8d3c2 commit 78e8763

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

pywt/tests/test_cwt_wavelets.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -387,6 +387,10 @@ def test_cwt_complex(dtype, tol, method):
387387
# verify same precision
388388
assert_equal(cfs.real.dtype, sst.dtype)
389389

390+
# verify number of time steps reduced by hop_size
391+
expected_time_len = int(np.ceil(len(sst) / hop_size))
392+
assert_equal(cfs.shape[1], expected_time_len)
393+
390394
# complex-valued transform equals sum of the transforms of the real
391395
# and imaginary components
392396
sst_complex = sst + 1j*sst

0 commit comments

Comments
 (0)