Added basic unit tests for symtridiagonal.

Author: loiseaujc

test/linalg/test_linalg_specialmatrices.fypp

@@ -20,10 +20,18 @@ contains
 
         testsuite = [ &
             new_unittest('tridiagonal', test_tridiagonal), &
-            new_unittest('tridiagonal error handling', test_tridiagonal_error_handling) &
+            new_unittest('tridiagonal error handling', test_tridiagonal_error_handling), &
+            new_unittest('symtridiagonal', test_symtridiagonal), &
+            new_unittest('symtridiagonal error handling', test_symtridiagonal_error_handling) &
         ]
     end subroutine
 
+    !----------------------------------------
+    !-----                              -----
+    !-----     TRIDIAGONAL MATRICES     -----
+    !-----                              -----
+    !----------------------------------------
+
     subroutine test_tridiagonal(error)
         !> Error handling
         type(error_type), allocatable, intent(out) :: error
@@ -94,6 +102,82 @@ contains
         #:endfor       
     end subroutine
 
+    !--------------------------------------------------
+    !-----                                        -----
+    !-----     SYMMETRIC TRIDIAGONAL MATRICES     -----
+    !-----                                        -----
+    !--------------------------------------------------
+
+    subroutine test_symtridiagonal(error)
+        !> Error handling
+        type(error_type), allocatable, intent(out) :: error
+        #:for k1, t1, s1 in (KINDS_TYPES)
+        block
+            integer, parameter :: wp = ${k1}$
+            integer, parameter :: n = 5
+            type(symtridiagonal_${s1}$_type) :: A
+            ${t1}$, allocatable :: Amat(:,:), dv(:), ev(:)
+            ${t1}$, allocatable :: x(:)
+            ${t1}$, allocatable :: y1(:), y2(:)
+
+            ! Initialize matrix.
+            allocate(ev(n-1), dv(n))
+            call random_number(dv) ; call random_number(ev)
+            A = symtridiagonal(dv, ev) ; Amat = dense(A)
+
+            ! Random vectors.
+            allocate(x(n), source = 0.0_wp)   ; call random_number(x)
+            allocate(y1(n), source = 0.0_wp)  ; allocate(y2(n), source=0.0_wp)
+
+            ! Test y = A @ x
+            y1 = matmul(Amat, x) ; call spmv(A, x, y2)
+            call check(error, all_close(y1, y2), .true.)
+            if (allocated(error)) return
+
+            ! Test y = A.T @ x
+            y1 = 0.0_wp ; y2 = 0.0_wp
+            y1 = matmul(transpose(Amat), x) ; call spmv(A, x, y2, op="T")
+            call check(error, all_close(y1, y2), .true.)
+            if (allocated(error)) return
+
+            #:if t1.startswith('complex') 
+            ! Test y = A.H @ x
+            y1 = 0.0_wp ; y2 = 0.0_wp
+            y1 = matmul(hermitian(Amat), x) ; call spmv(A, x, y2, op="H")
+            call check(error, all_close(y1, y2), .true.)
+            if (allocated(error)) return
+            #:endif
+        end block
+        #:endfor
+    end subroutine
+
+    subroutine test_symtridiagonal_error_handling(error)
+        !> Error handling
+        type(error_type), allocatable, intent(out) :: error
+        #:for k1, t1, s1 in (KINDS_TYPES)
+        block
+            integer, parameter :: wp = ${k1}$
+            integer, parameter :: n = 5
+            type(symtridiagonal_${s1}$_type) :: A
+            ${t1}$, allocatable :: dv(:), ev(:)
+            type(linalg_state_type) :: state
+            integer :: i
+
+            !> Test constructor from arrays.
+            ev = [(1.0_wp, i = 1, n-2)]
+            dv = [(2.0_wp, i = 1, n)]
+            A = symtridiagonal(dv, ev, state)
+            call check(error, state%ok(), .false.)
+            if (allocated(error)) return
+
+            !> Test contructor from constants.
+            A = symtridiagonal(dv(1), ev(1), -n, state)
+            call check(error, state%ok(), .false.)
+            if (allocated(error)) return
+        end block
+        #:endfor       
+    end subroutine
+
 end module