Test: add a new python test for audio channels muxing

Valentin Noel · Valentin Noel · commit 343ef4e829da · 2017-09-08T16:43:29.000+02:00
Audio channels are extracted from different file format sources, to
stereo and 5.1 output audio streams.
diff --git a/test/pyTest/testMuxAudioChannels.py b/test/pyTest/testMuxAudioChannels.py
@@ -0,0 +1,101 @@
+import os
+
+ # Check if environment is setup to run the tests
+if os.environ.get('AVTRANSCODER_TEST_AUDIO_WAVE_FILE') is None or \
+    os.environ.get('AVTRANSCODER_TEST_AUDIO_MOV_FILE') is None:
+    from nose.plugins.skip import SkipTest
+    raise SkipTest("Need to define environment variables "
+        "AVTRANSCODER_TEST_AUDIO_WAVE_FILE and "
+        "AVTRANSCODER_TEST_AUDIO_MOV_FILE")
+
+from nose.tools import *
+
+from pyAvTranscoder import avtranscoder as av
+
+def testMuxAudioChannelsFromDifferentFormatInputs_20():
+    """
+    Mux audio channels from different formats files, and generate one audio stereo stream
+    """
+    # inputs
+    inputFileName1 = os.environ['AVTRANSCODER_TEST_AUDIO_MOV_FILE']
+    inputFileName2 = os.environ['AVTRANSCODER_TEST_AUDIO_WAVE_FILE']
+    assert_not_equals(inputFileName1, inputFileName2)
+
+    inputs = av.InputStreamDescVector()
+    inputs.append(av.InputStreamDesc(inputFileName1, 1, 1))
+    inputs.append(av.InputStreamDesc(inputFileName2, 0, 2))
+
+    # output
+    outputFileName = "testMuxAudioChannelsFromDifferentFormatInputs_20.wav"
+    ouputFile = av.OutputFile(outputFileName)
+
+    transcoder = av.Transcoder(ouputFile)
+    transcoder.addStream(inputs, "wave24b48kstereo")
+
+    progress = av.ConsoleProgress()
+    processStat = transcoder.process( progress )
+
+    # check process stat returned
+    audioStat = processStat.getAudioStat(0)
+
+    inputFile1 = av.InputFile(inputFileName1)
+    inputFile2 = av.InputFile(inputFileName2)
+
+    src_audioStream1 = inputFile1.getProperties().getAudioProperties()[0]
+    src_audioStream2 = inputFile2.getProperties().getAudioProperties()[0]
+
+    min_src_duration = min(src_audioStream1.getDuration(), src_audioStream2.getDuration())
+
+    assert_equals(min_src_duration, audioStat.getDuration())
+
+    # check dst audio streams
+    dst_inputFile = av.InputFile(outputFileName)
+    dst_audioProperties = dst_inputFile.getProperties().getAudioProperties()
+    assert_equals(1, len(dst_audioProperties))
+    assert_equals(2, dst_audioProperties[0].getNbChannels())
+
+def testMuxAudioChannelsFromDifferentFormatInputs_51():
+    """
+    Mux audio channels from different formats files, and generate one audio stereo stream
+    """
+    # inputs
+    inputFileName1 = os.environ['AVTRANSCODER_TEST_AUDIO_MOV_FILE']
+    inputFileName2 = os.environ['AVTRANSCODER_TEST_AUDIO_WAVE_FILE']
+    assert_not_equals(inputFileName1, inputFileName2)
+
+    inputs = av.InputStreamDescVector()
+    inputs.append(av.InputStreamDesc(inputFileName1, 1, 1))
+    inputs.append(av.InputStreamDesc(inputFileName1, 1, 0))
+    inputs.append(av.InputStreamDesc(inputFileName2, 0, 2))
+    inputs.append(av.InputStreamDesc(inputFileName2, 0, 5))
+    inputs.append(av.InputStreamDesc(inputFileName2, 0, 1))
+    inputs.append(av.InputStreamDesc(inputFileName2, 0, 3))
+
+    # output
+    outputFileName = "testMuxAudioChannelsFromDifferentFormatInputs_51.wav"
+    ouputFile = av.OutputFile(outputFileName)
+
+    transcoder = av.Transcoder(ouputFile)
+    transcoder.addStream(inputs, "wave24b48k5_1")
+
+    progress = av.ConsoleProgress()
+    processStat = transcoder.process( progress )
+
+    # check process stat returned
+    audioStat = processStat.getAudioStat(0)
+
+    inputFile1 = av.InputFile(inputFileName1)
+    inputFile2 = av.InputFile(inputFileName2)
+
+    src_audioStream1 = inputFile1.getProperties().getAudioProperties()[0]
+    src_audioStream2 = inputFile2.getProperties().getAudioProperties()[0]
+
+    min_src_duration = min(src_audioStream1.getDuration(), src_audioStream2.getDuration())
+
+    assert_equals(min_src_duration, audioStat.getDuration())
+
+    # check dst audio streams
+    dst_inputFile = av.InputFile(outputFileName)
+    dst_audioProperties = dst_inputFile.getProperties().getAudioProperties()
+    assert_equals(1, len(dst_audioProperties))
+    assert_equals(6, dst_audioProperties[0].getNbChannels())
