add a test for predict and transform slurping fallbacks

oops

Author: ablaom

src/predict_transform.jl

@@ -4,10 +4,6 @@ function DOC_IMPLEMENTED_METHODS(name; overloaded=false)
     "[`LearnAPI.functions`](@ref) trait. "
 end
 
-const OPERATIONS = (:predict, :transform, :inverse_transform)
-const DOC_OPERATIONS_LIST_SYMBOL = join(map(op -> "`:$op`", OPERATIONS), ", ")
-const DOC_OPERATIONS_LIST_FUNCTION = join(map(op -> "`LearnAPI.$op`", OPERATIONS), ", ")
-
 DOC_MUTATION(op) =
     """
 
@@ -171,8 +167,8 @@ $(DOC_MUTATION(:transform))
 $(DOC_DATA_INTERFACE(:transform))
 
 """
-transform(model, data1, data2...; kwargs...) =
-    transform(model, (data1, datas...); kwargs...) # automatic slurping
+transform(model, data1, data2, datas...; kwargs...) =
+    transform(model, (data1, data2, datas...); kwargs...) # automatic slurping
 
 """
     inverse_transform(model, data)

test/predict_transform.jl

@@ -0,0 +1,19 @@
+using Test
+using LearnAPI
+
+struct Goose end
+
+LearnAPI.fit(algorithm::Goose) = Ref(algorithm)
+LearnAPI.algorithm(::Base.RefValue{Goose}) = Goose()
+LearnAPI.predict(::Base.RefValue{Goose}, ::Point, data) = sum(data)
+LearnAPI.transform(::Base.RefValue{Goose}, data) = prod(data)
+@trait Goose kinds_of_proxy = (Point(),)
+
+@testset "predict and transform argument slurping" begin
+    model = fit(Goose())
+    @test predict(model, Point(), 2, 3, 4) == 9
+    @test predict(model, 2, 3, 4) == 9
+    @test transform(model, 2, 3, 4) == 24
+end
+
+true

test/runtests.jl

@@ -4,6 +4,7 @@ test_files = [
     "tools.jl",
     "traits.jl",
     "clone.jl",
+    "predict_transform.jl",
     "patterns/regression.jl",
     "patterns/static_algorithms.jl",
     "patterns/ensembling.jl",