diff options
| author | Ruben Undheim <ruben.undheim@gmail.com> | 2017-05-14 09:45:38 +0000 |
|---|---|---|
| committer | Ruben Undheim <ruben.undheim@gmail.com> | 2017-05-19 18:03:27 +0000 |
| commit | 484acc4eabdc19d8c72c8c0724bae76ad975d955 (patch) | |
| tree | 4c09b6e157f5a876b996336610ec7868df0cb533 /gr-blocks/python | |
| parent | 7a6dace271a53faccec6d691477e2c5ccc5e9d49 (diff) | |
Address comment in pull request to use "with" with tempfile
Diffstat (limited to 'gr-blocks/python')
| -rw-r--r-- | gr-blocks/python/blocks/qa_file_source_sink.py | 52 |
1 files changed, 25 insertions, 27 deletions
diff --git a/gr-blocks/python/blocks/qa_file_source_sink.py b/gr-blocks/python/blocks/qa_file_source_sink.py index 0a5887d442..081f58228f 100644 --- a/gr-blocks/python/blocks/qa_file_source_sink.py +++ b/gr-blocks/python/blocks/qa_file_source_sink.py @@ -107,23 +107,22 @@ class test_file_source_sink(gr_unittest.TestCase): src_data = range(1000) expected_result = range(1000) - filename = "tmp.32f" - src = blocks.vector_source_f(src_data) - snk = blocks.file_sink(gr.sizeof_float, filename) - snk.set_unbuffered(True) - - src2 = blocks.file_source(gr.sizeof_float, filename) - src2.set_begin_tag(True) snk2 = blocks.vector_sink_f() - self.tb.connect(src, snk) - self.tb.run() + with tempfile.NamedTemporaryFile() as temp: + src = blocks.vector_source_f(src_data) + snk = blocks.file_sink(gr.sizeof_float, temp.name) + snk.set_unbuffered(True) - self.tb.disconnect(src, snk) - self.tb.connect(src2, snk2) - self.tb.run() + src2 = blocks.file_source(gr.sizeof_float, temp.name) + src2.set_begin_tag(True) - os.remove(filename) + self.tb.connect(src, snk) + self.tb.run() + + self.tb.disconnect(src, snk) + self.tb.connect(src2, snk2) + self.tb.run() result_data = snk2.data() self.assertFloatTuplesAlmostEqual(expected_result, result_data) @@ -134,24 +133,23 @@ class test_file_source_sink(gr_unittest.TestCase): expected_result = range(1000) expected_result.extend(range(1000)) - filename = "tmp.32f" - src = blocks.vector_source_f(src_data) - snk = blocks.file_sink(gr.sizeof_float, filename) - snk.set_unbuffered(True) - - src2 = blocks.file_source(gr.sizeof_float, filename, True) - src2.set_begin_tag(True) - hd = blocks.head(gr.sizeof_float, 2000) snk2 = blocks.vector_sink_f() - self.tb.connect(src, snk) - self.tb.run() + with tempfile.NamedTemporaryFile() as temp: + src = blocks.vector_source_f(src_data) + snk = blocks.file_sink(gr.sizeof_float, temp.name) + snk.set_unbuffered(True) - self.tb.disconnect(src, snk) - self.tb.connect(src2, hd, snk2) - self.tb.run() + src2 = blocks.file_source(gr.sizeof_float, temp.name, True) + src2.set_begin_tag(True) + hd = blocks.head(gr.sizeof_float, 2000) - os.remove(filename) + self.tb.connect(src, snk) + self.tb.run() + + self.tb.disconnect(src, snk) + self.tb.connect(src2, hd, snk2) + self.tb.run() result_data = snk2.data() self.assertFloatTuplesAlmostEqual(expected_result, result_data) |