From 617698706ccff16b43091edb9cf94d2d3eda7c5f Mon Sep 17 00:00:00 2001
From: Jerome Jiang <jianj@google.com>
Date: Thu, 12 May 2022 21:05:24 -0400
Subject: Add aq mode 0 and 3 to active map test

Bug: webm:1762
Change-Id: Ia827f6686e8d0cdc09f3d07d07dacaa4fcd801ab
---
 test/active_map_test.cc | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'test/active_map_test.cc')

diff --git a/test/active_map_test.cc b/test/active_map_test.cc
index 7f41009e0..543ec0d35 100644
--- a/test/active_map_test.cc
+++ b/test/active_map_test.cc
@@ -19,7 +19,8 @@ namespace {
 
 class ActiveMapTest
     : public ::libvpx_test::EncoderTest,
-      public ::libvpx_test::CodecTestWith2Params<libvpx_test::TestMode, int> {
+      public ::libvpx_test::CodecTestWith3Params<libvpx_test::TestMode, int,
+                                                 int> {
  protected:
   static const int kWidth = 208;
   static const int kHeight = 144;
@@ -37,7 +38,7 @@ class ActiveMapTest
                                   ::libvpx_test::Encoder *encoder) {
     if (video->frame() == 0) {
       encoder->Control(VP8E_SET_CPUUSED, cpu_used_);
-      encoder->Control(VP9E_SET_AQ_MODE, 3);
+      encoder->Control(VP9E_SET_AQ_MODE, GET_PARAM(3));
     } else if (video->frame() == 3) {
       vpx_active_map_t map = vpx_active_map_t();
       /* clang-format off */
@@ -88,5 +89,5 @@ TEST_P(ActiveMapTest, Test) {
 
 VP9_INSTANTIATE_TEST_SUITE(ActiveMapTest,
                            ::testing::Values(::libvpx_test::kRealTime),
-                           ::testing::Range(0, 10));
+                           ::testing::Range(5, 10), ::testing::Values(0, 3));
 }  // namespace
-- 
cgit v1.2.3