From 3db02cae380917607725a23abaa3461d4bcfb0f6 Mon Sep 17 00:00:00 2001
From: Harsh Chauhan <harsh2005.hc@gmail.com>
Date: Sun, 29 Mar 2026 20:07:10 +0530
Subject: [PATCH] test: add ReduceMean kFirst axis regression test

---
 tmva/sofie/test/TestCustomModelsFromONNX.cxx     |  15 +++++++++++++++
 .../test/input_models/ReduceMean_kFirst.onnx     | Bin 0 -> 117 bytes
 2 files changed, 15 insertions(+)
 create mode 100644 tmva/sofie/test/input_models/ReduceMean_kFirst.onnx

diff --git a/tmva/sofie/test/TestCustomModelsFromONNX.cxx b/tmva/sofie/test/TestCustomModelsFromONNX.cxx
index 94993f601a3c4..07bc0d7480d87 100644
--- a/tmva/sofie/test/TestCustomModelsFromONNX.cxx
+++ b/tmva/sofie/test/TestCustomModelsFromONNX.cxx
@@ -918,6 +918,21 @@ TEST(ONNX, Pow_broadcast){
 
 }
 
+TEST(ONNX, ReduceMean_kFirst)
+{
+   // ReduceMean over axis=0 (kFirst path) on a [3,4] tensor.
+   std::vector<float> input(12);
+   std::iota(input.begin(), input.end(), 0.0f);
+   std::vector<float> correct_output = {4, 5, 6, 7};
+
+   ASSERT_INCLUDE_AND_RUN(std::vector<float>, "ReduceMean_kFirst", input);
+
+   EXPECT_EQ(output.size(), correct_output.size());
+   for (size_t i = 0; i < output.size(); ++i) {
+      EXPECT_LE(std::abs(output[i] - correct_output[i]), DEFAULT_TOLERANCE);
+   }
+}
+
    TEST(ONNX, ReduceProd){
    constexpr float TOLERANCE = DEFAULT_TOLERANCE;
 
diff --git a/tmva/sofie/test/input_models/ReduceMean_kFirst.onnx b/tmva/sofie/test/input_models/ReduceMean_kFirst.onnx
new file mode 100644
index 0000000000000000000000000000000000000000..7d5e7bad145f69eed0f9ccb98e13e65bc4d00262
GIT binary patch
literal 117
zcmd<!waVr);9`srVvJPc3QA2WO-}VqP0Z8c=3+^#NG*0?Sis1x#m~i&otj#Zl9^j9
z0Tf^o5=5ws&vwf!DlUl<1{uP~#lyiU#KFbH!3@MKN&H-lkwRQt96&)9kbn~l7lQyV
E05*^o$^ZZW

literal 0
HcmV?d00001