jamie/wavinfo
1
0
Fork 0
mirror of https://github.com/iluvcapra/wavinfo.git synced 2026-01-02 09:50:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update test_wave_parsing.py

Browse Source 
Style
This commit is contained in:
Jamie Hardt
2019-01-05 12:25:34 -08:00
committed by GitHub
parent 0e1094421e
commit 1e0f31a794
1 changed files with 2 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_wave_parsing.py
View File

@@ -10,11 +10,8 @@ import wavinfo
FFPROBE='ffprobe' FFPROBE='ffprobe'
def ffprobe(path): def ffprobe(path):
arguments = [ FFPROBE , "-of", "json" , "-show_format", "-show_streams", path ] arguments = [ FFPROBE , "-of", "json" , "-show_format", "-show_streams", path ]
if int(sys.version[0]) < 3: if int(sys.version[0]) < 3:
process = subprocess.Popen(arguments, stdout=PIPE) process = subprocess.Popen(arguments, stdout=PIPE)
process.wait() process.wait()
@@ -27,7 +24,6 @@ def ffprobe(path):
return None return None
else: else:
process = subprocess.run(arguments, stdin=None, stdout=PIPE, stderr=PIPE) process = subprocess.run(arguments, stdin=None, stdout=PIPE, stderr=PIPE)
if process.returncode == 0: if process.returncode == 0:
output_str = process.stdout.decode('utf-8') output_str = process.stdout.decode('utf-8')
return json.loads(output_str) return json.loads(output_str)
@@ -35,16 +31,13 @@ def ffprobe(path):
return None return None
class TestWaveInfo(TestCase): class TestWaveInfo(TestCase):
def all_files(self): def all_files(self):
for dirpath, dirnames, filenames in os.walk('tests/test_files'): for dirpath, _, filenames in os.walk('tests/test_files'):
for filename in filenames: for filename in filenames:
name, ext = os.path.splitext(filename) _, ext = os.path.splitext(filename)
if ext in ['.wav','.WAV']: if ext in ['.wav','.WAV']:
yield os.path.join(dirpath, filename) yield os.path.join(dirpath, filename)
def test_sanity(self): def test_sanity(self):
for wav_file in self.all_files(): for wav_file in self.all_files():
info = wavinfo.WavInfoReader(wav_file) info = wavinfo.WavInfoReader(wav_file)
@@ -70,7 +63,6 @@ class TestWaveInfo(TestCase):
for wav_file in self.all_files(): for wav_file in self.all_files():
info = wavinfo.WavInfoReader(wav_file) info = wavinfo.WavInfoReader(wav_file)
ffprobe_info = ffprobe(wav_file) ffprobe_info = ffprobe(wav_file)
self.assertEqual( info.data.frame_count, int(ffprobe_info['streams'][0]['duration_ts'] )) self.assertEqual( info.data.frame_count, int(ffprobe_info['streams'][0]['duration_ts'] ))
def test_bext_against_ffprobe(self): def test_bext_against_ffprobe(self):