pyTest: fix file analysis of raw format (eAnalyseLevelFirstGop)

Clement Champetier · Clement Champetier · commit dfc2c98c11cc · 2016-11-03T11:30:42.000+01:00
diff --git a/test/pyTest/testProperties.py b/test/pyTest/testProperties.py
@@ -114,25 +114,26 @@ def testCheckRawVideoProperties():
     inputFile = av.InputFile(inputFileName)
     properties = inputFile.getProperties()
 
+    # Check format
     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
+    assert_equals(properties.getFileSize(), 256293L)
 
-    expectedFileSize = 256293L
-    assert_equals(properties.getFileSize(), expectedFileSize)
-
-    expectedBitRate = 177200L
-    expectedNbFrames = 200
-    expectedDuration = 8
-    expectedFps = 25
-
+    # Check video stream when analyse the header
+    videoStream = properties.getVideoProperties()[0]
+    assert_equals(videoStream.getFps(), 25)
+    assert_equals(videoStream.getNbFrames(), 0) # stream nbFrames is unknown
+    assert_equals(videoStream.getDuration(), 0) # stream duration is unknown
+    assert_equals(videoStream.getBitRate(), 0) # stream bitrate is unknown
+    # Check video stream when analyse the first GOP
+    inputFile.analyse(av.NoDisplayProgress(), av.eAnalyseLevelFirstGop)
     videoStream = properties.getVideoProperties()[0]
-    assert_equals(videoStream.getNbFrames(), expectedNbFrames)
-    assert_equals(videoStream.getDuration(), expectedDuration)
-    assert_equals(videoStream.getBitRate(), expectedBitRate)
-    assert_equals(videoStream.getFps(), expectedFps)
+    assert_equals(videoStream.getNbFrames(), 200)
+    assert_equals(videoStream.getDuration(), 8)
+    assert_equals(videoStream.getBitRate(), 177200L)
 
 
 def testCheckAudioProperties():
diff --git a/test/pyTest/testTranscoderRewrap.py b/test/pyTest/testTranscoderRewrap.py
@@ -164,6 +164,7 @@ def testRewrapRawVideoStream():
     # get src file of wrap
     inputFileName = os.environ['AVTRANSCODER_TEST_VIDEO_RAW_FILE']
     src_inputFile = av.InputFile(inputFileName)
+    src_inputFile.analyse(av.NoDisplayProgress(), av.eAnalyseLevelFirstGop)
     src_properties = src_inputFile.getProperties()
     src_videoStream = src_properties.getVideoProperties()[0]
 
@@ -180,6 +181,7 @@ def testRewrapRawVideoStream():
 
     # get dst file of wrap
     dst_inputFile = av.InputFile(outputFileName)
+    dst_inputFile.analyse(av.NoDisplayProgress(), av.eAnalyseLevelFirstGop)
     dst_properties = dst_inputFile.getProperties()
     dst_videoStream = dst_properties.getVideoProperties()[0]
 
