Change labels from strings to symbols in tests.

englhardt · Nov 5, 2018 · 7fc6209 · 7fc6209
1 parent 2b635c5
commit 7fc6209
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/test/classifiers/classifier_svdd_neg_test.jl b/test/classifiers/classifier_svdd_neg_test.jl
@@ -40,7 +40,7 @@
         actual = SVDD.predict(svdd_neg, dummy_data)
 
         @test_broken expected ≈ actual
-        @test sum(actual .> 0) == sum(labels .== "outlier")
+        @test sum(actual .> 0) == sum(labels .== :outlier)
     end
 
     @testset "with outlier examples" begin
@@ -51,7 +51,7 @@
         SVDD.fit!(svdd_neg, TEST_SOLVER)
         predictions = SVDD.predict(svdd_neg, dummy_data)
         @test sum(predictions .> 0) == 0
-        pools[labels .== "outlier"] .= :Lout
+        pools[labels .== :outlier] .= :Lout
 
         SVDD.set_pools!(svdd_neg, pools)
         @test_throws SVDD.ModelStateException SVDD.predict(svdd_neg, dummy_data)
@@ -72,7 +72,7 @@
         expected = SVDD.predict(svdd_neg, dummy_data)
         @test sum(expected .> 0) > 0
 
-        pools[labels .== "inlier"] .= :Lin
+        pools[labels .== :inlier] .= :Lin
         SVDD.set_pools!(svdd_neg, pools)
 
         SVDD.fit!(svdd_neg, TEST_SOLVER)

diff --git a/test/test_utils.jl b/test/test_utils.jl
@@ -10,10 +10,10 @@ function generate_mvn_with_outliers(n_dim, n_observations,
 
     if incl_outliers
         x = hcat(inliers, outliers)
-        labels = vcat(fill("inlier", n_observations), fill("outlier", 8))
+        labels = vcat(fill(:inlier, n_observations), fill(:outlier, 8))
     else
         x = inliers
-        labels = vcat(fill("inlier", n_observations))
+        labels = vcat(fill(:inlier, n_observations))
     end
     if normalized
         x = mapslices(normalize, x, dims=2)