Skip to content

Commit bc77e86

Browse files
Add norm tests
1 parent bfe75da commit bc77e86

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed

test/vnorm.jl

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
Random.seed!(0xe89edc3a6fb53d11)
2+
@testset "1-dimensional norms" begin
3+
x = rand(1000)
4+
for p (0.0, 0.5, 1.0, 1.5, 2.0, 3.0, Inf, -Inf)
5+
d = norm(x, p)
6+
@test vnorm(x, p) d
7+
@test vtnorm(x, p) d
8+
end
9+
end
10+
11+
@testset "multi-dimensional norms" begin
12+
x = rand(10,10,10,10);
13+
for dims (1, 2, 3, 4, (1,2), (1,3), (1,4), (2,3), (2,4), (3,4), (1,2,3), (1,2,4), (2,3,4))
14+
x′ = eachslice(x, dims=Tuple(setdiff(1:4, dims)))
15+
for p (0.0, 0.5, 1.0, 1.5, 2.0, 3.0, Inf, -Inf)
16+
d = norm.(x′, p)
17+
@test dropdims(vnorm(x, p, dims=dims), dims=dims) d
18+
@test dropdims(vtnorm(x, p, dims=dims), dims=dims) d
19+
end
20+
end
21+
for p (0.0, 0.5, 1.0, 1.5, 2.0, 3.0, Inf, -Inf)
22+
d = norm(x, p)
23+
for dims ((1,2,3,4), (1,2,3,4,5))
24+
@test first(vnorm(x, p, dims=dims)) d
25+
@test first(vtnorm(x, p, dims=dims)) d
26+
end
27+
end
28+
end

0 commit comments

Comments
 (0)