fixed tests, using only onehot for adjoint

Author: racinmat

src/onehot.jl

@@ -232,7 +232,7 @@ function Base.:(*)(A::AbstractMatrix, B::OneHotLike{<:Any, L, 1}) where L
   return NNlib.gather(A, _indices(B))
 end
 
-function Base.:(*)(A::AbstractMatrix, B::Adjoint{Bool, <:OneHotLike{<:Any, L, 1}}) where L
+function Base.:(*)(A::AbstractMatrix, B::Adjoint{Bool, <:OneHotMatrix})
   B_dim = length(_indices(parent(B)))
   size(A, 2) == B_dim || throw(DimensionMismatch("Matrix column must correspond with OneHot size: $(size(A, 2)) != $B_dim"))
   return NNlib.scatter(+, A, _indices(parent(B)), dstsize=(size(A,1), size(B,2)))

test/onehot.jl

@@ -59,9 +59,9 @@ end
   @test A * b3' == A * Array(b3')
   @test transpose(X) * b2 == transpose(X) * Array(b2)
   @test A * b4 == A[:,[1, 2, 2, 2, 3, 1]]
-  @test_broken A * b5' == A[:,[1, 2, 2, 2, 3, 1]]
+  @test A * b5' == hcat(A[:,[1, 2, 3, 3]], A[:,1]+A[:,2], zeros(Int64, 3))
   @test A * b6 == hcat(A[:,1], 2*A[:,2], A[:,2], A[:,1]+A[:,2])
-  @test_broken A * b7'
+  @test A * b7' == A[:,[1, 2, 3, 1, 2, 3]]
 
   @test_throws DimensionMismatch A*b1'
   @test_throws DimensionMismatch A*b2