1 files changed, 8 insertions, 6 deletions
diff --git a/nn/runtime/test/TestOpenmpSettings.cpp b/nn/runtime/test/TestOpenmpSettings.cpp
index 549473b46..59a794224 100644
--- a/nn/runtime/test/TestOpenmpSettings.cpp
+++ b/nn/runtime/test/TestOpenmpSettings.cpp
@@ -38,26 +38,28 @@ protected:
         ASSERT_EQ(blocktimeRestored, kOpenmpDefaultBlockTime);
     }
     static const int kOpenmpDefaultBlockTime;
+    static const int kPreferredBlockTime;
 };
 
 const int OpenmpSettingsTest::kOpenmpDefaultBlockTime = 200;
+const int OpenmpSettingsTest::kPreferredBlockTime = 20;
 
 using ::android::nn::ScopedOpenmpSettings;
 
-TEST_F(OpenmpSettingsTest, Test1) {
+TEST_F(OpenmpSettingsTest, TestkPreferredBlockTime) {
     ScopedOpenmpSettings s;
     const int blocktimeSet = kmp_get_blocktime();
-    ASSERT_EQ(blocktimeSet, 1);
+    ASSERT_EQ(blocktimeSet, kPreferredBlockTime);
 }
 
 TEST_F(OpenmpSettingsTest, Test2) {
     ScopedOpenmpSettings s1;
     const int blocktimeSet1 = kmp_get_blocktime();
-    ASSERT_EQ(blocktimeSet1, 1);
+    ASSERT_EQ(blocktimeSet1, kPreferredBlockTime);
 
     ScopedOpenmpSettings s2;
     const int blocktimeSet2 = kmp_get_blocktime();
-    ASSERT_EQ(blocktimeSet2, 1);
+    ASSERT_EQ(blocktimeSet2, kPreferredBlockTime);
 }
 
 TEST_F(OpenmpSettingsTest, TestThreaded) {
@@ -74,12 +76,12 @@ TEST_F(OpenmpSettingsTest, TestThreaded) {
             ScopedOpenmpSettings s;
 
             const int blocktimeSet2 = kmp_get_blocktime();
-            ASSERT_EQ(blocktimeSet2, 1);
+            ASSERT_EQ(blocktimeSet2, kPreferredBlockTime);
 
             usleep(sleepFor);
 
             const int blocktimeSet3 = kmp_get_blocktime();
-            ASSERT_EQ(blocktimeSet3, 1);
+            ASSERT_EQ(blocktimeSet3, kPreferredBlockTime);
         }));
     }
     std::for_each(threads.begin(), threads.end(), [](std::thread& t) {