Add tests for symmetric strength of connection

Author: ranjanan

src/gallery.jl

@@ -1 +1,3 @@
-poisson(n) = sparse(Tridiagonal(fill(-1, n-1), fill(2, n), fill(-1, n-1)))
+poisson(T, n) = sparse(Tridiagonal(fill(T(-1), n-1), 
+                        fill(T(2), n), fill(T(-1), n-1)))
+poisson(n) = poisson(Float64, n)

src/strength.jl

@@ -107,20 +107,26 @@ function strength_of_connection{T}(s::SymmetricStrength{T}, A, bsr_flag = false)
         for j in nzrange(A, i)
             row = A.rowval[j]
             val = A.nzval[j]
-            if val*val < eps_Aii * diags[row]
-                S.nzval[j] = 0
+            if row != i
+                if val*val < eps_Aii * diags[row]
+                    S.nzval[j] = 0
+                end
             end
         end
     end
 
     dropzeros!(S)
 
-    S.nzval .= abs.(S.nzval)
-    # for i = 1:size(S.nzval, 1)
+    # S.nzval .= abs.(S.nzval)
+    #for i = 1:size(S.nzval, 1)
     #     S.nzval[i] = abs(S.nzval[i])
-    # end
+    #end
 
     scale_cols_by_largest_entry!(S)
+    
+    for i = 1:size(S.nzval, 1)
+         S.nzval[i] = abs(S.nzval[i])
+    end
 
     S
 end

test/runtests.jl

@@ -4,6 +4,8 @@ using JLD
 using IterativeSolvers
 import AMG: V, coarse_solver, Pinv, Classical
 
+include("sa_tests.jl")
+
 @testset "AMG Tests" begin
 
 graph = load("test.jld")["G"]
@@ -234,4 +236,14 @@ end
 
 end
 
+# Smoothed Aggregation
+@testset "Smoothed Aggregation" begin
+
+@testset "Symmetric Strength of Connection" begin
+
+        test_symmetric_soc()
+end
+
+end
+
 end