add Symtridiagonal

JuliaDiff · Jun 21, 2019 · 3f73c05 · 3f73c05
1 parent 8184685
commit 3f73c05
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/coloring/high_level.jl b/src/coloring/high_level.jl
@@ -35,6 +35,6 @@ function matrix_colors(A::Bidiagonal)
     repeat(1:2,div(size(A,2),2)+1)[1:size(A,2)]
 end
 
-function matrix_colors(A::Tridiagonal)
+function matrix_colors(A::Union{Tridiagonal,SymTridiagonal})
     repeat(1:3,div(size(A,2),3)+1)[1:size(A,2)]
 end
diff --git a/test/test_specialmatrices.jl b/test/test_specialmatrices.jl
@@ -9,6 +9,7 @@ diagonal=Diagonal(dense)
 bidiagonalU=Bidiagonal(dense,:U)
 bidiagonalL=Bidiagonal(dense,:L)
 tridiagonal=Tridiagonal(dense)
+symtridiagonal=SymTridiagonal(dense)
 
 @test matrix_colors(dense)==1:n
 @test matrix_colors(uptri)==1:n
@@ -17,4 +18,5 @@ tridiagonal=Tridiagonal(dense)
 @test matrix_colors(diagonal)==fill(1,n)
 @test matrix_colors(bidiagonalU)==[1,2,1,2,1,2,1,2,1,2]
 @test matrix_colors(bidiagonalL)==[1,2,1,2,1,2,1,2,1,2]
-@test matrix_colors(tridiagonal)==[1,2,3,1,2,3,1,2,3,1]
+@test matrix_colors(tridiagonal)==[1,2,3,1,2,3,1,2,3,1]
+@test matrix_colors(symtridiagonal)==[1,2,3,1,2,3,1,2,3,1]