From d74e8d022cc268251b56eca445b1abb65f145556 Mon Sep 17 00:00:00 2001 From: Scott Schneider Date: Wed, 15 Oct 2025 12:43:39 -0700 Subject: [PATCH 1/2] Mark slow tests --- test/test_ops.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/test_ops.py b/test/test_ops.py index fddd4043c..585d62f15 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -1384,6 +1384,7 @@ def decode(self, file_path) -> torch.Tensor: frames, *_ = get_frames_in_range(decoder, start=0, stop=60) return frames + @pytest.mark.slow @pytest.mark.parametrize("format", ("mov", "mp4", "mkv", "webm")) def test_video_encoder_round_trip(self, tmp_path, format): # Test that decode(encode(decode(asset))) == decode(asset) @@ -1422,6 +1423,7 @@ def test_video_encoder_round_trip(self, tmp_path, format): assert psnr(s_frame, rt_frame) > 30 assert_close(s_frame, rt_frame, atol=atol, rtol=0) + @pytest.mark.slow @pytest.mark.skipif(in_fbcode(), reason="ffmpeg CLI not available") @pytest.mark.parametrize( "format", ("mov", "mp4", "avi", "mkv", "webm", "flv", "gif") From ee000a8e976d3de9aa97327cdcd3276e8dcc92ce Mon Sep 17 00:00:00 2001 From: Scott Schneider Date: Thu, 16 Oct 2025 06:01:56 -0700 Subject: [PATCH 2/2] Just mark webm --- test/test_ops.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/test/test_ops.py b/test/test_ops.py index 585d62f15..0c1d90cfc 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -1384,8 +1384,9 @@ def decode(self, file_path) -> torch.Tensor: frames, *_ = get_frames_in_range(decoder, start=0, stop=60) return frames - @pytest.mark.slow - @pytest.mark.parametrize("format", ("mov", "mp4", "mkv", "webm")) + @pytest.mark.parametrize( + "format", ("mov", "mp4", "mkv", pytest.param("webm", marks=pytest.mark.slow)) + ) def test_video_encoder_round_trip(self, tmp_path, format): # Test that decode(encode(decode(asset))) == decode(asset) ffmpeg_version = get_ffmpeg_major_version() @@ -1423,10 +1424,18 @@ def test_video_encoder_round_trip(self, tmp_path, format): assert psnr(s_frame, rt_frame) > 30 assert_close(s_frame, rt_frame, atol=atol, rtol=0) - @pytest.mark.slow @pytest.mark.skipif(in_fbcode(), reason="ffmpeg CLI not available") @pytest.mark.parametrize( - "format", ("mov", "mp4", "avi", "mkv", "webm", "flv", "gif") + "format", + ( + "mov", + "mp4", + "avi", + "mkv", + "flv", + "gif", + pytest.param("webm", marks=pytest.mark.slow), + ), ) def test_video_encoder_against_ffmpeg_cli(self, tmp_path, format): ffmpeg_version = get_ffmpeg_major_version()