Skip to content

Commit d835cf8

Browse files
committed
Add a test
1 parent b768e06 commit d835cf8

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

test/interface/fitpredict.jl

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,3 +84,10 @@ end
8484
fp = MLJBase.fitted_params(mach)
8585
@test unique(fp.classes) == [1,2,3]
8686
end
87+
88+
@testset "Fitting a classifier with verbosity=1" begin
89+
mdl = LogisticClassifier()
90+
X, y = MLJBase.@load_iris
91+
mach = MLJBase.machine(mdl, X, y)
92+
@test_logs (:info,"Training Machine{LogisticClassifier,…}.") (:info,"Solver: LBFGS()") MLJBase.fit!(mach; verbosity=1)
93+
end

0 commit comments

Comments
 (0)