Tests

tests/__init__.py

@@ -0,0 +1,2 @@
+from . import test_wave_parsing
+

tests/test_files/WaveAgentSoundReport.csv

@@ -0,0 +1 @@
+SOUND REPORT

tests/test_wave_parsing.py

@@ -0,0 +1,99 @@
+import os.path
+import json
+import subprocess
+
+from unittest import TestCase
+
+import wavinfo
+
+FFPROBE='/usr/local/bin/ffprobe'
+
+
+def ffprobe(path):
+
+    arguments = [ FFPROBE , "-of", "json" , "-show_format", "-show_streams", path ]
+
+    process = subprocess.run(arguments, stdin=None, capture_output=True)
+
+    if process.returncode == 0:
+        return json.loads(process.stdout)
+    else:
+        return None
+
+
+class TestWaveInfo(TestCase):
+
+
+    def all_files(self):
+        for dirpath, dirnames, filenames in os.walk('tests/test_files'):
+            for filename in filenames:
+                name, ext = os.path.splitext(filename)
+                if ext == '.wav':
+                    yield os.path.join(dirpath, filename)
+
+
+    def test_sanity(self):
+        for wav_file in self.all_files():
+            info = wavinfo.WavInfoReader(wav_file)
+            self.assertTrue(info is not None)
+
+    def test_fmt_against_ffprobe(self):
+        for wav_file in self.all_files():
+            info = wavinfo.WavInfoReader(wav_file)
+            ffprobe_info = ffprobe(wav_file)
+
+            self.assertEqual( info.fmt.channel_count , ffprobe_info['streams'][0]['channels']  )
+            self.assertEqual( info.fmt.sample_rate   , int(ffprobe_info['streams'][0]['sample_rate'])  )
+            self.assertEqual( info.fmt.bits_per_sample, int(ffprobe_info['streams'][0]['bits_per_raw_sample']) )
+
+            if info.fmt.audio_format == 1:
+                self.assertTrue(ffprobe_info['streams'][0]['codec_name'].startswith('pcm')  )
+                byte_rate = int(ffprobe_info['streams'][0]['sample_rate']) \
+                        * ffprobe_info['streams'][0]['channels'] \
+                        * int(ffprobe_info['streams'][0]['bits_per_raw_sample']) / 8
+                self.assertEqual( info.fmt.byte_rate     , byte_rate  )
+
+    def test_data_against_ffprobe(self):
+        for wav_file in self.all_files():
+            info = wavinfo.WavInfoReader(wav_file)
+            ffprobe_info = ffprobe(wav_file)
+
+            self.assertEqual( info.data.frame_count, int(ffprobe_info['streams'][0]['duration_ts'] ))
+
+    def test_bext_against_ffprobe(self):
+        for wav_file in self.all_files():
+            info = wavinfo.WavInfoReader(wav_file)
+            ffprobe_info = ffprobe(wav_file)
+
+            self.assertEqual( info.bext.description, ffprobe_info['format']['tags']['comment']  )
+            self.assertEqual( info.bext.originator, ffprobe_info['format']['tags']['encoded_by']  )
+            self.assertEqual( info.bext.originator_ref, ffprobe_info['format']['tags']['originator_reference']  )
+
+            # these don't always reflect the bext info
+            #self.assertEqual( info.bext.originator_date, ffprobe_info['format']['tags']['date']  )
+            #self.assertEqual( info.bext.originator_time, ffprobe_info['format']['tags']['creation_time']  )
+            self.assertEqual( info.bext.time_reference, int(ffprobe_info['format']['tags']['time_reference'])  )
+            self.assertEqual( info.bext.coding_history, ffprobe_info['format']['tags']['coding_history']  )
+
+    def test_ixml(self):
+        expected = {'A101_4.WAV': {'project' : 'BMH', 'scene': 'A101', 'take': '4',
+                        'tape': '18Y12M31', 'family_uid': 'USSDVGR1112089007124015008231000'},
+                    'A101_3.WAV': {'project' : 'BMH', 'scene': 'A101', 'take': '3',
+                        'tape': '18Y12M31', 'family_uid': 'USSDVGR1112089007124014008228300'},
+                    'A101_2.WAV': {'project' : 'BMH', 'scene': 'A101', 'take': '2',
+                        'tape': '18Y12M31', 'family_uid': 'USSDVGR1112089007124004008218600'},
+                    'A101_1.WAV': {'project' : 'BMH', 'scene': 'A101', 'take': '1',
+                        'tape': '18Y12M31', 'family_uid': 'USSDVGR1112089007124001008206300'},
+                }
+
+        for wav_file in self.all_files():
+            basename =  os.path.basename(wav_file)
+            if basename in expected:
+                info = wavinfo.WavInfoReader(wav_file)
+                e = expected[basename]
+
+                self.assertEqual( e['project'], info.ixml.project )
+                self.assertEqual( e['scene'], info.ixml.scene )
+                self.assertEqual( e['take'], info.ixml.take )
+                self.assertEqual( e['tape'], info.ixml.tape )
+                self.assertEqual( e['family_uid'], info.ixml.family_uid )