From 3a2c2b454472b7a72f1b85621ac88d9359ea45ea Mon Sep 17 00:00:00 2001
From: Raphael Steiner <raphael.steiner@huawei.com>
Date: Thu, 20 Nov 2025 13:36:27 +0100
Subject: [PATCH] Increased Sarkar Test Coverage

---
 tests/coarser.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/tests/coarser.cpp b/tests/coarser.cpp
index a58f3706..be189ae8 100644
--- a/tests/coarser.cpp
+++ b/tests/coarser.cpp
@@ -703,6 +703,20 @@ BOOST_AUTO_TEST_CASE(SarkarML_test) {
     test_coarser_same_graph<graph_t>(coarser);
 }
 
+BOOST_AUTO_TEST_CASE(SarkarMLBufferMerge_test) {
+    using graph_t = computational_dag_edge_idx_vector_impl_def_t;
+    // using graph_t = computational_dag_vector_impl_def_t;
+
+    SarkarParams::MulParameters<v_workw_t<graph_t>> params;
+    params.commCostVec = {1, 2, 10, 50, 100};
+    params.buffer_merge_mode = SarkarParams::BufferMergeMode::FULL;
+
+    SarkarMul<graph_t, graph_t> coarser;
+    coarser.setParameters(params);
+
+    test_coarser_same_graph<graph_t>(coarser);
+}
+
 BOOST_AUTO_TEST_CASE(SquashAML_test) {
     using graph_t = computational_dag_edge_idx_vector_impl_def_t;
     // using graph_t = computational_dag_vector_impl_def_t;