test: singleton dims AoS

avik-pal · web-flow · commit 1d931145ffb5 · 2024-11-17T20:41:59.000-05:00
diff --git a/test/ad.jl b/test/ad.jl
@@ -1,6 +1,9 @@
 using ArrayInterface, ReverseDiff, Tracker, Test
 x = ReverseDiff.track([4.0])
 @test ArrayInterface.aos_to_soa(x) isa ReverseDiff.TrackedArray
+x = reshape([ReverseDiff.track(rand(1, 1, 1))[1]], 1, 1, 1)
+@test ArrayInterface.aos_to_soa(x) isa ReverseDiff.TrackedArray
+@test ndims(ArrayInterface.aos_to_soa(x)) == 3
 x = reduce(vcat, ReverseDiff.track([4.0,4.0]))
 @test ArrayInterface.aos_to_soa(x) isa ReverseDiff.TrackedArray
 x = [ReverseDiff.track([4.0])[1]]
