increase code cov

sisl · Dec 16, 2024 · e3d38aa · e3d38aa
1 parent a907982
commit e3d38aa
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/src/constructors/epca4.jl b/src/constructors/epca4.jl
@@ -1,6 +1,6 @@
 struct EPCA4{
     FT1<:Function,
-    FT2<:Function,
+    FT2<:Union{Function, FunctionWrapper},
     MT<:AbstractMatrix{<:Real},
     OT<:Options
 } <: EPCA
@@ -29,7 +29,7 @@ function EPCA(
     indim::Integer,
     outdim::Integer,
     Bg::Function,
-    g::Function,
+    g::Union{Function, FunctionWrapper},
     ::Val{(:Bg, :g)};
     options::Options = Options()
 )

diff --git a/test/utils.jl b/test/utils.jl
@@ -216,5 +216,21 @@ function run_EPCA_tests(
             X;
             rtol=1e-7
         )
+
+        test_equivalence(
+            "Bg, G",
+            M1,
+            A1,
+            EPCA(
+                indim,
+                outdim,
+                Bg,
+                G,
+                Val((:Bg, :G));
+                options = options
+            ),
+            X;
+            rtol=1e-7
+        )
     end
 end