Add OutputFile unit test

valnoel · valnoel · commit 6671f364e0a3 · 2020-01-23T17:23:24.000+01:00
diff --git a/test/pyTest/testOutputFile.py b/test/pyTest/testOutputFile.py
@@ -95,3 +95,20 @@ def testGetUnexistedOutputStream():
     outputFileName = "testGetUnexistedOutputStream.mov"
     ouputFile = av.OutputFile(outputFileName)
     ouputFile.getStream(0)
+
+
+def testAddingCustomStream():
+    """
+    Create an OutputFile, and add a custom stream and try to access that stream.
+    """
+    outputFileName = "testAddingCustomStream.mov"
+    ouputFile = av.OutputFile(outputFileName)
+
+    codec = av.AudioCodec(av.eCodecTypeEncoder, "pcm_s24le");
+    addedOutputStream = ouputFile.addCustomStream(codec)
+
+    retrievedOutputStream = ouputFile.getStream(0)
+
+    assert_equals(addedOutputStream.getStreamIndex(), retrievedOutputStream.getStreamIndex())
+    assert_equals(addedOutputStream.getStreamDuration(), retrievedOutputStream.getStreamDuration())
+    assert_equals(addedOutputStream.getNbFrames(), retrievedOutputStream.getNbFrames())
