Commit 2eba1fb7 authored by Andrew Quinn's avatar Andrew Quinn
Browse files

Fix wavelet tests after tweak to get_morlet_basis

parent 84108d3b

sails/tests/test_wavelet.py

+3 −2
Original line number Diff line number Diff line
@@ -13,10 +13,11 @@ class MorletBasisTests(unittest.TestCase): 
        sample_rate = 128

        freqs = np.linspace(1, 50, 49)

        ncycles = 5

        win_len = 5



        from ..wavelet import get_morlet_basis



        mlt = get_morlet_basis(freqs, ncycles, sample_rate)

        mlt = get_morlet_basis(freqs, ncycles, win_len, sample_rate)



        # Check we have same number of morlet basis waves as input freqs

        assert(len(mlt) == freqs.shape[0])
@@ -26,7 +27,7 @@ class MorletBasisTests(unittest.TestCase): 
        assert(np.all(np.diff([len(m) for m in mlt]) <= 0))



        # Example with faster sample_rate

        mlt2 = get_morlet_basis(freqs, ncycles, sample_rate*2)

        mlt2 = get_morlet_basis(freqs, ncycles, win_len, sample_rate*2)

        check = np.c_[[len(m) for m in mlt], [len(m) for m in mlt2]]



        # Morlets from double sample rate should be twice as long (or off-by-one)