Mercurial > repos > public > sbplib_julia

changeset 477:79a88269d7d0 feature/outer_product

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add some tests
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 02 Nov 2020 22:12:24 +0100
parents 1b9af062ba2c
children 2dc2eac27f75
files test/testLazyTensors.jl
diffstat 1 files changed, 17 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/test/testLazyTensors.jl	Mon Nov 02 21:34:33 2020 +0100
+++ b/test/testLazyTensors.jl	Mon Nov 02 22:12:24 2020 +0100
@@ -399,7 +399,7 @@
 end
 
 
-@testset "LazyIdentityOuterProduct" begin
+@testset "LazyOuterProduct" begin
     struct ScalingOperator{T,D} <: TensorMapping{T,D,D}
         λ::T
         size::NTuple{D,Int}
@@ -430,7 +430,22 @@
     @test A⊗B == AB
     @test A⊗B⊗C == ABC
 
-    # TODO: Include some tests where the domain has different size and dimension
+    A = rand(3,2)
+    B = rand(2,4,3)
+
+    v₁ = rand(2,4,3)
+    v₂ = rand(4,3,2)
+
+    Ã = LazyLinearMap(A,(1,),(2,))
+    B̃ = LazyLinearMap(B,(1,),(2,3))
+
+    ÃB̃ = LazyOuterProduct(Ã,B̃)
+    @tullio ABv[i,k] := A[i,j]*B[k,l,m]*v₁[j,l,m]
+    @test ÃB̃*v₁ ≈ ABv
+
+    B̃Ã = LazyOuterProduct(B̃,Ã)
+    @tullio BAv[k,i] := A[i,j]*B[k,l,m]*v₂[l,m,j]
+    @test B̃Ã*v₂ ≈ BAv
 
 end