Skip to content

Commit

Permalink
test svtav1
Browse files Browse the repository at this point in the history
  • Loading branch information
JoeSchiff committed Dec 29, 2024
1 parent 892c999 commit ee823cb
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions tests/test_encode.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,30 @@ def test_encoding_with_unicode_filename(self) -> None:
with av.open(path) as input:
assert_rgb_rotate(self, input)

def test_libsvtav1(self) -> None:
if not "libsvtav1" in av.codecs_available:
pytest.skip()

with av.open(self.sandboxed("output.mp4"), "w") as output:
stream = output.add_stream("libsvtav1", 24)
assert isinstance(stream, VideoStream)

for i in range(24):
frame = av.VideoFrame(200, 100, "rgb24")
frame.pts = i * 2000
frame.time_base = Fraction(1, 48000)
for packet in stream.encode(frame):
assert packet.time_base == Fraction(1, 24)
output.mux(packet)

for packet in stream.encode(None):
assert packet.time_base == Fraction(1, 24)
output.mux(packet)

assert output.streams[0].codec.name == "libsvtav1"
assert output.streams[0].codec.is_encoder is True
assert output.streams[0].frames == 24


class TestBasicAudioEncoding(TestCase):
def test_default_options(self) -> None:
Expand Down

0 comments on commit ee823cb

Please sign in to comment.