add tests for new interface

Author: BatyLeo

src/regularized/soft_argmax.jl

@@ -8,7 +8,7 @@ Corresponds to regularized prediction on the probability simplex with entropic p
 struct SoftArgmax <: AbstractRegularized end
 
 (::SoftArgmax)(z::AbstractVector; kwargs...) = soft_argmax(z)
-compute_regularization(::SoftArgmax, y::AbstractVector) = soft_argmax_regularization(y)
+compute_regularization(::SoftArgmax, y::AbstractArray) = soft_argmax_regularization(y)
 
 function soft_argmax(z::AbstractVector)
     s = exp.(z)

src/regularized/sparse_argmax.jl

@@ -8,7 +8,7 @@ Corresponds to regularized prediction on the probability simplex with square nor
 struct SparseArgmax <: AbstractRegularized end
 
 (::SparseArgmax)(z::AbstractVector; kwargs...) = sparse_argmax(z)
-compute_regularization(::SparseArgmax, y::AbstractVector) = sparse_argmax_regularization(y)
+compute_regularization(::SparseArgmax, y::AbstractArray) = sparse_argmax_regularization(y)
 
 function sparse_argmax(z::AbstractVector; kwargs...)
     p, _ = simplex_projection_and_support(z)

test/interface.jl

@@ -0,0 +1,8 @@
+@testitem "Test interfaces are correctly implemented" default_imports = false begin
+    using InferOpt, RequiredInterfaces, Test
+    const RI = RequiredInterfaces
+
+    @test RI.check_interface_implemented(AbstractRegularized, RegularizedFrankWolfe)
+    @test RI.check_interface_implemented(AbstractRegularized, SoftArgmax)
+    @test RI.check_interface_implemented(AbstractRegularized, SparseArgmax)
+end