More tests

Author: mtfishman

test/test_fillarrays.jl

@@ -48,6 +48,38 @@ using TestExtras: @constinferred
   @test arg1(permutedims!(b, a, (2, 1))) ≡ Eye(2)
   @test arg2(permutedims!(b, a, (2, 1))) == permutedims(arg2(a), (2, 1))
 
+  a = randn(3, 3) ⊗ Eye(2)
+  @test size(a) == (6, 6)
+  @test a + a == (2 * arg1(a)) ⊗ Eye(2)
+  @test 2a == (2 * arg1(a)) ⊗ Eye(2)
+  @test a * a == (arg1(a) * arg1(a)) ⊗ Eye(2)
+  @test arg2(a[(:) × (:), (:) × (:)]) ≡ Eye(2)
+  @test arg2(view(a, (:) × (:), (:) × (:))) ≡ Eye(2)
+  @test arg2(a[Base.Slice(Base.OneTo(2)) × (:), (:) × (:)]) ≡ Eye(2)
+  @test arg2(view(a, Base.Slice(Base.OneTo(2)) × (:), (:) × (:))) ≡ Eye(2)
+  @test arg2(a[(:) × (:), Base.Slice(Base.OneTo(2)) × (:)]) ≡ Eye(2)
+  @test arg2(view(a, (:) × (:), Base.Slice(Base.OneTo(2)) × (:))) ≡ Eye(2)
+  @test arg2(a[Base.Slice(Base.OneTo(2)) × (:), Base.Slice(Base.OneTo(2)) × (:)]) ≡ Eye(2)
+  @test arg2(view(a, Base.Slice(Base.OneTo(2)) × (:), Base.Slice(Base.OneTo(2)) × (:))) ≡
+    Eye(2)
+  @test arg2(adapt(JLArray, a)) ≡ Eye(2)
+  @test arg1(adapt(JLArray, a)) == jl(arg1(a))
+  @test arg1(adapt(JLArray, a)) isa JLArray
+  @test arg2(similar(a, (cartesianrange(2 × 3), cartesianrange(2 × 3)))) ≡ Eye(3)
+  @test arg2(similar(typeof(a), (cartesianrange(2 × 3), cartesianrange(2 × 3)))) ≡ Eye(3)
+  @test arg2(similar(a, Float32, (cartesianrange(2 × 3), cartesianrange(2 × 3)))) ≡
+    Eye{Float32}(3)
+  @test arg2(copy(a)) ≡ Eye(2)
+  @test arg2(copy(a)) == arg2(a)
+  b = similar(a)
+  @test arg2(copyto!(b, a)) ≡ Eye(2)
+  @test arg2(copyto!(b, a)) == arg2(a)
+  @test arg2(permutedims(a, (2, 1))) ≡ Eye(2)
+  @test arg1(permutedims(a, (2, 1))) == permutedims(arg1(a), (2, 1))
+  b = similar(a)
+  @test arg2(permutedims!(b, a, (2, 1))) ≡ Eye(2)
+  @test arg1(permutedims!(b, a, (2, 1))) == permutedims(arg1(a), (2, 1))
+
   a = δ(2, 2) ⊗ randn(3, 3)
   @test size(a) == (6, 6)
   @test a + a == δ(2, 2) ⊗ (2 * arg2(a))
@@ -79,6 +111,38 @@ using TestExtras: @constinferred
   @test arg1(permutedims!(b, a, (2, 1))) ≡ δ(2, 2)
   @test arg2(permutedims!(b, a, (2, 1))) == permutedims(arg2(a), (2, 1))
 
+  a = randn(3, 3) ⊗ δ(2, 2)
+  @test size(a) == (6, 6)
+  @test a + a == (2 * arg1(a)) ⊗ δ(2, 2)
+  @test 2a == (2 * arg1(a)) ⊗ δ(2, 2)
+  @test a * a == (arg1(a) * arg1(a)) ⊗ δ(2, 2)
+  @test arg2(a[(:) × (:), (:) × (:)]) ≡ δ(2, 2)
+  @test arg2(view(a, (:) × (:), (:) × (:))) ≡ δ(2, 2)
+  @test arg2(a[Base.Slice(Base.OneTo(2)) × (:), (:) × (:)]) ≡ δ(2, 2)
+  @test arg2(view(a, Base.Slice(Base.OneTo(2)) × (:), (:) × (:))) ≡ δ(2, 2)
+  @test arg2(a[(:) × (:), Base.Slice(Base.OneTo(2)) × (:)]) ≡ δ(2, 2)
+  @test arg2(view(a, (:) × (:), Base.Slice(Base.OneTo(2)) × (:))) ≡ δ(2, 2)
+  @test arg2(a[Base.Slice(Base.OneTo(2)) × (:), Base.Slice(Base.OneTo(2)) × (:)]) ≡ δ(2, 2)
+  @test arg2(view(a, Base.Slice(Base.OneTo(2)) × (:), Base.Slice(Base.OneTo(2)) × (:))) ≡
+    δ(2, 2)
+  @test arg2(adapt(JLArray, a)) ≡ δ(2, 2)
+  @test arg1(adapt(JLArray, a)) == jl(arg1(a))
+  @test arg1(adapt(JLArray, a)) isa JLArray
+  @test arg2(similar(a, (cartesianrange(2 × 3), cartesianrange(2 × 3)))) ≡ δ(3, 3)
+  @test arg2(similar(typeof(a), (cartesianrange(2 × 3), cartesianrange(2 × 3)))) ≡ δ(3, 3)
+  @test arg2(similar(a, Float32, (cartesianrange(2 × 3), cartesianrange(2 × 3)))) ≡
+    δ(Float32, (3, 3))
+  @test arg2(copy(a)) ≡ δ(2, 2)
+  @test arg2(copy(a)) == arg2(a)
+  b = similar(a)
+  @test arg2(copyto!(b, a)) ≡ δ(2, 2)
+  @test arg2(copyto!(b, a)) == arg2(a)
+  @test arg2(permutedims(a, (2, 1))) ≡ δ(2, 2)
+  @test arg1(permutedims(a, (2, 1))) == permutedims(arg1(a), (2, 1))
+  b = similar(a)
+  @test arg2(permutedims!(b, a, (2, 1))) ≡ δ(2, 2)
+  @test arg1(permutedims!(b, a, (2, 1))) == permutedims(arg1(a), (2, 1))
+
   # Views
   a = @constinferred(Eye(2) ⊗ randn(3, 3))
   b = @constinferred(view(a, (:) × (2:3), (:) × (2:3)))