extend tests to cover lables on connecting lines

JuliaAI · Nov 2, 2023 · f4f28bb · f4f28bb
1 parent 1e44838
commit f4f28bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -42,7 +42,7 @@ wt = DecisionTree.wrap(dt, (featurenames = feature_names, classlabels = class_la
 @testset "visual test" begin
     # plot the tree using the `TreeRecipe`
     # this is a rather visual test, to see, if the tree gets plotted correctly
-    plot(wt)    # this calls automatically the `TreeRecipe`
+    plot(wt; connect_labels = ["yes", "no"])    # this calls automatically the `TreeRecipe`
 end
 
 @testset "structural test" begin