From f662b8fd0480692f81b5c857b7fb9f3f8f318302 Mon Sep 17 00:00:00 2001
From: termi-official <termi-official@users.noreply.github.com>
Date: Mon, 17 Mar 2025 15:56:42 +0100
Subject: [PATCH] Fix test

---
 src/HYPRE.jl     |  1 +
 src/precs.jl     | 16 +++++-----------
 test/runtests.jl |  8 ++++----
 3 files changed, 10 insertions(+), 15 deletions(-)

diff --git a/src/HYPRE.jl b/src/HYPRE.jl
index 993fbd7..7a6f4f9 100644
--- a/src/HYPRE.jl
+++ b/src/HYPRE.jl
@@ -3,6 +3,7 @@
 module HYPRE
 
 using MPI: MPI
+import LinearAlgebra
 
 export HYPREMatrix, HYPREVector
 
diff --git a/src/precs.jl b/src/precs.jl
index 8956ac7..bb6d0f8 100644
--- a/src/precs.jl
+++ b/src/precs.jl
@@ -1,5 +1,3 @@
-import LinearAlgebra
-
 struct BoomerAMGPrecWrapper{MatType}
     P::HYPRE.BoomerAMG
     A::MatType
@@ -19,17 +17,13 @@ struct BoomerAMGPrecBuilder{SFun, Tk}
 end
 
 # Syntactic sugar wth some defaults
-function BoomerAMGPrecBuilder(settings_fun! = (amg, A, p) -> nothing; MaxIter = 1, Tol = 0.0, kwargs...)
-    return construct_boomeramg_prec_builder(settings_fun!; MaxIter, Tol, kwargs)
-end
-
-# Helper to package kwargs
-function construct_boomeramg_prec_builder(settings_fun!; kwargs...)
+function BoomerAMGPrecBuilder(settings_fun! = (amg, A, p) -> nothing; kwargs...)
     return BoomerAMGPrecBuilder(settings_fun!, kwargs)
 end
 
 function (b::BoomerAMGPrecBuilder)(A, p)
-    amg = HYPRE.BoomerAMG(; b.kwargs)
-    settings_fun!(amg, A, p)
-    return (BoomerAMGPrecWrapper(amg, A), I)
+    amg = HYPRE.BoomerAMG(; b.kwargs...)
+    Internals.set_precond_defaults(amg)
+    b.settings_fun!(amg, A, p)
+    return (BoomerAMGPrecWrapper(amg, A), LinearAlgebra.I)
 end
diff --git a/test/runtests.jl b/test/runtests.jl
index 8c5ce75..1918f26 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -28,9 +28,9 @@ end
     x = zeros(100)
     # Solve
     tol = 1e-9
-    function set_debug_printlevel(amg, A, p)
-        HYPRE.HYPRE_BoomerAMGSetPrintLevel(amg, 3)
-    end
+    # function set_debug_printlevel(amg, A, p)
+    #     HYPRE.HYPRE_BoomerAMGSetPrintLevel(amg, 3)
+    # end
     bamg = HYPRE.BoomerAMGPrecBuilder(
         (amg, A, p) -> nothing;
         MaxIter = 1,
@@ -39,7 +39,7 @@ end
     prob = LinearProblem(A, b)
     solver = KrylovJL_CG(precs = bamg)
     x = solve(prob, solver, atol=1.0e-14)
-    @test x ≈ A \ b atol=tol
+    @test x ≈ A \ b atol=√tol
 end