Add tests

Author: eliascarv

test/Project.toml

@@ -1,2 +1,6 @@
 [deps]
+DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
+MLJ = "add582a8-e3ab-11e8-2d5e-e98b27df1bc7"
+MLJDecisionTreeInterface = "c6f25543-311c-4c74-83dc-3ea6d1015661"
+StatsLearnModels = "c146b59d-1589-421c-8e09-a22e554fd05c"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

test/runtests.jl

@@ -1,6 +1,16 @@
-using StatsLearnModels
+import StatsLearnModels as SLM
+using MLJ, MLJDecisionTreeInterface
+using DataFrames
 using Test
 
 @testset "StatsLearnModels.jl" begin
-    # Write your tests here.
+  iris = DataFrame(load_iris())
+  input = iris[:, Not(:target)]
+  output = iris[:, [:target]]
+  Tree = @load(DecisionTreeClassifier, pkg=DecisionTree, verbosity=0)
+  train, test = partition(1:nrow(input), 0.7, rng=123)
+  fmodel = SLM.fit(Tree(), input[train, :], output[train, :])
+  pred = SLM.predict(fmodel, input[test, :])
+  accuracy = count(pred.target .== output.target[test]) / length(test)
+  @test accuracy > 0.9
 end