softmax tests

FluxML · Aug 22, 2017 · 19cbce5 · 19cbce5
1 parent 6379100
commit 19cbce5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/test/activation.jl b/test/activation.jl
@@ -0,0 +1,9 @@
+@testset "Activation Functions" begin
+
+xs = rand(5,5)
+
+@test all(sum(softmax(xs), 1) .≈ 1)
+
+@test sum(softmax(vec(xs))) ≈ 1
+
+end
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,5 +1,8 @@
 using NNlib
 using Base.Test
 
-# write your own tests here
-@test 1 == 2
+@testset "NNlib" begin
+
+include("activation.jl")
+
+end