better coverage

src/test/scala/org/apache/spark/smile/ImplicitSuite.scala

@@ -5,6 +5,7 @@ import org.scalatest.FunSuite
 import smile.classification.knn
 import smile.data._
 import smile.validation.{Accuracy, ClassificationMeasure, _}
+import org.apache.spark.sql.functions._
 
 class ImplicitSuite extends FunSuite with DatasetSuiteBase {
 
@@ -22,4 +23,18 @@ class ImplicitSuite extends FunSuite with DatasetSuiteBase {
 
   }
 
+  test("toSmileDataset with weights") {
+
+    import org.apache.spark.smile.implicits._
+
+    val mushrooms = spark.read.format("libsvm").load("data/mushrooms.svm").withColumn("weight",lit(1.0))
+
+    val (x,y) = mushrooms.toSmileDataset().unzipInt
+
+    val res = cv(x, y, 5, Seq(new Accuracy().asInstanceOf[ClassificationMeasure]): _*) { (x, y) => knn(x, y, 3) }
+
+    assert(res(0) == 1)
+
+  }
+
 }