Add tests for reshape.

JuliaGPU · Feb 7, 2019 · 4cfb6e9 · 4cfb6e9
1 parent 74a0499
commit 4cfb6e9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -47,6 +47,10 @@ Adapt.adapt_structure(to, xs::Wrapper) = Wrapper(adapt(to, xs.arr))
 @test adapt(CustomArray, view(val.arr,:,:)) == view(val,:,:)
 @test adapt(CustomArray, view(val.arr,:,:)) isa SubArray{<:Any,<:Any,<:CustomArray}
 
+# NOTE: manual creation of ReshapedArray because Base.Array has an optimized `reshape`
+@test adapt(CustomArray, Base.ReshapedArray(val.arr,(2,2),())) == reshape(val,(2,2))
+@test adapt(CustomArray, Base.ReshapedArray(val.arr,(2,2),())) isa Base.ReshapedArray{<:Any,<:Any,<:CustomArray}
+
 
 using LinearAlgebra