Mercurial > repos > public > sbplib_julia

annotate LazyTensor/test/runtests.jl @ 182:6ae99288bc6f boundary_conditions

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Temporarily remove LazyTensor package before rename
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 20 Jun 2019 21:06:05 +0200
parents b7397ae8afaf
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
1 using Test
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
2 using LazyTensor
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
3
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
5
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
6 @testset "Generic Mapping methods" begin
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7 struct DummyMapping{T,R,D} <: LazyTensor.Mapping{T,R,D} end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 LazyTensor.apply(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 @test range_dim(DummyMapping{Int,2,3}()) == 2
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
10 @test domain_dim(DummyMapping{Int,2,3}()) == 3
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11 @test apply(DummyMapping{Int,2,3}(), zeros(Int, (0,0,0)),0) == :apply
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
12 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
13
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
14 struct DummyOperator{T,D} <: LazyTensor.Operator{T,D} end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15 @testset "Generic Operator methods" begin
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16 @test range_size(DummyOperator{Int,2}(), (3,5)) == (3,5)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17 @test domain_size(DummyOperator{Float64, 3}(), (3,3,1)) == (3,3,1)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
20 @testset "Mapping transpose" begin
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
21 struct DummyMapping{T,R,D} <: LazyTensor.Mapping{T,R,D} end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
22
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
23 LazyTensor.apply(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
24 LazyTensor.apply_transpose(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply_transpose
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
25
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
26 LazyTensor.range_size(m::DummyMapping{T,R,D}, domain_size) where {T,R,D} = :range_size
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
27 LazyTensor.domain_size(m::DummyMapping{T,R,D}, range_size) where {T,R,D} = :domain_size
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
28
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
29 m = DummyMapping{Float64,2,3}()
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
30 @test m'' == m
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
31 @test apply(m',zeros(Float64,(0,0)),0) == :apply_transpose
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
32 @test apply(m'',zeros(Float64,(0,0,0)),0) == :apply
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
33 @test apply_transpose(m', zeros(Float64,(0,0,0)),0) == :apply
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
35 @test range_size(m', (0,0)) == :domain_size
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36 @test domain_size(m', (0,0,0)) == :range_size
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
38
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
39 @testset "TensorApplication" begin
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
40 struct DummyMapping{T,R,D} <: LazyTensor.Mapping{T,R,D} end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
41
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42 LazyTensor.apply(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = (:apply,v,i)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
43 LazyTensor.apply_transpose(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply_transpose
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
44
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
45 LazyTensor.range_size(m::DummyMapping{T,R,D}, domain_size) where {T,R,D} = 2 .* domain_size
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
46 LazyTensor.domain_size(m::DummyMapping{T,R,D}, range_size) where {T,R,D} = range_size.รท2
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
48
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49 m = DummyMapping{Int, 1, 1}()
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
50 v = [0,1,2]
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
51 @test m*v isa AbstractVector{Int}
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
52 @test size(m*v) == 2 .*size(v)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
53 @test (m*v)[0] == (:apply,v,0)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
54 @test m*m*v isa AbstractVector{Int}
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
55 @test (m*m*v)[0] == (:apply,m*v,0)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
56 end