more tests

cossio · Jun 12, 2020 · 78063bd · 78063bd · cossio · Jun 12, 2020
1 parent ffc3233
commit 78063bd
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 1 deletion.
diff --git a/test/array.jl b/test/array.jl
@@ -1,4 +1,5 @@
 using OneHot
+using Base: tail, front
 
 X = OneHotArray(rand(1:4, 5), 4)
 
@@ -10,5 +11,13 @@ A = randn(3,4)
 @test A * X == A[:, X.c]
 @inferred A * X
 
+B = randn(3,3)
+@test_throws DimensionMismatch B * X
+
 @test X[:,1:3] isa OneHotArray{2}
 @inferred X[:,1:3]
+
+for I in eachindex(X)
+    i = Tuple(I)
+    @test X[I] == (first(i) == X.c[tail(i)...])
+end
diff --git a/test/stats.jl b/test/stats.jl
@@ -6,6 +6,12 @@ using OneHot, Test, Random
     for I in CartesianIndices(cl)
         @test maximum(A[:,I]) == A[cl[I],I]
     end
+
+    X = OneHotArray(rand(1:4,5,5), 4)
+    cl = OneHot.classify(X)
+    for I in CartesianIndices(cl)
+        @test maximum(X[:,I]) == X[cl[I],I]
+    end
 end
 
 @testset "entropy" begin

diff --git a/test/util.jl b/test/util.jl
@@ -1,5 +1,5 @@
 using OneHot, Test, Random
-using OneHot: argmaxdrop, argmaxdropfirst, tuplen
+using OneHot: argmaxdrop, argmaxdropfirst, tuplen, columns
 
 @testset "argmaxdrop" begin
 	@inferred argmaxdrop(randn(3,2,4); dims=2)
@@ -21,3 +21,12 @@ end
     @inferred tuplen(Val(5))
     @test tuplen(Val(5)) == (1,2,3,4,5)
 end
+
+@testset "columns" begin
+	A = randn(4,5)
+	@test columns(A) == collect(eachcol(A))
+	@inferred columns(A)
+	A = randn(5,5,3)
+	@test vec(columns(A)) == collect(eachcol(reshape(A,5,15)))
+	@inferred columns(A)
+end