add webm tolerance to the round trip test

Author: Dan-Flores

test/test_encoders.py

@@ -743,9 +743,10 @@ def test_round_trip(self, tmp_path, format, method):
         assert source_frames.shape == round_trip_frames.shape
         assert source_frames.dtype == round_trip_frames.dtype
 
+        atol = 3 if format == "webm" else 2
         for s_frame, rt_frame in zip(source_frames, round_trip_frames):
             assert psnr(s_frame, rt_frame) > 30
-            torch.testing.assert_close(s_frame, rt_frame, atol=2, rtol=0)
+            torch.testing.assert_close(s_frame, rt_frame, atol=atol, rtol=0)
 
     @pytest.mark.parametrize(
         "format",
@@ -858,13 +859,12 @@ def test_video_encoder_against_ffmpeg_cli(self, tmp_path, format, pixel_format):
         # There may be additional subtle differences in the encoder.
         percentage = 94 if ffmpeg_version == 6 or format == "avi" else 99
 
-        atol = 3 if format == "webm" else 2
         # Check that PSNR between both encoded versions is high
         for ff_frame, enc_frame in zip(ffmpeg_frames, encoder_frames):
             res = psnr(ff_frame, enc_frame)
             assert res > 30
             assert_tensor_close_on_at_least(
-                ff_frame, enc_frame, percentage=percentage, atol=atol
+                ff_frame, enc_frame, percentage=percentage, atol=2
             )
 
     def test_to_file_like_custom_file_object(self):