Skip to content

Commit 967489a

Browse files
committed
filter: minor reorg of pfb_channelizer test cases after merge from maint.
1 parent 1992cc2 commit 967489a

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

gr-filter/python/filter/qa_pfb_channelizer.py

+16-16
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def setUp(self):
4343
self.fs = 5000
4444
# Input samp rate to channelizer.
4545
self.ifs = self.M*self.fs
46-
46+
4747
self.taps = filter.firdes.low_pass_2(
4848
1, self.ifs, self.fs/2, self.fs/10,
4949
attenuation_dB=80,
@@ -63,6 +63,18 @@ def test_0001(self):
6363
self.check_channelizer(filter.pfb.channelizer_hier_ccf(
6464
self.M, n_filterbanks=1, taps=self.taps))
6565

66+
def test_0002(self):
67+
"""Test roundig error handling for oversample rate (ok)."""
68+
channels, oversample = 36, 25.
69+
filter.pfb.channelizer_ccf(channels, taps=self.taps,
70+
oversample_rate=channels/oversample)
71+
72+
def test_0003(self):
73+
"""Test roundig error handling for oversample rate, (bad)."""
74+
self.assertRaises(RuntimeError,
75+
filter.pfb.channelizer_ccf,
76+
36, taps=self.taps, oversample_rate=10.1334)
77+
6678
def get_input_data(self):
6779
"""
6880
Get the raw data generated by addition of sinusoids.
@@ -108,7 +120,7 @@ def check_channelizer(self, channelizer_block):
108120

109121
expected_data = self.get_expected_data(L)
110122
received_data = [snk.data() for snk in snks]
111-
123+
112124
for expected, received in zip(expected_data, received_data):
113125
self.compare_data(expected, received)
114126

@@ -119,12 +131,12 @@ def compare_data(self, expected, received):
119131
expected = [x/expected[0] for x in expected]
120132
received = [x/received[0] for x in received]
121133
self.assertComplexTuplesAlmostEqual(expected, received, 3)
122-
134+
123135

124136
def get_freq(self, data):
125137
freqs = []
126138
for r1, r2 in zip(data[:-1], data[1:]):
127-
diff = cmath.phase(r1) - cmath.phase(r2)
139+
diff = cmath.phase(r1) - cmath.phase(r2)
128140
if diff > math.pi:
129141
diff -= 2*math.pi
130142
if diff < -math.pi:
@@ -150,18 +162,6 @@ def get_expected_data(self, L):
150162
1j*math.sin(2.*math.pi*f*x), t) for f in self.freqs]
151163
return expected_data
152164

153-
def test_0002(self):
154-
"""Test roundig error handling for oversample rate (ok)."""
155-
channels, oversample = 36, 25.
156-
filter.pfb.channelizer_ccf(channels, taps=self.taps,
157-
oversample_rate=channels/oversample)
158-
159-
def test_0003(self):
160-
"""Test roundig error handling for oversample rate, (bad)."""
161-
self.assertRaises(RuntimeError,
162-
filter.pfb.channelizer_ccf,
163-
36, taps=self.taps, oversample_rate=10.1334)
164-
165165

166166
if __name__ == '__main__':
167167
gr_unittest.run(test_pfb_channelizer, "test_pfb_channelizer.xml")

0 commit comments

Comments
 (0)