#!/usr/bin/env python
#
-# Copyright 2006 Free Software Foundation, Inc.
+# Copyright 2006,2007 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
class test_goertzel(gr_unittest.TestCase):
def setUp(self):
- self.fg = gr.flow_graph()
+ self.tb = gr.top_block()
def tearDown(self):
- self.fg = None
+ self.tb = None
def make_tone_data(self, rate, freq):
return [cos(2*pi*x*freq/rate) for x in range(rate)]
src = gr.vector_source_f(src_data, False)
dft = gr.goertzel_fc(rate, rate, freq)
dst = gr.vector_sink_c()
- self.fg.connect(src, dft, dst)
- self.fg.run()
+ self.tb.connect(src, dft, dst)
+ self.tb.run()
return dst.data()
def test_001(self): # Measure single tone magnitude
src_data = self.make_tone_data(rate, freq)
expected_result = 0.5
actual_result = abs(self.transform(src_data, rate, bin)[0])
- self.assertAlmostEqual(expected_result, actual_result, places=5)
+ self.assertAlmostEqual(expected_result, actual_result, places=4)
def test_002(self): # Measure off frequency magnitude
rate = 8000
src_data = self.make_tone_data(rate, freq)
expected_result = 0.0
actual_result = abs(self.transform(src_data, rate, bin)[0])
- self.assertAlmostEqual(expected_result, actual_result, places=5)
+ self.assertAlmostEqual(expected_result, actual_result, places=4)
if __name__ == '__main__':
gr_unittest.main()