fix col ids in DynamicSparseMartrix show function (#41)

* fix matrix display * mineure
atoptima · Apr 21, 2023 · ee779aa · ee779aa
1 parent 2b99d1f
commit ee779aa
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/matrix.jl b/src/matrix.jl
@@ -120,13 +120,15 @@ function Base.show(io::IO, matrix::DynamicSparseMatrix{K,L,T}) where {K,L,T}
     #col major iteration
     pma = matrix.colmajor.pcsc.pma
     semaphores = matrix.colmajor.pcsc.semaphores
-    j = 1
+    col_keys = matrix.colmajor.col_keys
+    tmp = 0
     for (index, elmt) in enumerate(pma.array)
         if index in semaphores
-            j += 1
+            tmp += 1
             print(io, "\n")
         else
             if !isnothing(elmt)
+                j = col_keys[tmp]
                 (i, value) = elmt
                 println(io, " [$(j), $(i)] = $(value) ")
             end

diff --git a/test/unit/display.jl b/test/unit/display.jl
@@ -7,8 +7,8 @@ end
 
 
 function test_matrix_display()
-    I = [1, 2, 3, 2, 6, 7, 1, 6, 8] #rows
-    J = [1, 1, 1, 2, 2, 2, 3, 3, 3] #columns
+    I = ['a', 'a', 'a', 'b', 'b', 'c', 'd', 'd', 'd'] #rows
+    J = ['x', 'x', 'x', 'y', 'y', 'y', 'z', 'z', 'z'] #columns
     V = [20, 30, 40, 20, 40, 50, 30, 50, 70] #value
     matrix = DynamicSparseArrays.dynamicsparse(I, J, V)
     @show matrix