Base testing fixes

madsjulia · May 14, 2017 · f234d03 · f234d03
1 parent 588f785
commit f234d03
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -3,19 +3,22 @@ import SVR
 currentdir = pwd()
 cd(dirname(@__FILE__))
 
-y_true = vec(readdlm("mg.result"))
+@Base.Test.testset "SVR" begin
 
-x, y = SVR.readlibsvmfile("mg.libsvm")
+    y_true = vec(readdlm("mg.result"))
 
-pmodel = SVR.train(y, x');
-y_pr = SVR.predict(pmodel, x');
-@assert maximum(abs.(y_pr .- y_true)) < 1e-4
-SVR.savemodel(pmodel, "mg.model")
-SVR.freemodel(pmodel)
+    x, y = SVR.readlibsvmfile("mg.libsvm")
 
-pmodel = SVR.loadmodel("mg.model")
-y_pr = SVR.predict(pmodel, x');
-# @assert maximum(abs.(y_pr .- y_true)) < 1e-4
-SVR.freemodel(pmodel)
+    pmodel = SVR.train(y, x');
+    y_pr = SVR.predict(pmodel, x');
+    @Base.Test.test isapprox(maximum(abs.(y_pr .- y_true)), 0, atol=1e-4)
+    SVR.savemodel(pmodel, "mg.model")
+    SVR.freemodel(pmodel)
+
+    pmodel = SVR.loadmodel("mg.model")
+    y_pr = SVR.predict(pmodel, x');
+    # @assert maximum(abs.(y_pr .- y_true)) < 1e-4
+    SVR.freemodel(pmodel)
+end
 
 cd(currentdir)