add classify tests

cossio · May 9, 2020 · 1e6e20d · 1e6e20d · cossio · May 9, 2020
1 parent ec0e609
commit 1e6e20d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/test/classify.jl b/test/classify.jl
@@ -0,0 +1,9 @@
+using OneHot, Test, Random
+
+@testset "classify" begin
+    A = randn(5,5,4)
+    cl = OneHot.classify(A)
+    for I in CartesianIndices(cl)
+        @test maximum(A[:,I]) == A[cl[I],I]
+    end
+end
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -2,3 +2,4 @@ using SafeTestsets
 
 @safetestset "util" begin include("util.jl") end
 @safetestset "onehot" begin include("onehot.jl") end
+@safetestset "classify" begin include("classify.jl") end
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,3 +2,4 @@ using SafeTestsets

		@safetestset "util" begin include("util.jl") end
		@safetestset "onehot" begin include("onehot.jl") end
		@safetestset "classify" begin include("classify.jl") end