def test_to_image_rgb24(self):
sizes = [
(318, 238),
(320, 240),
(500, 500),
]
for width, height in sizes:
frame = VideoFrame(width, height, format='rgb24')
# fill video frame data
for plane in frame.planes:
ba = bytearray(plane.buffer_size)
pos = 0
for row in range(height):
for i in range(plane.line_size):
ba[pos] = i % 256
pos += 1
plane.update(ba)
# construct expected image data
expected = bytearray(height * width * 3)
pos = 0
for row in range(height):
for i in range(width * 3):
expected[pos] = i % 256
pos += 1
img = frame.to_image()
self.assertEqual(img.size, (width, height))
self.assertEqual(img.tobytes(), expected)
def test_to_image_with_dimensions(self):
frame = VideoFrame(640, 480, format="rgb24")
img = frame.to_image(width=320, height=240)
self.assertEqual(img.size, (320, 240))
