from ffmpegio.utils import prod

from ffmpegio import plugins

def test_rawdata_bytes():

hook = plugins.get_hook()

dtype = "|u1"

shape = (2, 2, 3)

b = b"\0" * prod(shape)

data = hook.bytes_to_video(b=b, dtype=dtype, shape=shape, squeeze=False)

assert data["buffer"] == b

assert data["dtype"] == dtype

assert data["shape"][1:] == shape

assert hook.video_info(obj=data) == (shape, dtype)

assert hook.video_bytes(obj=data) == b

data = hook.bytes_to_video(b=b, dtype=dtype, shape=shape, squeeze=True)

assert data["shape"] == shape

dtype = "<f4"

shape = (2,)

b = b"\0" * (1024 * prod(shape))

data = hook.bytes_to_audio(b=b, dtype=dtype, shape=shape, squeeze=False)

assert data["buffer"] == b

assert data["dtype"] == dtype

assert data["shape"][1:] == shape

assert hook.audio_info(obj=data) == (shape, dtype)

assert hook.audio_bytes(obj=data) == b

def test_use():

import numpy as np

plugins.use("read_numpy")

dtype = "|u1"

shape = (2, 2, 3)

b = b"\0" * prod(shape)

hook = plugins.get_hook()

data = hook.bytes_to_video(b=b, dtype=dtype, shape=shape, squeeze=False)

assert isinstance(data, np.ndarray)

plugins.use("read_bytes")

hook = plugins.get_hook()

data = hook.bytes_to_video(b=b, dtype=dtype, shape=shape, squeeze=False)

assert isinstance(data, dict)

if __name__ == "__main__":

print(plugins.pm.get_plugins())