4 files changed, 141 insertions, 1 deletions
diff --git a/gr-blocks/python/qa_burst_tagger.py b/gr-blocks/python/qa_burst_tagger.py
index 5e249cb8b9..c89ece4e15 100644
--- a/gr-blocks/python/qa_burst_tagger.py
+++ b/gr-blocks/python/qa_burst_tagger.py
@@ -33,7 +33,6 @@ class test_burst_tagger(gr_unittest.TestCase):
         self.tb = None
 
     def test_001(self):
-        # Just run some data through and make sure it doesn't puke.
         src_data  = ( 1,  2,  3,  4,  5,  6,  7,  8,  9,  10)
         trg_data = (-1, -1,  1,  1, -1, -1,  1,  1, -1,  -1)
         src = gr.vector_source_i(src_data)
diff --git a/gr-blocks/python/qa_tag_file_sink.py b/gr-blocks/python/qa_tag_file_sink.py
new file mode 100644
index 0000000000..80e41a7dd0
--- /dev/null
+++ b/gr-blocks/python/qa_tag_file_sink.py
@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+#
+# Copyright 2013 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
+import blocks_swig as blocks
+import os, struct
+
+class test_tag_file_sink(gr_unittest.TestCase):
+
+    def setUp(self):
+        self.tb = gr.top_block()
+
+    def tearDown(self):
+        self.tb = None
+
+    def test_001(self):
+        src_data = ( 1,  2,  3,  4,  5,  6,  7,  8,  9,  10)
+        trg_data = (-1, -1,  1,  1, -1, -1,  1,  1, -1,  -1)
+        src = gr.vector_source_i(src_data)
+        trg = gr.vector_source_s(trg_data)
+        op  = blocks.burst_tagger(gr.sizeof_int)
+        snk = blocks.tagged_file_sink(gr.sizeof_int, 1)
+        self.tb.connect(src, (op,0))
+        self.tb.connect(trg, (op,1))
+        self.tb.connect(op, snk)
+        self.tb.run()
+
+        # Tagged file sink gets 2 burst tags at index 2 and index 5.
+        # Creates two new files, each with two integers in them from
+        # src_data at these indexes (3,4) and (7,8).
+        file0 = "file{0}_0_2.00000000.dat".format(snk.unique_id())
+        file1 = "file{0}_1_6.00000000.dat".format(snk.unique_id())
+
+        # Open the files and read in the data, then remove the files
+        # to clean up the directory.
+        outfile0 = file(file0, 'rb')
+	outfile1 = file(file1, 'rb')
+	data0 = outfile0.read(8)
+	data1 = outfile1.read(8)
+        outfile0.close()
+        outfile1.close()
+	os.remove(file0)
+	os.remove(file1)
+
+        # Convert the 8 bytes from the files into a tuple of 2 ints.
+        idata0 = struct.unpack('ii', data0)
+        idata1 = struct.unpack('ii', data1)
+
+        self.assertEqual(idata0, (3, 4))
+        self.assertEqual(idata1, (7, 8))
+
+if __name__ == '__main__':
+    gr_unittest.run(test_tag_file_sink, "test_tag_file_sink.xml")
diff --git a/gr-blocks/python/qa_wavfile.py b/gr-blocks/python/qa_wavfile.py
new file mode 100755
index 0000000000..b40b9b7ec3
--- /dev/null
+++ b/gr-blocks/python/qa_wavfile.py
@@ -0,0 +1,69 @@
+#!/usr/bin/env python
+#
+# Copyright 2008,2010,2013 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
+import blocks_swig as blocks
+
+import os
+from os.path import getsize
+
+g_in_file = os.path.join(os.getenv("srcdir"), "test_16bit_1chunk.wav")
+
+class test_wavefile(gr_unittest.TestCase):
+
+    def setUp(self):
+        self.tb = gr.top_block()
+
+    def tearDown(self):
+        self.tb = None
+
+    def test_001_checkwavread(self):
+	wf = blocks.wavfile_source(g_in_file)
+	self.assertEqual(wf.sample_rate(), 8000)
+
+    def test_002_checkwavcopy(self):
+	infile  = g_in_file
+	outfile = "test_out.wav"
+
+	wf_in  = blocks.wavfile_source(infile)
+	wf_out = blocks.wavfile_sink(outfile,
+                                     wf_in.channels(),
+                                     wf_in.sample_rate(),
+                                     wf_in.bits_per_sample())
+	self.tb.connect(wf_in, wf_out)
+	self.tb.run()
+	wf_out.close()
+
+	self.assertEqual(getsize(infile), getsize(outfile))
+
+	in_f  = file(infile,  'rb')
+	out_f = file(outfile, 'rb')
+
+	in_data  = in_f.read()
+	out_data = out_f.read()
+        out_f.close()
+	os.remove(outfile)
+
+	self.assertEqual(in_data, out_data)
+
+if __name__ == '__main__':
+    gr_unittest.run(test_wavefile, "test_wavefile.xml")
diff --git a/gr-blocks/python/test_16bit_1chunk.wav b/gr-blocks/python/test_16bit_1chunk.wav
new file mode 100644
index 0000000000..0fe12a7a13
--- /dev/null
+++ b/gr-blocks/python/test_16bit_1chunk.wav