Test: add CodedData unit tests

valnoel · valnoel · commit 9db1c51ec960 · 2020-02-14T12:50:39.000+01:00
diff --git a/test/pyTest/testCodedData.py b/test/pyTest/testCodedData.py
@@ -0,0 +1,35 @@
+from nose.tools import *
+
+from pyAvTranscoder import avtranscoder as av
+
+def testCodedDataConstructors():
+    """
+    Try to create a CodedData instances from different constructors.
+    """
+    dataSize = 1024
+    codedData = av.CodedData(dataSize)
+    assert_equals(dataSize, codedData.getSize())
+
+    codedDataCopy = av.CodedData(codedData)
+    assert_equals(dataSize, codedDataCopy.getSize())
+
+
+def testCodedDataManagement():
+    """
+    Try to resize and assign CodedData data.
+    """
+    dataSize = 1024
+    codedData = av.CodedData()
+    codedData.resize(dataSize)
+    assert_equals(dataSize, codedData.getSize())
+
+    newDataSize = 128
+    codedData.assign(newDataSize, 1)
+    assert_equals(newDataSize, codedData.getSize())
+    data = codedData.getData()
+    for i in range(0, newDataSize):
+        assert_equals('\x01', data[i])
+
+    newDataSize = 256
+    codedData.resize(newDataSize)
+    assert_equals(newDataSize, codedData.getSize())
