set reusable params for parametrized round trip test

Author: Daniel Flores

test/test_ops.py

@@ -1406,19 +1406,20 @@ def test_video_encoder_round_trip(self, tmp_path, format, method):
         asset = TEST_SRC_2_720P
         source_frames = self.decode(file_path=str(asset.path)).data
 
-        frame_rate = 30  # Frame rate is fixed with num frames decoded
+        params = dict(
+            frame_rate=30, crf=0
+        )  # Frame rate is fixed with num frames decoded
         if method == "to_file":
             encoded_path = str(tmp_path / f"encoder_output.{format}")
             encode_video_to_file(
                 frames=source_frames,
-                frame_rate=frame_rate,
                 filename=encoded_path,
-                crf=0,
+                **params,
             )
             round_trip_frames = self.decode(file_path=encoded_path).data
         else:  # to_tensor
             encoded_tensor = encode_video_to_tensor(
-                source_frames, frame_rate, format, crf=0
+                source_frames, format=format, **params
             )
             round_trip_frames = self.decode(tensor=encoded_tensor).data
 
@@ -1454,16 +1455,15 @@ def test_video_encoder_against_ffmpeg_cli(self, tmp_path, format):
                 )
         asset = TEST_SRC_2_720P
         source_frames = self.decode(str(asset.path)).data
-        frame_rate = 30
 
         # Encode with FFmpeg CLI
         temp_raw_path = str(tmp_path / "temp_input.raw")
         with open(temp_raw_path, "wb") as f:
             f.write(source_frames.permute(0, 2, 3, 1).cpu().numpy().tobytes())
 
         ffmpeg_encoded_path = str(tmp_path / f"ffmpeg_output.{format}")
+        frame_rate = 30
         crf = 0
-        quality_params = ["-crf", str(crf)]
         # Some codecs (ex. MPEG4) do not support CRF.
         # Flags not supported by the selected codec will be ignored.
         ffmpeg_cmd = [
@@ -1479,7 +1479,8 @@ def test_video_encoder_against_ffmpeg_cli(self, tmp_path, format):
             str(frame_rate),
             "-i",
             temp_raw_path,
-            *quality_params,
+            "-crf",
+            str(crf),
             ffmpeg_encoded_path,
         ]
         subprocess.run(ffmpeg_cmd, check=True)