add test

JuliaAI · Oct 15, 2024 · e9f863b · e9f863b
1 parent 8a00b3b
commit e9f863b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -5,6 +5,7 @@ test_files = [
     "traits.jl",
     "clone.jl",
     "accessor_functions.jl",
+    "target_features.jl",
     "patterns/regression.jl",
     "patterns/static_algorithms.jl",
     "patterns/ensembling.jl",

diff --git a/test/target_features.jl b/test/target_features.jl
@@ -0,0 +1,11 @@
+using Test
+using LearnAPI
+
+struct Avacado end
+
+@test isnothing(LearnAPI.target(Avacado(), "salsa"))
+@test isnothing(LearnAPI.weights(Avacado(), "salsa"))
+@test LearnAPI.features(Avacado(), "salsa") == "salsa"
+@test LearnAPI.features(Avacado(), (:X, :y)) == :X
+
+true