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 4218358
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions tests/test_decode.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,3 +238,26 @@ def test_hardware_decode(self) -> None:
frame_count += 1

assert frame_count == video_stream.frames

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

output = av.open("output.mp4", "w")
stream = output.add_stream("libsvtav1")

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

0 comments on commit 4218358

Please sign in to comment.