more tests

LuxDL · Oct 18, 2024 · 24a6c97 · 24a6c97
1 parent 39a4dea
commit 24a6c97
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/misc_tests.jl b/test/misc_tests.jl
@@ -167,12 +167,13 @@ end
         x
     end
     Functors.@functor Tleaf
-
     MLDataDevices.isleaf(::Tleaf) = true
-
     Adapt.adapt_structure(dev::CPUDevice, t::Tleaf) = Tleaf(2 .* dev(t.x))
 
     cpu = cpu_device()
     t = Tleaf(ones(2))
-    @test cpu(t).x == 2 .* ones(2)
+    t = cpu(t)
+    @test y.x == 2 .* ones(2)
+    y = cpu([(t,)])
+    @test y[1][1].x == 2 .* ones(2)
 end