Mercurial > repos > public > sbplib_julia
diff LazyTensors/test/runtests.jl @ 232:a20bb4fac23d boundary_conditions
Improve tests for LazyTensors
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Wed, 26 Jun 2019 17:41:44 +0200 |
| parents | 2aa33d0eef90 |
| children | 856caf960d89 |
line wrap: on
line diff
--- a/LazyTensors/test/runtests.jl Wed Jun 26 15:06:28 2019 +0200 +++ b/LazyTensors/test/runtests.jl Wed Jun 26 17:41:44 2019 +0200 @@ -21,10 +21,11 @@ LazyTensors.apply(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply_transpose - LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size) where {T,R,D} = :range_size - LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size) where {T,R,D} = :domain_size + LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size::NTuple{D,Integer}) where {T,R,D} = :range_size + LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size::NTuple{R,Integer}) where {T,R,D} = :domain_size m = DummyMapping{Float64,2,3}() + @test m' isa TensorMapping{Float64, 3,2} @test m'' == m @test apply(m',zeros(Float64,(0,0)),0) == :apply_transpose @test apply(m'',zeros(Float64,(0,0,0)),0) == :apply @@ -40,8 +41,8 @@ LazyTensors.apply(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = (:apply,v,i) LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply_transpose - LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size) where {T,R,D} = 2 .* domain_size - LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size) where {T,R,D} = range_size.÷2 + LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size::NTuple{D,Integer}) where {T,R,D} = 2 .* domain_size + LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size::NTuple{R,Integer}) where {T,R,D} = range_size.÷2 m = DummyMapping{Int, 1, 1}()