@@ -43,7 +43,7 @@ def setUp(self):
43
43
self .fs = 5000
44
44
# Input samp rate to channelizer.
45
45
self .ifs = self .M * self .fs
46
-
46
+
47
47
self .taps = filter .firdes .low_pass_2 (
48
48
1 , self .ifs , self .fs / 2 , self .fs / 10 ,
49
49
attenuation_dB = 80 ,
@@ -63,6 +63,18 @@ def test_0001(self):
63
63
self .check_channelizer (filter .pfb .channelizer_hier_ccf (
64
64
self .M , n_filterbanks = 1 , taps = self .taps ))
65
65
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
+
66
78
def get_input_data (self ):
67
79
"""
68
80
Get the raw data generated by addition of sinusoids.
@@ -108,7 +120,7 @@ def check_channelizer(self, channelizer_block):
108
120
109
121
expected_data = self .get_expected_data (L )
110
122
received_data = [snk .data () for snk in snks ]
111
-
123
+
112
124
for expected , received in zip (expected_data , received_data ):
113
125
self .compare_data (expected , received )
114
126
@@ -119,12 +131,12 @@ def compare_data(self, expected, received):
119
131
expected = [x / expected [0 ] for x in expected ]
120
132
received = [x / received [0 ] for x in received ]
121
133
self .assertComplexTuplesAlmostEqual (expected , received , 3 )
122
-
134
+
123
135
124
136
def get_freq (self , data ):
125
137
freqs = []
126
138
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 )
128
140
if diff > math .pi :
129
141
diff -= 2 * math .pi
130
142
if diff < - math .pi :
@@ -150,18 +162,6 @@ def get_expected_data(self, L):
150
162
1j * math .sin (2. * math .pi * f * x ), t ) for f in self .freqs ]
151
163
return expected_data
152
164
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
-
165
165
166
166
if __name__ == '__main__' :
167
167
gr_unittest .run (test_pfb_channelizer , "test_pfb_channelizer.xml" )
0 commit comments