Merge pull request #253 from tweska/knobel

bug fix for MMWrite (invalid output on array format)
GraphBLAS · Jul 1, 2024 · 64a629f · 64a629f
2 parents 1beaf9d + 275a986
commit 64a629f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/data/comments_full.mtx b/data/comments_full.mtx
@@ -2,7 +2,7 @@
 %%GraphBLAS type double
 % comments for full.mtx
 % this file was created by test_MMRead.c
-3 3 9
+3 3
 .646
 .709
 .754

diff --git a/src/utility/LAGraph_MMWrite.c b/src/utility/LAGraph_MMWrite.c
@@ -459,8 +459,18 @@ int LAGraph_MMWrite
         nvals_to_print = nself_edges + (nvals - nself_edges) / 2 ;
     }
 
-    FPRINTF (f, "%" PRIu64 " %" PRIu64 " %" PRIu64 "\n",
-        nrows, ncols, nvals_to_print) ;
+    if (MM_fmt == MM_array)
+    {
+        // write `nrows ncols` if the array format is used
+        FPRINTF (f, "%" PRIu64 " %" PRIu64 "\n",
+            nrows, ncols) ;
+    }
+    else
+    {
+        // otherwise write `nrows ncols nvals` for the coordinate format
+        FPRINTF (f, "%" PRIu64 " %" PRIu64 " %" PRIu64 "\n",
+            nrows, ncols, nvals_to_print) ;
+    }
 
     if (nvals_to_print == 0)
     {
-Original file line number
+Diff line change
@@ Expand Up / @@ -2,7 +2,7 @@ @@
     %%GraphBLAS type double
     % comments for full.mtx
     % this file was created by test_MMRead.c
-3 9
+3
     .646
     .709
     .754
@@ Expand Down @@