pyTest: added test to check properties of a raw video stream

Clement Champetier · Clement Champetier · commit 0fcd0cce06b4 · 2016-06-20T18:33:50.000+02:00
diff --git a/test/pyTest/testProperties.py b/test/pyTest/testProperties.py
@@ -3,12 +3,14 @@
 # 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_VIDEO_MP4_FILE') is None or \
-    os.environ.get('AVTRANSCODER_TEST_VIDEO_MOV_FILE') is None:
+    os.environ.get('AVTRANSCODER_TEST_VIDEO_MOV_FILE') is None or \
+    os.environ.get('AVTRANSCODER_TEST_VIDEO_RAW_FILE') is None:
     from nose.plugins.skip import SkipTest
     raise SkipTest("Need to define environment variables "
         "AVTRANSCODER_TEST_AUDIO_WAVE_FILE and "
         "AVTRANSCODER_TEST_VIDEO_MP4_FILE and "
-        "AVTRANSCODER_TEST_VIDEO_MOV_FILE")
+        "AVTRANSCODER_TEST_VIDEO_MOV_FILE and "
+        "AVTRANSCODER_TEST_VIDEO_RAW_FILE")
 
 from nose.tools import *
 
@@ -98,6 +100,36 @@ def testCheckVideoProperties():
     assert_equals( videoStream.getFps(), expectedFps )
 
 
+
+def testCheckRawVideoProperties():
+    """
+    Check properties of a raw video stream.
+    A raw stream does not contain header (so the duration, number of frames... needs to be computed).
+    """
+    inputFileName = os.environ['AVTRANSCODER_TEST_VIDEO_RAW_FILE']
+    progress = av.NoDisplayProgress()
+    inputFile = av.InputFile(inputFileName)
+    inputFile.analyse(progress, av.eAnalyseLevelFirstGop)
+    properties = inputFile.getProperties()
+
+    assert_true(properties.isRawFormat())
+    assert_equals(properties.getNbStreams(), 1)
+    assert_equals(properties.getNbVideoStreams(), 1)
+    assert_equals(properties.getDuration(), 0) # file duration is unknown
+    assert_equals(properties.getBitRate(), 0) # file bitrate is unknown
+
+    expectedBitRate = 260800L
+    expectedNbFrames = 200
+    expectedDuration = 8
+    expectedFps = 25
+
+    videoStream = properties.getVideoProperties()[0]
+    assert_equals(videoStream.getNbFrames(), expectedNbFrames)
+    assert_equals(videoStream.getDuration(), expectedDuration)
+    assert_equals(videoStream.getBitRate(), expectedBitRate)
+    assert_equals(videoStream.getFps(), expectedFps)
+
+
 def testCheckAudioProperties():
     """
     Check properties of an audio stream.
diff --git a/tools/travis/python.nosetests.sh b/tools/travis/python.nosetests.sh
@@ -12,6 +12,7 @@ git clone https://github.com/avTranscoder/avTranscoder-data.git
 export AVTRANSCODER_TEST_VIDEO_AVI_FILE=`pwd`/avTranscoder-data/video/BigBuckBunny/BigBuckBunny_480p_stereo.avi
 export AVTRANSCODER_TEST_VIDEO_MP4_FILE=`pwd`/avTranscoder-data/video/BigBuckBunny/BigBuckBunny_HD.mp4
 export AVTRANSCODER_TEST_VIDEO_MOV_FILE=`pwd`/avTranscoder-data/video/BigBuckBunny/BigBuckBunny_640p_h264.mov
+export AVTRANSCODER_TEST_VIDEO_RAW_FILE=`pwd`/avTranscoder-data/video/Intro/Intro_raw_1080p.h264
 export AVTRANSCODER_TEST_AUDIO_WAVE_FILE=`pwd`/avTranscoder-data/audio/frequenciesPerChannel.wav
 export AVTRANSCODER_TEST_AUDIO_MOV_FILE=`pwd`/avTranscoder-data/video/BigBuckBunny/BigBuckBunny_1080p_5_1.mov
 export AVTRANSCODER_TEST_IMAGE_PNG_FILE=`pwd`/avTranscoder-data/image/BigBuckBunny/bbb-splash.thumbnail.png
