fixing hpdRF_parallelTree unit tests

vertica · Oct 14, 2015 · 4ae7651 · 4ae7651
1 parent 6ff7c7c
commit 4ae7651
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/algorithms/HPdclassifier/inst/tests/test_hpdRF_parallelTree.R b/algorithms/HPdclassifier/inst/tests/test_hpdRF_parallelTree.R
@@ -334,7 +334,8 @@ expect_no_error({old_rf_model <- deploy.hpdRF_parallelTree(model)})
 data = generateData(100,2,TRUE,TRUE)
 predictions = predict(old_rf_model, getpartition(data))
 responses = getpartition(data)$X1
-responses = factor(responses,levels = attr(predictions,"levels"))
+responses <- factor(responses,levels = attr(predictions,"levels"))
+names(responses) <- 1:length(responses)
 expect_equal(predictions,responses)
 })