Fix multion_reg Spark tests

juliasilge · juliasilge · commit 441fc79c25f9 · 2020-07-06T15:45:40.000-06:00
diff --git a/tests/testthat/test_multinom_reg_spark.R b/tests/testthat/test_multinom_reg_spark.R
@@ -66,7 +66,7 @@ test_that('spark execution', {
 
   expect_equal(
     colnames(spark_class_prob),
-    c("pred_versicolor", "pred_virginica", "pred_setosa")
+    c("pred_VF", "pred_F", "pred_L", "pred_M")
   )
 
   expect_equivalent(

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,7 @@ test_that('spark execution', {`
`66`	`66`
`67`	`67`	`expect_equal(`
`68`	`68`	`colnames(spark_class_prob),`
`69`		`- c("pred_versicolor", "pred_virginica", "pred_setosa")`
	`69`	`+ c("pred_VF", "pred_F", "pred_L", "pred_M")`
`70`	`70`	`)`
`71`	`71`
`72`	`72`	`expect_equivalent(`