summaryrefslogtreecommitdiff
path: root/gr-wavelet/python/wavelet/qa_classify.py
diff options
context:
space:
mode:
authorDouglas Anderson <danderson@ntia.doc.gov>2017-02-12 15:52:19 -0800
committerJohnathan Corgan <johnathan@corganlabs.com>2017-02-26 18:21:22 -0800
commit9e625c4821f4c63421b3d3747c0c4f358fef6c5f (patch)
tree41dedbe053417be7314cdce15d64fbbb89db4d8d /gr-wavelet/python/wavelet/qa_classify.py
parente5aabcc6a4a9335f3ef8abf5f89104b626e9364d (diff)
python3: update non-GRC components to use python2 or python3
Diffstat (limited to 'gr-wavelet/python/wavelet/qa_classify.py')
-rw-r--r--[-rwxr-xr-x]gr-wavelet/python/wavelet/qa_classify.py18
1 files changed, 10 insertions, 8 deletions
diff --git a/gr-wavelet/python/wavelet/qa_classify.py b/gr-wavelet/python/wavelet/qa_classify.py
index 4752620f4d..7f562c628e 100755..100644
--- a/gr-wavelet/python/wavelet/qa_classify.py
+++ b/gr-wavelet/python/wavelet/qa_classify.py
@@ -20,6 +20,8 @@
# Boston, MA 02110-1301, USA.
#
+from __future__ import division
+
import numpy
from gnuradio import gr, gr_unittest, wavelet, analog, blocks
import copy
@@ -79,14 +81,14 @@ class test_classify(gr_unittest.TestCase):
def test_002_(self):
src_data = numpy.array([-1.0,
- -1.0/2.0,
- -1.0/3.0,
- -1.0/4.0,
- -1.0/5.0])
+ -1.0 / 2.0,
+ -1.0 / 3.0,
+ -1.0 / 4.0,
+ -1.0 / 5.0])
trg_data = copy.deepcopy(src_data)
src = blocks.vector_source_f(src_data, False, len(src_data))
- st = blocks.stretch_ff(-1.0/5.0, len(src_data))
+ st = blocks.stretch_ff(-1.0 / 5.0, len(src_data))
dst = blocks.vector_sink_f(len(src_data))
self.tb.connect(src, st)
self.tb.connect(st, dst)
@@ -155,11 +157,11 @@ class test_classify(gr_unittest.TestCase):
dwav = numpy.array(src_data)
wvps = numpy.zeros(3)
# wavelet power spectrum
- scl = 1.0/sqr(dwav[0])
+ scl = 1.0 / sqr(dwav[0])
k = 1
for e in range(len(wvps)):
- wvps[e] = scl*sqr(dwav[k:k+(01<<e)]).sum()
- k += 01<<e
+ wvps[e] = scl*sqr(dwav[k:k+(0o1<<e)]).sum()
+ k += 0o1<<e
src = blocks.vector_source_f(src_data, False, len(src_data))
kon = wavelet.wvps_ff(len(src_data))