Change assert to account for new outputs

microsoft · Feb 26, 2025 · 8c5f2fb · 8c5f2fb
1 parent 91a9c9d
commit 8c5f2fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/pp_api_test/test_feature_extraction.cc b/test/pp_api_test/test_feature_extraction.cc
@@ -65,7 +65,7 @@ TEST(ExtractorTest, TestPhi4AudioFeatureExtraction) {
   err = OrtxTensorResultGetAt(result.get(), 1, tensor.ToBeAssigned());
   ASSERT_EQ(err, kOrtxOK);
   err = OrtxGetTensorData(tensor.get(), reinterpret_cast<const void**>(&data), &shape, &num_dims);
-  ASSERT_EQ(num_dims, 1);
+  ASSERT_EQ(num_dims, 2);
   ASSERT_EQ(std::vector<int64_t>(shape, shape + num_dims), std::vector<int64_t>({3}));
   const float* actual_output = reinterpret_cast<const float*>(data);
   ASSERT_FLOAT_EQ(actual_output[0], 138.0f);
@@ -100,7 +100,7 @@ TEST(ExtractorTest, TestPhi4AudioFeatureExtraction8k) {
   err = OrtxTensorResultGetAt(result.get(), 1, tensor.ToBeAssigned());
   ASSERT_EQ(err, kOrtxOK);
   err = OrtxGetTensorData(tensor.get(), reinterpret_cast<const void**>(&data), &shape, &num_dims);
-  ASSERT_EQ(num_dims, 1);
+  ASSERT_EQ(num_dims, 2);
   ASSERT_EQ(std::vector<int64_t>(shape, shape + num_dims), std::vector<int64_t>({1}));
 }