Mercurial > repos > public > sbplib_julia

annotate LazyTensors/test/runtests.jl @ 204:9d9f7b0e514b boundary_conditions

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Modify the implementation of elementwise operations to be more direct and add a few tests
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 24 Jun 2019 14:04:38 +0200
parents b5c9be7f391c
children 70e1f3401d82
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
184
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
2 using LazyTensors
180
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 @testset "Generic Mapping methods" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
5 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
184
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
6 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7 @test range_dim(DummyMapping{Int,2,3}()) == 2
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 @test domain_dim(DummyMapping{Int,2,3}()) == 3
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 @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
10 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
12 @testset "Generic Operator methods" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
13 struct DummyOperator{T,D} <: TensorOperator{T,D} end
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
14 @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
15 @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
16 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18 @testset "Mapping transpose" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
19 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
20
184
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
21 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
22 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply_transpose
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
23
184
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
24 LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size) where {T,R,D} = :range_size
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
25 LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size) where {T,R,D} = :domain_size
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
26
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
27 m = DummyMapping{Float64,2,3}()
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
28 @test m'' == m
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
29 @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
30 @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
31 @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
32
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
33 @test range_size(m', (0,0)) == :domain_size
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34 @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
35 end
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
37 @testset "TensorApplication" begin
186
715ff09bb2ce Rename and export types in LazyTensors follow julia conventions
Jonatan Werpers <jonatan@werpers.com>
parents: 184
diff changeset
38 struct DummyMapping{T,R,D} <: TensorMapping{T,R,D} end
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
39
184
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
40 LazyTensors.apply(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = (:apply,v,i)
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
41 LazyTensors.apply_transpose(m::DummyMapping{T,R,D}, v, i) where {T,R,D} = :apply_transpose
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42
184
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
43 LazyTensors.range_size(m::DummyMapping{T,R,D}, domain_size) where {T,R,D} = 2 .* domain_size
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
44 LazyTensors.domain_size(m::DummyMapping{T,R,D}, range_size) where {T,R,D} = range_size.รท2
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
45
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
46
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47 m = DummyMapping{Int, 1, 1}()
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
48 v = [0,1,2]
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49 @test m*v isa AbstractVector{Int}
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
50 @test size(m*v) == 2 .*size(v)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
51 @test (m*v)[0] == (:apply,v,0)
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
52 @test m*m*v isa AbstractVector{Int}
191
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
53 @test (m*m*v)[1] == (:apply,m*v,1)
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
54 @test (m*m*v)[3] == (:apply,m*v,3)
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
55 @test (m*m*v)[6] == (:apply,m*v,6)
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
56 @test_broken BoundsError == (m*m*v)[0]
25d2ef206fe9 Add a few test cases for TensorApplication
Jonatan Werpers <jonatan@werpers.com>
parents: 186
diff changeset
57 @test_broken BoundsError == (m*m*v)[7]
198
b5c9be7f391c Fix up some formatting
Jonatan Werpers <jonatan@werpers.com>
parents: 191
diff changeset
58 end
204
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
59
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
60
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
61 @testset "Lazy elementwise operations" begin
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
62 struct DummyLazyArray{D} <: LazyArray{Int,D}
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
63 size::NTuple{D,Int}
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
64 end
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
65 Base.size(a::DummyLazyArray) = a.size
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
66 Base.getindex(a::DummyLazyArray, I...) = prod(I)
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
67
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
68
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
69 a = DummyLazyArray((3,))
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
70
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
71 @test (a+a)[3] == 6
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
72 @test (a-a)[3] == 0
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
73 @test (a*a)[3] == 9
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
74
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
75 a = DummyLazyArray((4,))
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
76 b = [3,2,1,2]
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
77
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
78 @test (a+b)[4] == 6
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
79 @test (a-b)[4] == 2
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
80 @test (a*b)[4] == 8
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
81
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
82 @test (b+a)[4] == 6
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
83 @test (b-a)[4] == -2
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
84 @test (b*a)[4] == 8
9d9f7b0e514b Modify the implementation of elementwise operations to be more direct and add a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 198
diff changeset
85 end