pyTest: add tests to check Video/Audio frames

Clement Champetier · Clement Champetier · commit 9c90c2d579ab · 2016-10-31T12:21:46.000+01:00
diff --git a/test/pyTest/testAudioFrame.py b/test/pyTest/testAudioFrame.py
@@ -0,0 +1,52 @@
+from nose.tools import *
+
+from pyAvTranscoder import avtranscoder as av
+
+
+@raises(MemoryError)
+def testInvalidAudioFrameAutoAllocated():
+    """
+    Try to create an invalid AudioFrame automatically allocated.
+    """
+    desc = av.AudioFrameDesc(4800, 1, "toto")
+    av.AudioFrame(desc)
+
+
+@raises(MemoryError)
+def testInvalidAudioFrameManualAllocated():
+    """
+    Try to create an invalid AudioFrame manually allocated.
+    """
+    sampleRate = 48000
+    nbChannels = 1
+    sampleFormat = "titi"
+    desc = av.AudioFrameDesc(sampleRate, nbChannels, sampleFormat)
+    frame = av.AudioFrame(desc, False)
+
+    assert_equals(frame.isDataAllocated(), False)
+    assert_equals(frame.getSize(), 0)
+    assert_equals(frame.getSampleRate(), sampleRate)
+    assert_equals(frame.getNbChannels(), nbChannels)
+    assert_equals(frame.getChannelLayoutDesc(), "mono")
+    assert_equals(frame.getNbSamplesPerChannel(), 1920)
+    assert_equals(frame.getBytesPerSample(), 0)
+    assert_equals(av.getSampleFormatName(frame.getSampleFormat()), "")
+
+    frame.allocateData()
+
+def testAudioFrame():
+    """
+    Check the size and the data buffer of a AudioFrame.
+    """
+    sampleRate = 48000
+    nbChannels = 1
+    sampleFormat = "s32"
+    desc = av.AudioFrameDesc(sampleRate, nbChannels, sampleFormat)
+    frame = av.AudioFrame(desc)
+
+    assert_equals(frame.isDataAllocated(), True)
+    assert_equals(frame.isAudioFrame(), True)
+    assert_equals(frame.getSize(), frame.getNbSamplesPerChannel() * frame.getBytesPerSample() * nbChannels)
+
+    frame.freeData()
+    assert_equals(frame.isDataAllocated(), False)
diff --git a/test/pyTest/testVideoFrame.py b/test/pyTest/testVideoFrame.py
@@ -0,0 +1,49 @@
+from nose.tools import *
+
+from pyAvTranscoder import avtranscoder as av
+
+
+@raises(MemoryError)
+def testInvalidVideoFrameAutoAllocated():
+    """
+    Try to create an invalid VideoFrame automatically allocated.
+    """
+    desc = av.VideoFrameDesc(1920, 1080, "toto")
+    av.VideoFrame(desc)
+
+
+@raises(MemoryError)
+def testInvalidVideoFrameManualAllocated():
+    """
+    Try to create an invalid VideoFrame manually allocated.
+    """
+    width = 1920
+    height = 1080
+    pixelFormat = "titi"
+    desc = av.VideoFrameDesc(width, height, pixelFormat)
+    frame = av.VideoFrame(desc, False)
+
+    assert_equals(frame.isDataAllocated(), False)
+    assert_equals(frame.getSize(), 0)
+    assert_equals(frame.getWidth(), width)
+    assert_equals(frame.getHeight(), height)
+    assert_equals(av.getPixelFormatName(frame.getPixelFormat()), "")
+
+    frame.allocateData()
+
+def testVideoFrame():
+    """
+    Check the size and the data buffer of a VideoFrame.
+    """
+    width = 1920
+    height = 1080
+    pixelFormat = "yuv420p"
+    desc = av.VideoFrameDesc(width, height, pixelFormat)
+    frame = av.VideoFrame(desc)
+
+    assert_equals(frame.isDataAllocated(), True)
+    assert_equals(frame.isVideoFrame(), True)
+    assert_equals(frame.getSize(), width * height * 1.5)
+
+    frame.freeData()
+    assert_equals(frame.isDataAllocated(), False)
