def test_001_t (self):
"""
Simple test to see if decimation is working at all.
"""
input_len = 2048
nsamples = 1024
decimation = 2
pspectrum_len = 512
n = 1
m = 64
self.siggen = siggen.signal_generator(n_sinusoids = n,
SNR = 20, samp_rate = 32e3,
nsamples = input_len)
self.esprit = specest.esprit(n, m, nsamples, pspectrum_len, decimation)
self.sink = gr.vector_sink_f(vlen=pspectrum_len)
self.tb.connect(self.siggen, self.esprit, self.sink)
self.tb.run ()
self.assertEqual(len(self.sink.data()), input_len / nsamples / decimation * pspectrum_len)
def test_002_t (self):
"""
Simple test to see if the decimation setters and getters work at all.
"""
input_len = 1024
nsamples = 256
decimation = 1
pspectrum_len = 512
n = 1
m = 64
self.siggen = siggen.signal_generator(n_sinusoids = n,
SNR = 20, samp_rate = 32e3,
nsamples = input_len)
self.esprit = specest.esprit(n, m, nsamples, pspectrum_len, decimation)
self.assertEqual(self.esprit.decimation(), 1)
self.esprit.set_decimation(2)
self.assertEqual(self.esprit.decimation(), 2)
self.sink = gr.vector_sink_f(vlen=pspectrum_len)
self.tb.connect(self.siggen, self.esprit, self.sink)
self.tb.run ()
self.assertEqual(len(self.sink.data()), input_len / nsamples / 2 * pspectrum_len)
