go-astiav/codec_test.go
2022-02-13 19:08:18 +01:00

66 lines
1.9 KiB
Go

package astiav_test
import (
"testing"
"github.com/asticode/go-astiav"
"github.com/stretchr/testify/require"
)
func TestCodec(t *testing.T) {
c := astiav.FindDecoder(astiav.CodecIDMp3)
require.NotNil(t, c)
require.Nil(t, c.ChannelLayouts())
require.True(t, c.IsDecoder())
require.False(t, c.IsEncoder())
require.Nil(t, c.PixelFormats())
require.Equal(t, []astiav.SampleFormat{astiav.SampleFormatFltp, astiav.SampleFormatFlt}, c.SampleFormats())
require.Equal(t, "mp3float", c.Name())
require.Equal(t, "mp3float", c.String())
c = astiav.FindDecoderByName("aac")
require.NotNil(t, c)
require.Equal(t, []astiav.ChannelLayout{
astiav.ChannelLayoutMono,
astiav.ChannelLayoutStereo,
astiav.ChannelLayoutSurround,
astiav.ChannelLayout4Point0,
astiav.ChannelLayout5Point0Back,
astiav.ChannelLayout5Point1Back,
astiav.ChannelLayout7Point1WideBack,
}, c.ChannelLayouts())
require.True(t, c.IsDecoder())
require.False(t, c.IsEncoder())
require.Equal(t, []astiav.SampleFormat{astiav.SampleFormatFltp}, c.SampleFormats())
require.Equal(t, "aac", c.Name())
require.Equal(t, "aac", c.String())
c = astiav.FindEncoder(astiav.CodecIDH264)
require.NotNil(t, c)
require.False(t, c.IsDecoder())
require.True(t, c.IsEncoder())
require.Equal(t, []astiav.PixelFormat{
astiav.PixelFormatVideotoolbox,
astiav.PixelFormatNv12,
astiav.PixelFormatYuv420P,
}, c.PixelFormats())
require.Nil(t, c.SampleFormats())
require.Equal(t, "h264_videotoolbox", c.Name())
require.Equal(t, "h264_videotoolbox", c.String())
c = astiav.FindEncoderByName("mjpeg")
require.NotNil(t, c)
require.False(t, c.IsDecoder())
require.True(t, c.IsEncoder())
require.Equal(t, []astiav.PixelFormat{
astiav.PixelFormatYuvj420P,
astiav.PixelFormatYuvj422P,
astiav.PixelFormatYuvj444P,
}, c.PixelFormats())
require.Equal(t, "mjpeg", c.Name())
require.Equal(t, "mjpeg", c.String())
c = astiav.FindDecoderByName("invalid")
require.Nil(t, c)
}