| #!/usr/bin/env python |
| # -*- coding: utf-8 -*- |
| # @file |
| # @author Roman Khassraf <rkhassraf@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. |
| # |
| # |
| |
| from gnuradio import gr, gr_unittest, blocks |
| import grgsm_swig as grgsm |
| import os |
| import pmt |
| import sys |
| import tempfile |
| |
| class qa_burst_printer (gr_unittest.TestCase): |
| |
| def setUp (self): |
| self.tb = gr.top_block() |
| self.tmpfile = tempfile.NamedTemporaryFile() |
| self.prevfd = os.dup(sys.stdout.fileno()) |
| os.dup2(self.tmpfile.fileno(), sys.stdout.fileno()) |
| self.prev = sys.stdout |
| sys.stdout = os.fdopen(self.prevfd, "w") |
| |
| def tearDown (self): |
| self.tb = None |
| os.dup2(self.prevfd, self.prev.fileno()) |
| sys.stdout = self.prev |
| self.tmpfile.close() |
| |
| def getOutput(self): |
| self.tmpfile.seek(0) |
| return self.tmpfile.read() |
| |
| def getOutputExpected(self, expected_lines): |
| out = "" |
| for l in expected_lines: |
| out = out + l + "\n" |
| return out |
| |
| def test_001_complete_bursts_prefix (self): |
| """ |
| Complete bursts, without any prefix |
| """ |
| framenumbers_input = [2569043, 2569044, 2569045, 2569046] |
| timeslots_input = [2, 2, 2, 2] |
| bursts_input = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| expected_lines = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| src = grgsm.burst_source(framenumbers_input, timeslots_input, bursts_input) |
| printer = grgsm.bursts_printer(pmt.intern(""), False, False, False, False) |
| self.tb.msg_connect(src, "out", printer, "bursts") |
| self.tb.run() |
| |
| self.assertEqual(self.getOutput(), self.getOutputExpected(expected_lines)) |
| |
| def test_002_complete_bursts_prefix (self): |
| """ |
| Complete bursts, with a string prefix |
| """ |
| framenumbers_input = [2569043, 2569044, 2569045, 2569046] |
| timeslots_input = [2, 2, 2, 2] |
| bursts_input = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| expected_lines = [ |
| "Test 0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "Test 0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "Test 0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "Test 0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| src = grgsm.burst_source(framenumbers_input, timeslots_input, bursts_input) |
| printer = grgsm.bursts_printer(pmt.intern("Test "), False, False, False, False) |
| self.tb.msg_connect(src, "out", printer, "bursts") |
| self.tb.run() |
| |
| self.assertEqual(self.getOutput(), self.getOutputExpected(expected_lines)) |
| |
| def test_003_complete_bursts_fnr (self): |
| """ |
| Complete bursts, no prefix, with frame number |
| """ |
| framenumbers_input = [2569043, 2569044, 2569045, 2569046] |
| timeslots_input = [2, 2, 2, 2] |
| bursts_input = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| expected_lines = [ |
| "2569043: 0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "2569044: 0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "2569045: 0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "2569046: 0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| src = grgsm.burst_source(framenumbers_input, timeslots_input, bursts_input) |
| printer = grgsm.bursts_printer(pmt.intern(""), True, False, False, False) |
| self.tb.msg_connect(src, "out", printer, "bursts") |
| self.tb.run() |
| |
| self.assertEqual(self.getOutput(), self.getOutputExpected(expected_lines)) |
| |
| def test_004_complete_bursts_fcount (self): |
| """ |
| Complete bursts, no prefix, with frame count |
| """ |
| framenumbers_input = [2569043, 2569044, 2569045, 2569046] |
| timeslots_input = [2, 2, 2, 2] |
| bursts_input = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| expected_lines = [ |
| "3967625: 0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "3967658: 0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "3967691: 0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "3967724: 0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| src = grgsm.burst_source(framenumbers_input, timeslots_input, bursts_input) |
| printer = grgsm.bursts_printer(pmt.intern(""), False, True, False, False) |
| self.tb.msg_connect(src, "out", printer, "bursts") |
| self.tb.run() |
| |
| self.assertEqual(self.getOutput(), self.getOutputExpected(expected_lines)) |
| |
| def test_005_complete_bursts_fnr_fcount (self): |
| """ |
| Complete bursts, no prefix, with frame number and frame count |
| """ |
| framenumbers_input = [2569043, 2569044, 2569045, 2569046] |
| timeslots_input = [2, 2, 2, 2] |
| bursts_input = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| expected_lines = [ |
| "2569043 3967625: 0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "2569044 3967658: 0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "2569045 3967691: 0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "2569046 3967724: 0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| src = grgsm.burst_source(framenumbers_input, timeslots_input, bursts_input) |
| printer = grgsm.bursts_printer(pmt.intern(""), True, True, False, False) |
| self.tb.msg_connect(src, "out", printer, "bursts") |
| self.tb.run() |
| |
| self.assertEqual(self.getOutput(), self.getOutputExpected(expected_lines)) |
| |
| def test_006_payload_prefix_fnr_fcount (self): |
| """ |
| Payload only, no prefix, with frame number and frame count |
| Bursts 4-7 are dummy bursts |
| """ |
| framenumbers_input = [2569043, 2569044, 2569045, 2569046, 1099602, 1099603, 1099604, 1099605] |
| timeslots_input = [2, 2, 2, 2, 4, 4, 4, 4] |
| bursts_input = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000" |
| |
| ] |
| |
| expected_lines = [ |
| "test_006: 2569043 3967625: 110000001001001111011111001111100000000101000111100000000100101101010000001111010100010110111101011101011100000101", |
| "test_006: 2569044 3967658: 100010111110111111000001010000101101111101011111010101110110110111101101111110000011011010111011111001011101000011", |
| "test_006: 2569045 3967691: 000100001111011111010101100100011000000000011011010110001001010100101011111001000111100000100000111111000000101110", |
| "test_006: 2569046 3967724: 110101011111001000101011010110000001110110001111111010010111000000001010010111001111111011001000000001001000011101", |
| "test_006: 1099602 1699146: 111110110111011000001010010011100000100100010000000111110111010010100011001100111001111010011111000100101111101010", |
| "test_006: 1099603 1699179: 111110110111011000001010010011100000100100010000000111110111010010100011001100111001111010011111000100101111101010", |
| "test_006: 1099604 1699212: 111110110111011000001010010011100000100100010000000111110111010010100011001100111001111010011111000100101111101010", |
| "test_006: 1099605 1699245: 111110110111011000001010010011100000100100010000000111110111010010100011001100111001111010011111000100101111101010" |
| ] |
| |
| src = grgsm.burst_source(framenumbers_input, timeslots_input, bursts_input) |
| printer = grgsm.bursts_printer(pmt.intern("test_006: "), True, True, True, False) |
| self.tb.msg_connect(src, "out", printer, "bursts") |
| self.tb.run() |
| |
| self.assertEqual(self.getOutput(), self.getOutputExpected(expected_lines)) |
| |
| def test_007_payload_prefix_fnr_fcount (self): |
| """ |
| Payload only, no prefix, with frame number, frame count, and ignoring dummy bursts |
| Bursts 4-7 are dummy bursts |
| """ |
| framenumbers_input = [2569043, 2569044, 2569045, 2569046, 1099602, 1099603, 1099604, 1099605] |
| timeslots_input = [2, 2, 2, 2, 4, 4, 4, 4] |
| bursts_input = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000" |
| |
| ] |
| |
| expected_lines = [ |
| "test_007: 2569043 3967625: 110000001001001111011111001111100000000101000111100000000100101101010000001111010100010110111101011101011100000101", |
| "test_007: 2569044 3967658: 100010111110111111000001010000101101111101011111010101110110110111101101111110000011011010111011111001011101000011", |
| "test_007: 2569045 3967691: 000100001111011111010101100100011000000000011011010110001001010100101011111001000111100000100000111111000000101110", |
| "test_007: 2569046 3967724: 110101011111001000101011010110000001110110001111111010010111000000001010010111001111111011001000000001001000011101" |
| ] |
| |
| src = grgsm.burst_source(framenumbers_input, timeslots_input, bursts_input) |
| printer = grgsm.bursts_printer(pmt.intern("test_007: "), True, True, True, True) |
| self.tb.msg_connect(src, "out", printer, "bursts") |
| self.tb.run() |
| |
| self.assertEqual(self.getOutput(), self.getOutputExpected(expected_lines)) |
| |
| def test_008_complete_prefix_fnr_fcount (self): |
| """ |
| Complete bursts, no prefix, with frame number, frame count, and ignoring dummy bursts |
| Bursts 4-7 are dummy bursts |
| """ |
| framenumbers_input = [2569043, 2569044, 2569045, 2569046, 1099602, 1099603, 1099604, 1099605] |
| timeslots_input = [2, 2, 2, 2, 4, 4, 4, 4] |
| bursts_input = [ |
| "0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000", |
| "0001111101101110110000010100100111000001001000100000001111100011100010111000101110001010111010010100011001100111001111010011111000100101111101010000" |
| |
| ] |
| |
| expected_lines = [ |
| "test_008: 2569043 3967625: 0001100000010010011110111110011111000000001010001111000000001000110101110010000011010111100101101010000001111010100010110111101011101011100000101000", |
| "test_008: 2569044 3967658: 0001000101111101111110000010100001011011111010111110101011101000110101110010000011010111110110111101101111110000011011010111011111001011101000011000", |
| "test_008: 2569045 3967691: 0000001000011110111110101011001000110000000000110110101100011000110101110010000011010111001010100101011111001000111100000100000111111000000101110000", |
| "test_008: 2569046 3967724: 0001101010111110010001010110101100000011101100011111110100101000110101110010000011010111111000000001010010111001111111011001000000001001000011101000" |
| ] |
| |
| src = grgsm.burst_source(framenumbers_input, timeslots_input, bursts_input) |
| printer = grgsm.bursts_printer(pmt.intern("test_008: "), True, True, False, True) |
| self.tb.msg_connect(src, "out", printer, "bursts") |
| self.tb.run() |
| |
| self.assertEqual(self.getOutput(), self.getOutputExpected(expected_lines)) |
| |
| |
| if __name__ == '__main__': |
| gr_unittest.run(qa_burst_printer, "qa_burst_printer.xml") |