146 lines
4.9 KiB
Python
146 lines
4.9 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# @file
|
|
# @author (C) 2019 by Vasil Velichkov <vvvelichkov@gmail.com>
|
|
# @section LICENSE
|
|
#
|
|
# Gr-gsm is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 3, or (at your option)
|
|
# any later version.
|
|
#
|
|
# Gr-gsm is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with gr-gsm; see the file COPYING. If not, write to
|
|
# the Free Software Foundation, Inc., 51 Franklin Street,
|
|
# Boston, MA 02110-1301, USA.
|
|
#
|
|
#
|
|
|
|
import unittest
|
|
import numpy as np
|
|
from gnuradio import gr, gr_unittest, blocks
|
|
import grgsm
|
|
import pmt
|
|
import qa_gsm_demapper_data as test_data
|
|
|
|
class qa_gsm_bcch_ccch_demapper (gr_unittest.TestCase):
|
|
|
|
def setUp (self):
|
|
self.tb = gr.top_block ()
|
|
self.maxDiff = None
|
|
|
|
def tearDown (self):
|
|
self.tb = None
|
|
|
|
def test_downlink (self):
|
|
"""
|
|
BCCH_CCCH demapper downlink test
|
|
"""
|
|
src = grgsm.burst_source(test_data.frames, test_data.timeslots, test_data.bursts)
|
|
src.set_arfcn(0); # downlink
|
|
demapper = grgsm.gsm_bcch_ccch_demapper(timeslot_nr=0)
|
|
dst = grgsm.burst_sink()
|
|
|
|
self.tb.msg_connect(src, "out", demapper, "bursts")
|
|
self.tb.msg_connect(demapper, "bursts", dst, "in")
|
|
self.tb.run ()
|
|
|
|
b = test_data.bursts
|
|
self.assertEqual([
|
|
b[ 2], b[ 3], b[ 4], b[ 5], #BCCH
|
|
b[ 6], b[ 7], b[ 8], b[ 9], #CCCH skip 10-11
|
|
b[ 12], b[ 13], b[ 14], b[ 15],
|
|
b[ 16], b[ 17], b[ 18], b[ 19], #skip 20-21
|
|
b[ 22], b[ 23], b[ 24], b[ 25],
|
|
b[ 26], b[ 27], b[ 28], b[ 29], #skip 30-31
|
|
b[ 32], b[ 33], b[ 34], b[ 35],
|
|
b[ 36], b[ 37], b[ 38], b[ 39], #skip 40-41
|
|
b[ 42], b[ 43], b[ 44], b[ 45],
|
|
b[ 46], b[ 47], b[ 48], b[ 49], #skip 50-52
|
|
b[ 53], b[ 54], b[ 55], b[ 56], #BCCH
|
|
b[ 57], b[ 58], b[ 59], b[ 60], #CCCH skip 61-62
|
|
b[ 63], b[ 64], b[ 65], b[ 66],
|
|
b[ 67], b[ 68], b[ 69], b[ 70], #skip 71-72
|
|
b[ 73], b[ 74], b[ 75], b[ 76],
|
|
b[ 77], b[ 78], b[ 79], b[ 80], #skip 81-82
|
|
b[ 83], b[ 84], b[ 85], b[ 86],
|
|
b[ 87], b[ 88], b[ 89], b[ 90], #skip 91-92
|
|
b[ 93], b[ 94], b[ 95], b[ 96],
|
|
b[ 97], b[ 98], b[ 99], b[100], #skip 101-103
|
|
b[104], b[105], b[106], b[107] #BCCH
|
|
], list(dst.get_burst_data()))
|
|
|
|
self.assertEqual([
|
|
1, 1, 1, 1, #BCCH
|
|
2, 2, 2, 2, #CCCH
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
1, 1, 1, 1, #BCCH
|
|
2, 2, 2, 2, #CCCH
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
2, 2, 2, 2,
|
|
1, 1, 1, 1, #BCCH
|
|
], list(dst.get_sub_types()))
|
|
|
|
self.assertEqual([
|
|
0, 0, 0, 0, #BCCH
|
|
0, 0, 0, 0, #CCCH 0
|
|
1, 1, 1, 1,
|
|
2, 2, 2, 2,
|
|
3, 3, 3, 3,
|
|
4, 4, 4, 4,
|
|
5, 5, 5, 5,
|
|
6, 6, 6, 6,
|
|
7, 7, 7, 7,
|
|
8, 8, 8, 8, #CCCH 8
|
|
0, 0, 0, 0, #BCCH
|
|
0, 0, 0, 0, #CCCH 0
|
|
1, 1, 1, 1,
|
|
2, 2, 2, 2,
|
|
3, 3, 3, 3,
|
|
4, 4, 4, 4,
|
|
5, 5, 5, 5,
|
|
6, 6, 6, 6,
|
|
7, 7, 7, 7,
|
|
8, 8, 8, 8, #CCCH 8
|
|
0, 0, 0, 0, #BCCH
|
|
], list(dst.get_sub_slots()))
|
|
|
|
def test_uplink (self):
|
|
"""
|
|
BCCH_CCCH demapper uplink test
|
|
"""
|
|
src = grgsm.burst_source(test_data.frames, test_data.timeslots, test_data.bursts)
|
|
src.set_arfcn(0x2240); #uplink flag is 40
|
|
demapper = grgsm.gsm_bcch_ccch_demapper(timeslot_nr=0)
|
|
dst = grgsm.burst_sink()
|
|
|
|
self.tb.msg_connect(src, "out", demapper, "bursts")
|
|
self.tb.msg_connect(demapper, "bursts", dst, "in")
|
|
self.tb.run ()
|
|
|
|
b = test_data.bursts
|
|
self.assertEqual(b, list(dst.get_burst_data()))
|
|
self.assertEqual([3]*len(b), list(dst.get_sub_types()))
|
|
self.assertEqual([0]*len(b), list(dst.get_sub_slots()))
|
|
|
|
if __name__ == '__main__':
|
|
gr_unittest.run(qa_gsm_bcch_ccch_demapper, "qa_gsm_bcch_ccch_demapper.xml")
|