From 410a8df64723d3d56a6d0d728e0d0a0e1da18c1f Mon Sep 17 00:00:00 2001
From: Karl Pierce <kmp5@vt.edu>
Date: Fri, 8 Dec 2023 15:58:11 -0500
Subject: [PATCH] Add more tests

---
 .../lib/UnallocatedArrays/test/runtests.jl    | 22 ++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/NDTensors/src/lib/UnallocatedArrays/test/runtests.jl b/NDTensors/src/lib/UnallocatedArrays/test/runtests.jl
index e9cdf0c107..4017775ba9 100644
--- a/NDTensors/src/lib/UnallocatedArrays/test/runtests.jl
+++ b/NDTensors/src/lib/UnallocatedArrays/test/runtests.jl
@@ -115,6 +115,21 @@ using .NDTensorsTestUtils: devices_list
     @test F2 isa UnallocatedFill
     @test F2[1, 1, 1] == elt(6.0)
     @test alloctype(F2) == alloctype(F)
+
+    #F2 .+= elt(2.0) ## This is broken
+    F2 = F2 .+ elt(2.0)
+    @test F2 isa UnallocatedFill
+    @test F2[1,1,1] == elt(8.0)
+    @test alloctype(F2) == alloctype(F)
+
+    F = UnallocatedFill(Fill(elt(2.0), (2,3)), dev(Matrix{elt}))
+    R = Z + F
+    @test R isa UnallocatedFill
+    @test alloctype(R) == alloctype(Z)
+
+    R = F + Z
+    @test R isa UnallocatedFill
+    @test alloctype(R) == alloctype(Z)
   end
 
   ## TODO make other kron tests
@@ -146,7 +161,12 @@ using .NDTensorsTestUtils: devices_list
   R = Z + Z
   @test_broken R isa UnallocatedZeros
   @test_broken alloctype(R) == alloctype(Z)
-  R = Z .+ eltype(Z)(2.0)
+  R = Z .+ elt(2.0)
   @test_broken R isa UnallocatedFill
+
+  F = UnallocatedFill(Fill(elt(2),(2,3)), dev(Matrix{elt}))
+  R = F + F
+  @test_broken R isa UnallocatedFill
+
 end
 end