Fix tests

ITensor · May 8, 2024 · 87a60c8 · 87a60c8
1 parent 66e1dcc
commit 87a60c8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/test/test_cudiag.jl b/test/test_cudiag.jl
@@ -1,10 +1,10 @@
-using ITensors,
-  ITensors.NDTensors,
-  ITensorGPU,
-  LinearAlgebra, # For tr()
-  Combinatorics, # For permutations()
-  CUDA,
-  Test
+using ITensors
+using ITensors.NDTensors: NDTensors
+using ITensorGPU
+using LinearAlgebra: tr
+using Combinatorics: permutations
+using CUDA
+using Test: @test, @test_broken, @testset
 
 @testset "cuITensor $T Contractions" for T in (Float64, ComplexF64)
   mi, mj, mk, ml, ma = 2, 3, 4, 5, 6, 7

diff --git a/test/test_cuitensor.jl b/test/test_cuitensor.jl
@@ -60,7 +60,7 @@ using ITensors,
   @testset "Test CuVector(cuITensor)" begin
     v = CuVector(ones(SType, dim(a)))
     A = cuITensor(v, a)
-    @allowscalar begin
+    CUDA.@allowscalar begin
       @test v == CuVector(A)
     end
   end