pyTest: split reader tests to testVideo/AudioReader.py

Clement Champetier · Clement Champetier · commit 5cd30bcd50d2 · 2016-10-31T12:27:36.000+01:00
diff --git a/test/pyTest/testAudioReader.py b/test/pyTest/testAudioReader.py
@@ -0,0 +1,65 @@
+import os
+
+# Check if environment is setup to run the tests
+if os.environ.get('AVTRANSCODER_TEST_AUDIO_WAVE_FILE') is None:
+    from nose.plugins.skip import SkipTest
+    raise SkipTest("Need to define environment variables AVTRANSCODER_TEST_AUDIO_WAVE_FILE")
+
+from nose.tools import *
+
+from pyAvTranscoder import avtranscoder as av
+
+
+def testAudioReaderChannelsExtraction():
+    """
+    Read the same audio stream with several AudioReaders.
+    Compare decoded frames from reader of all channels, and of one channel.
+    """
+    inputFileName = os.environ['AVTRANSCODER_TEST_AUDIO_WAVE_FILE']
+    inputFile = av.InputFile(inputFileName)
+    streamIndex = inputFile.getProperties().getAudioProperties()[0].getStreamIndex()
+    channelIndex = 0
+
+    # create reader to read all channels of the audio stream
+    readerOfAllChannels = av.AudioReader(inputFile, streamIndex)
+    nbChannels = readerOfAllChannels.getOutputNbChannels()
+    # read first frame
+    frame = readerOfAllChannels.readNextFrame()
+    sizeOfFrameWithAllChannels = frame.getSize()
+
+    # create reader to read one channel of the audio stream
+    readerOfOneChannel = av.AudioReader(inputFile, streamIndex, channelIndex)
+    # read first frame
+    frame = readerOfOneChannel.readNextFrame()
+    sizeOfFrameWithOneChannels = frame.getSize()
+
+    assert_equals( sizeOfFrameWithAllChannels / nbChannels, sizeOfFrameWithOneChannels )
+
+
+def testAudioReaderWithGenerator():
+    """
+    Read an audio stream with the AudioReader.
+    When there is no more data to decode, switch to a generator and process some frames.
+    """
+    inputFileName = os.environ['AVTRANSCODER_TEST_AUDIO_WAVE_FILE']
+    inputFile = av.InputFile(inputFileName)
+    reader = av.AudioReader(inputFile)
+
+    # read all frames and check their size
+    while True:
+        frame = reader.readNextFrame()
+        if not frame:
+            break
+        assert_greater(frame.getSize(), 0)
+
+    # check if there is no next frame
+    assert_equals( reader.readNextFrame(), None )
+
+    # generate 10 frames of silence
+    reader.continueWithGenerator()
+    for i in xrange(0, 9):
+        frame = reader.readNextFrame()
+        # assuming we generate data of 1920 samples of 2 bytes
+        nbSamplesPerChannel = 1920
+        bytesPerSample = 2
+        assert_equals(frame.getSize(), reader.getOutputNbChannels() * nbSamplesPerChannel * bytesPerSample )
diff --git a/test/pyTest/testVideoReader.py b/test/pyTest/testVideoReader.py
@@ -1,12 +1,9 @@
 import os
 
 # Check if environment is setup to run the tests
-if os.environ.get('AVTRANSCODER_TEST_VIDEO_AVI_FILE') is None or \
-    os.environ.get('AVTRANSCODER_TEST_AUDIO_WAVE_FILE') is None:
+if os.environ.get('AVTRANSCODER_TEST_VIDEO_AVI_FILE') is None:
     from nose.plugins.skip import SkipTest
-    raise SkipTest("Need to define environment variables "
-        "AVTRANSCODER_TEST_VIDEO_AVI_FILE and "
-        "AVTRANSCODER_TEST_AUDIO_WAVE_FILE")
+    raise SkipTest("Need to define environment variables AVTRANSCODER_TEST_VIDEO_AVI_FILE")
 
 from nose.tools import *
 
@@ -51,32 +48,6 @@ def testVideoReaderReferenceInputFile():
     assert_equals( reader.readNextFrame(), None )
 
 
-def testAudioReaderChannelsExtraction():
-    """
-    Read the same audio stream with several AudioReaders.
-    Compare decoded frames from reader of all channels, and of one channel.
-    """
-    inputFileName = os.environ['AVTRANSCODER_TEST_AUDIO_WAVE_FILE']
-    inputFile = av.InputFile(inputFileName)
-    streamIndex = inputFile.getProperties().getAudioProperties()[0].getStreamIndex()
-    channelIndex = 0
-
-    # create reader to read all channels of the audio stream
-    readerOfAllChannels = av.AudioReader(inputFile, streamIndex)
-    nbChannels = readerOfAllChannels.getOutputNbChannels()
-    # read first frame
-    frame = readerOfAllChannels.readNextFrame()
-    sizeOfFrameWithAllChannels = frame.getSize()
-
-    # create reader to read one channel of the audio stream
-    readerOfOneChannel = av.AudioReader(inputFile, streamIndex, channelIndex)
-    # read first frame
-    frame = readerOfOneChannel.readNextFrame()
-    sizeOfFrameWithOneChannels = frame.getSize()
-
-    assert_equals( sizeOfFrameWithAllChannels / nbChannels, sizeOfFrameWithOneChannels )
-
-
 def testVideoReaderWithGenerator():
     """
     Read a video stream with the VideoReader.
@@ -100,32 +71,3 @@ def testVideoReaderWithGenerator():
         frame = reader.readNextFrame()
         bytesPerPixel = reader.getOutputBitDepth() / 8
         assert_equals( frame.getSize(), reader.getOutputWidth() * reader.getOutputHeight() * bytesPerPixel )
-
-
-def testAudioReaderWithGenerator():
-    """
-    Read an audio stream with the AudioReader.
-    When there is no more data to decode, switch to a generator and process some frames.
-    """
-    inputFileName = os.environ['AVTRANSCODER_TEST_AUDIO_WAVE_FILE']
-    inputFile = av.InputFile(inputFileName)
-    reader = av.AudioReader(inputFile)
-
-    # read all frames and check their size
-    while True:
-        frame = reader.readNextFrame()
-        if not frame:
-            break
-        assert_greater(frame.getSize(), 0)
-
-    # check if there is no next frame
-    assert_equals( reader.readNextFrame(), None )
-
-    # generate 10 frames of silence
-    reader.continueWithGenerator()
-    for i in xrange(0, 9):
-        frame = reader.readNextFrame()
-        # assuming we generate data of 1920 samples of 2 bytes
-        nbSamplesPerChannel = 1920
-        bytesPerSample = 2
-        assert_equals(frame.getSize(), reader.getOutputNbChannels() * nbSamplesPerChannel * bytesPerSample )
