summaryrefslogtreecommitdiff
path: root/gr-wavelet/python/qa_classify.py
diff options
context:
space:
mode:
Diffstat (limited to 'gr-wavelet/python/qa_classify.py')
-rwxr-xr-xgr-wavelet/python/qa_classify.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/gr-wavelet/python/qa_classify.py b/gr-wavelet/python/qa_classify.py
index 5701bce8c1..c2972c92f2 100755
--- a/gr-wavelet/python/qa_classify.py
+++ b/gr-wavelet/python/qa_classify.py
@@ -26,6 +26,8 @@ import copy
#import pygsl.wavelet as wavelet # FIXME: pygsl not checked for in config
import math
import wavelet_swig
+import analog_swig as analog
+import blocks_swig as blocks
def sqr(x):
return x*x
@@ -66,7 +68,7 @@ class test_classify(gr_unittest.TestCase):
trg_data = src_data * 0.5
src = gr.vector_source_f(src_data)
dst = gr.vector_sink_f()
- rail = gr.rail_ff(-0.5, 0.5)
+ rail = analog.rail_ff(-0.5, 0.5)
self.tb.connect(src, rail)
self.tb.connect(rail, dst)
self.tb.run()
@@ -87,7 +89,7 @@ class test_classify(gr_unittest.TestCase):
trg_data = copy.deepcopy(src_data)
src = gr.vector_source_f(src_data, False, len(src_data))
- st = gr.stretch_ff(-1.0/5.0, len(src_data))
+ st = blocks.stretch_ff(-1.0/5.0, len(src_data))
dst = gr.vector_sink_f(len(src_data))
self.tb.connect(src, st)
self.tb.connect(st, dst)