diff options
| author | Jeff Long <willcode4@gmail.com> | 2018-01-16 18:19:05 -0500 |
|---|---|---|
| committer | Martin Braun <martin.braun@ettus.com> | 2018-02-08 16:01:08 -0800 |
| commit | 6ae02d1d6d04a5f529ab3fea496c589f8d14229d (patch) | |
| tree | 3491ab847228dc2bb7c315c7b421eeb1423533c5 /gr-blocks/python/blocks/qa_file_sink.py | |
| parent | 2270a41d12ee028905302233d8178568d6da6daa (diff) | |
file_source: add optional offset (in items) and len (in items)
- Adds unit tests, and splits existing unit tests into different files:
qa_file_source.py - tests file_source_f
qa_file_sink.py - tests file_sink_f
qa_file_descriptor_source_sink.py - tests file_descriptor_*_f
Diffstat (limited to 'gr-blocks/python/blocks/qa_file_sink.py')
| -rw-r--r-- | gr-blocks/python/blocks/qa_file_sink.py | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/gr-blocks/python/blocks/qa_file_sink.py b/gr-blocks/python/blocks/qa_file_sink.py new file mode 100644 index 0000000000..a7296183ef --- /dev/null +++ b/gr-blocks/python/blocks/qa_file_sink.py @@ -0,0 +1,60 @@ +#!/usr/bin/env python +# +# Copyright 2018 Free Software Foundation, Inc. +# +# This file is part of GNU Radio +# +# GNU Radio 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. +# +# GNU Radio 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 GNU Radio; 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 os +import tempfile +import pmt +import array + +class test_file_sink(gr_unittest.TestCase): + + def setUp (self): + os.environ['GR_CONF_CONTROLPORT_ON'] = 'False' + self.tb = gr.top_block () + + def tearDown (self): + self.tb = None + + def test_file_sink(self): + data = range(1000) + expected_result = data + + with tempfile.NamedTemporaryFile() as temp: + src = blocks.vector_source_f(data) + snk = blocks.file_sink(gr.sizeof_float, temp.name) + snk.set_unbuffered(True) + self.tb.connect(src, snk) + self.tb.run() + + # Check file length (float: 4 * nsamples) + st = os.stat(temp.name) + self.assertEqual(st.st_size, 4 * len(data)) + + # Check file contents + datafile = open(temp.name, 'r') + result_data = array.array('f') + result_data.fromfile(datafile, len(data)) + self.assertFloatTuplesAlmostEqual(expected_result, result_data) + +if __name__ == '__main__': + gr_unittest.run(test_file_sink, "test_file_sink.xml") |