From 24e5f031aeae3468f6d9cb726171a31e38847554 Mon Sep 17 00:00:00 2001
From: MohamedLaghdafHABIBOULLAH <mohamed-laghdaf-2.habiboullah@polymtl.ca>
Date: Mon, 4 Nov 2024 14:01:11 -0500
Subject: [PATCH] add unit test for DiagonalBFGS

---
 test/test_diag.jl | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/test/test_diag.jl b/test/test_diag.jl
index 9349039..3c02886 100644
--- a/test/test_diag.jl
+++ b/test/test_diag.jl
@@ -120,6 +120,10 @@ end
   mul!(u, C, v)
   @test (@allocated mul!(u, C, v)) == 0
   @test (@wrappedallocs push!(C, u, v)) == 0
+  D = DiagonalBFGS(d)
+  mul!(u, D, v)
+  @test (@allocated mul!(u, D, v)) == 0
+  @test (@wrappedallocs push!(D, u, v)) == 0
 end
 
 @testset "reset" begin