[test] Fix the rank compare test of SNPE filter
authorYongjoo Ahn <yongjoo1.ahn@samsung.com>
Mon, 11 Sep 2023 08:02:48 +0000 (17:02 +0900)
committerjaeyun-jung <39614140+jaeyun-jung@users.noreply.github.com>
Tue, 12 Sep 2023 02:44:31 +0000 (11:44 +0900)
- Deal with the recent rank update in nnstreamer.

Signed-off-by: Yongjoo Ahn <yongjoo1.ahn@samsung.com>
tests/nnstreamer_filter_snpe/unittest_filter_snpe.cc

index 2ae0274..10e35c0 100644 (file)
@@ -80,18 +80,17 @@ TEST (nnstreamerFilterSnpe, getModelInfo00)
   EXPECT_EQ (ret, 0);
 
   EXPECT_EQ (in_info.num_tensors, 1U);
-  EXPECT_EQ (in_info.info[0].dimension[0], 1U);
-  EXPECT_EQ (in_info.info[0].dimension[1], 1U);
-  EXPECT_EQ (in_info.info[0].dimension[2], 1U);
-  EXPECT_EQ (in_info.info[0].dimension[3], 1U);
   EXPECT_EQ (in_info.info[0].type, _NNS_FLOAT32);
+  EXPECT_EQ (in_info.info[0].dimension[0], 1U);
+  for (int i = 1; i < NNS_TENSOR_RANK_LIMIT; ++i)
+    EXPECT_EQ (in_info.info[0].dimension[i], 0U);
 
   EXPECT_EQ (out_info.num_tensors, 1U);
-  EXPECT_EQ (out_info.info[0].dimension[0], 1U);
-  EXPECT_EQ (out_info.info[0].dimension[1], 1U);
-  EXPECT_EQ (out_info.info[0].dimension[2], 1U);
-  EXPECT_EQ (out_info.info[0].dimension[3], 1U);
   EXPECT_EQ (out_info.info[0].type, _NNS_FLOAT32);
+  EXPECT_EQ (out_info.info[0].dimension[0], 1U);
+  for (int i = 1; i < NNS_TENSOR_RANK_LIMIT; ++i)
+    EXPECT_EQ (out_info.info[0].dimension[i], 0U);
+
 
   sp->close (&prop, &data);
   gst_tensors_info_free (&in_info);