Add a scalar diag test

ITensor · Aug 14, 2024 · 03ad5b4 · 03ad5b4
1 parent d460a81
commit 03ad5b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/NDTensors/test/test_diag.jl b/NDTensors/test/test_diag.jl
@@ -63,7 +63,7 @@ using .NDTensorsTestUtils: devices_list, is_supported_eltype
       @test x == dev(diagm(0 => vr))
       @test x == D
     end
-
+    @test sqrt(contract(D, (-1,-2), D, (-1,-2))[]) ≈ norm(D)
     # This if statement corresponds to the reported bug:
     # https://github.com/JuliaGPU/Metal.jl/issues/364
     if !(dev == NDTensors.mtl && elt === ComplexF32)