use isapprox, etc

Author: Michael Abbott

test/batchedmul.jl

@@ -23,29 +23,29 @@ function bmm_adjtest(a,b; adjA = false, adjB = false)
 end
 
 
-@testset "Batched Matrix Multiplication" for TB in [Float64, Float32]
+@testset "Batched Matrices: Float64 * $TB" for TB in [Float64, Float32]
 
     A = randn(7,5,3)
     B = randn(TB, 5,7,3)
     C = randn(7,6,3)
 
     @test batched_mul(A, B) == bmm_test(A, B)
     @test batched_mul(batched_transpose(A), batched_transpose(B)) == bmm_test(A, B; transA = true, transB = true)
-    @test batched_mul(batched_transpose(A), C) == bmm_test(A, C; transA = true)
-    @test batched_mul(A, batched_transpose(A)) == bmm_test(A, A; transB = true)
+    @test batched_mul(batched_transpose(A), C) ≈ bmm_test(A, C; transA = true)
+    @test batched_mul(A, batched_transpose(A)) ≈ bmm_test(A, A; transB = true)
 
 
     cA = randn(Complex{Float64}, 7,5,3)
     cB = randn(Complex{TB}, 5,7,3)
     cC = randn(Complex{Float64}, 7,6,3)
 
     @test batched_mul(cA, cB) == bmm_adjtest(cA, cB)
-    @test batched_mul(batched_adjoint(cA), batched_adjoint(cB)) == bmm_adjtest(cA, cB; adjA = true, adjB = true)
-    @test batched_mul(batched_adjoint(cA), cC) == bmm_adjtest(cA, cC; adjA = true)
-    @test batched_mul(cA, batched_adjoint(cA)) == bmm_adjtest(cA, cA; adjB = true)
+    @test batched_mul(batched_adjoint(cA), batched_adjoint(cB)) ≈ bmm_adjtest(cA, cB; adjA = true, adjB = true)
+    @test batched_mul(batched_adjoint(cA), cC) ≈ bmm_adjtest(cA, cC; adjA = true)
+    @test batched_mul(cA, batched_adjoint(cA)) ≈ bmm_adjtest(cA, cA; adjB = true)
 
-    @test batched_transpose(batched_transpose(A)) == A
-    @test batched_adjoint(batched_adjoint(cA)) == cA
+    @test batched_transpose(batched_transpose(A)) === A
+    @test batched_adjoint(batched_adjoint(cA)) === cA
 
     TBi = TB==Float64 ? Int64 : Int32
     iA = rand(1:99, 7,5,3)