Include dim in mpscopy testsets

JuliaGPU · Sep 30, 2024 · 3a4ceb4 · 3a4ceb4
1 parent b999285
commit 3a4ceb4
Showing 1 changed file with 10 additions and 12 deletions.
diff --git a/test/mps/copy.jl b/test/mps/copy.jl
@@ -31,21 +31,19 @@ end
 @testset "MPSMatrixCopy" begin
     Ts = collect(values(MPS.jl_mps_to_typ))
     Ts = Ts[.!(Ts .<: IGNORE_UNION)]
-    @testset "$T" for T in Ts
-        for dim in ((16,16), (10,500), (500,10), (256,512))
-            srcMat = MtlArray(rand(T, dim))
+    @testset "$T: $dim" for T in Ts, dim in ((16,16), (10,500), (500,10), (256,512))
+        srcMat = MtlArray(rand(T, dim))
 
-            dstMat = copytest(srcMat, false, false)
-            @test dstMat == srcMat
+        dstMat = copytest(srcMat, false, false)
+        @test dstMat == srcMat
 
-            dstMat = copytest(srcMat, true, false)
-            @test dstMat == srcMat'
+        dstMat = copytest(srcMat, true, false)
+        @test dstMat == srcMat'
 
-            dstMat = copytest(srcMat, false, true)
-            @test dstMat == srcMat'
+        dstMat = copytest(srcMat, false, true)
+        @test dstMat == srcMat'
 
-            dstMat = copytest(srcMat, true, true)
-            @test dstMat == srcMat
-        end
+        dstMat = copytest(srcMat, true, true)
+        @test dstMat == srcMat
     end
 end