[ffmpeg-qsv] add av1e playback_seek test case

Signed-off-by: Wang Hangjie <[email protected]>
intel · Sep 26, 2024 · fabcc96 · fabcc96
1 parent fdbf05e
commit fabcc96
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/lib/parameters.py b/lib/parameters.py
@@ -950,6 +950,8 @@ def gen_vp9_seek_lp_parameters(spec):
 
 gen_avc_seek_parameters = gen_vp9_seek_parameters
 gen_avc_seek_lp_parameters = gen_vp9_seek_lp_parameters
+gen_av1_seek_parameters = gen_vp9_seek_parameters
+gen_av1_seek_lp_parameters = gen_vp9_seek_lp_parameters
 
 def gen_av1_cqp_variants(spec, strapi=False):
   for case, params in spec.items():

diff --git a/test/ffmpeg-qsv/encode/av1.py b/test/ffmpeg-qsv/encode/av1.py
@@ -144,6 +144,24 @@ def test_strapi(self, case, gop, bframes, tilecols, tilerows, bitrate, fps, qual
     self.init(spec, case, gop, bframes, tilecols, tilerows, bitrate, fps, quality, profile)
     self.encode()
 
+class seek(AV1EncoderTest):
+  def init(self, tspec, case, rcmode, bitrate, maxrate, fps, seek):
+    vars(self).update(tspec[case].copy())
+    vars(self).update(
+      case      = case,
+      bitrate   = bitrate,
+      maxrate   = maxrate,
+      minrate   = bitrate,
+      rcmode    = rcmode,
+      fps       = fps,
+      seek      = seek,
+    )
+
+  @slash.parametrize(*gen_av1_seek_parameters(spec))
+  def test(self, case, rcmode, bitrate, maxrate, fps, seek):
+    self.init(spec, case, rcmode, bitrate, maxrate, fps, seek)
+    self.encode()
+
 class cqp_lp(AV1EncoderLPTest):
   def init(self, tspec, case, gop, bframes, tilecols, tilerows,qp, quality, profile):
     vars(self).update(tspec[case].copy())
@@ -276,3 +294,20 @@ def test_strapi(self, case, gop, bframes, tilecols, tilerows, bitrate, fps, qual
     self.init(spec, case, gop, bframes, tilecols, tilerows, bitrate, fps, quality, profile)
     self.encode()
 
+class seek_lp(AV1EncoderLPTest):
+  def init(self, tspec, case, rcmode, bitrate, maxrate, fps, seek):
+    vars(self).update(tspec[case].copy())
+    vars(self).update(
+      case      = case,
+      bitrate   = bitrate,
+      maxrate   = maxrate,
+      minrate   = bitrate,
+      rcmode    = rcmode,
+      fps       = fps,
+      seek      = seek,
+    )
+
+  @slash.parametrize(*gen_av1_seek_lp_parameters(spec))
+  def test(self, case, rcmode, bitrate, maxrate, fps, seek):
+    self.init(spec, case, rcmode, bitrate, maxrate, fps, seek)
+    self.encode()